diff --git a/utils/call_llm.py b/utils/call_llm.py index 81ef2a2..145e493 100644 --- a/utils/call_llm.py +++ b/utils/call_llm.py @@ -58,8 +58,8 @@ def call_llm(prompt: str, use_cache: bool = True) -> str: client = genai.Client( api_key=os.getenv("GEMINI_API_KEY", ""), ) - model = os.getenv("GEMINI_MODEL", "gemini-2.5-pro-exp-03-25") - # model = os.getenv("GEMINI_MODEL", "gemini-2.5-flash-preview-04-17") + model = os.getenv("GEMINI_MODEL", "gemini-2.5-pro") + # model = os.getenv("GEMINI_MODEL", "gemini-2.5-flash") response = client.models.generate_content(model=model, contents=[prompt]) response_text = response.text