Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 2 additions & 4 deletions api/llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -199,15 +199,13 @@ def _define_ontology() -> Ontology:
def _create_kg_agent(repo_name: str):
global ontology

openapi_model = OpenAiGenerativeModel("gpt-4o")
#gemini_model = GeminiGenerativeModel("gemini-1.5-flash-001")
#gemini_model_pro = GeminiGenerativeModel("gemini-1.5-pro")
model = LiteModel(model_name="gemini/gemini-2.0-flash-exp")

#ontology = _define_ontology()
code_graph_kg = KnowledgeGraph(
name=repo_name,
ontology=ontology,
model_config=KnowledgeGraphModelConfig.with_model(openapi_model),
model_config=KnowledgeGraphModelConfig.with_model(model),
host=os.getenv('FALKORDB_HOST', 'localhost'),
port=os.getenv('FALKORDB_PORT', 6379),
username=os.getenv('FALKORDB_USERNAME', None),
Expand Down
Loading
Loading