diff --git a/.github/dependabot.yml b/.github/dependabot.yml new file mode 100644 index 0000000..10048ed --- /dev/null +++ b/.github/dependabot.yml @@ -0,0 +1,7 @@ +version: 2 +updates: + - package-ecosystem: "pip" # See documentation for possible values + directory: "/" # Location of package manifests + schedule: + interval: "daily" + target-branch: "staging" diff --git a/api/llm.py b/api/llm.py index 3e1a99d..b28c407 100644 --- a/api/llm.py +++ b/api/llm.py @@ -1,3 +1,4 @@ +import os import logging from graphrag_sdk.models.openai import OpenAiGenerativeModel @@ -201,6 +202,10 @@ def _create_kg_agent(repo_name: str): name=repo_name, ontology=ontology, model_config=KnowledgeGraphModelConfig.with_model(openapi_model), + host=os.getenv('FALKORDB_HOST', 'localhost'), + port=os.getenv('FALKORDB_PORT', 6379), + username=os.getenv('FALKORDB_USERNAME', None), + password=os.getenv('FALKORDB_PASSWORD', None), ) return code_graph_kg.chat_session()