From 7f2c9757eb8ad0064909bbd4e6d26eb2e639df24 Mon Sep 17 00:00:00 2001 From: Gal Shubeli Date: Tue, 11 Feb 2025 10:22:36 +0200 Subject: [PATCH 1/2] update-gemini-model --- api/llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/llm.py b/api/llm.py index a13da86..0b61aff 100644 --- a/api/llm.py +++ b/api/llm.py @@ -237,7 +237,7 @@ def _define_ontology() -> Ontology: def _create_kg_agent(repo_name: str): global ontology - model_name = os.getenv('MODEL_NAME', 'gemini/gemini-2.0-flash-exp') + model_name = os.getenv('MODEL_NAME', 'gemini/gemini-2.0-flash-001') model = LiteModel(model_name) From 7be2007b59376860e8d44ef88b3668fa8c6edeb3 Mon Sep 17 00:00:00 2001 From: Guy Korland Date: Tue, 11 Feb 2025 11:25:16 +0200 Subject: [PATCH 2/2] Update default model name in _create_kg_agent --- api/llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/llm.py b/api/llm.py index 0b61aff..bd45a95 100644 --- a/api/llm.py +++ b/api/llm.py @@ -237,7 +237,7 @@ def _define_ontology() -> Ontology: def _create_kg_agent(repo_name: str): global ontology - model_name = os.getenv('MODEL_NAME', 'gemini/gemini-2.0-flash-001') + model_name = os.getenv('MODEL_NAME', 'gemini/gemini-2.0-flash') model = LiteModel(model_name)