chore: Temporarily remove embedding env vars for code graph action (#647)
<!-- .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:
parent
cad9e0ce44
commit
9b9fe48843
3 changed files with 6 additions and 6 deletions
|
|
@ -19,9 +19,5 @@ jobs:
|
|||
secrets:
|
||||
OPENAI_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_PASSWORD: ${{ secrets.GRAPHISTRY_PASSWORD }}
|
||||
|
|
|
|||
|
|
@ -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(get_data_list_for_user, dataset_name="repo_docs", user=user),
|
||||
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(
|
||||
extract_graph_from_data, graph_model=KnowledgeGraph, task_config={"batch_size": 50}
|
||||
),
|
||||
|
|
|
|||
|
|
@ -7,7 +7,11 @@ from cognee.shared.utils import setup_logging
|
|||
|
||||
|
||||
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():
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue