File tree Expand file tree Collapse file tree 3 files changed +1443
-382
lines changed Expand file tree Collapse file tree 3 files changed +1443
-382
lines changed Original file line number Diff line number Diff line change @@ -199,15 +199,13 @@ def _define_ontology() -> Ontology:
199
199
def _create_kg_agent (repo_name : str ):
200
200
global ontology
201
201
202
- openapi_model = OpenAiGenerativeModel ("gpt-4o" )
203
- #gemini_model = GeminiGenerativeModel("gemini-1.5-flash-001")
204
- #gemini_model_pro = GeminiGenerativeModel("gemini-1.5-pro")
202
+ model = LiteModel (model_name = "gemini/gemini-2.0-flash-exp" )
205
203
206
204
#ontology = _define_ontology()
207
205
code_graph_kg = KnowledgeGraph (
208
206
name = repo_name ,
209
207
ontology = ontology ,
210
- model_config = KnowledgeGraphModelConfig .with_model (openapi_model ),
208
+ model_config = KnowledgeGraphModelConfig .with_model (model ),
211
209
host = os .getenv ('FALKORDB_HOST' , 'localhost' ),
212
210
port = os .getenv ('FALKORDB_PORT' , 6379 ),
213
211
username = os .getenv ('FALKORDB_USERNAME' , None ),
You can’t perform that action at this time.
0 commit comments