gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: Temporarily remove embedding env vars for code graph action (#647)

Browse source 
<!-- .github/pull_request_template.md -->

## Description
Temporarily remove embedding env variables for code graph action so the
action can run

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **Chores**
- Removed legacy secret configuration from the testing workflow to
streamline the CI process and enhance maintainability.
- **Improvements**
  - Updated the argument name in the code graph pipeline for clarity.
- Enhanced the handling of results in the example script to support
asynchronous processing.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->
This commit is contained in:
Igor Ilic 2025-03-17 14:58:03 +01:00 • committed by GitHub
parent cad9e0ce44
commit 9b9fe48843
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
.github/workflows/test_code_graph_example.yml vendored
View file

@ -19,9 +19,5 @@ jobs:
secrets: secrets:
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }}
EMBEDDING_PROVIDER: "fastembed"
EMBEDDING_MODEL: "sentence-transformers/all-MiniLM-L6-v2"
EMBEDDING_DIMENSIONS: 384
EMBEDDING_MAX_TOKENS: 256
GRAPHISTRY_USERNAME: ${{ secrets.GRAPHISTRY_USERNAME }} GRAPHISTRY_USERNAME: ${{ secrets.GRAPHISTRY_USERNAME }}
GRAPHISTRY_PASSWORD: ${{ secrets.GRAPHISTRY_PASSWORD }} GRAPHISTRY_PASSWORD: ${{ secrets.GRAPHISTRY_PASSWORD }}

2
cognee/api/v1/cognify/code_graph_pipeline.py
View file

@ -61,7 +61,7 @@ async def run_code_graph_pipeline(repo_path, include_docs=False):
Task(ingest_data, dataset_name="repo_docs", user=user), Task(ingest_data, dataset_name="repo_docs", user=user),
Task(get_data_list_for_user, dataset_name="repo_docs", user=user), Task(get_data_list_for_user, dataset_name="repo_docs", user=user),
Task(classify_documents), Task(classify_documents),
Task(extract_chunks_from_documents, max_chunk_tokens=get_max_chunk_tokens()), Task(extract_chunks_from_documents, max_chunk_size=get_max_chunk_tokens()),
Task( Task(
extract_graph_from_data, graph_model=KnowledgeGraph, task_config={"batch_size": 50} extract_graph_from_data, graph_model=KnowledgeGraph, task_config={"batch_size": 50}
), ),

6
examples/python/code_graph_example.py
View file

@ -7,7 +7,11 @@ from cognee.shared.utils import setup_logging
async def main(repo_path, include_docs): async def main(repo_path, include_docs):
return await run_code_graph_pipeline(repo_path, include_docs) run_status = False
async for run_status in run_code_graph_pipeline(repo_path, include_docs=include_docs):
run_status = run_status
return run_status
def parse_args(): def parse_args():