From dc2990e552a4c55993ec670645524187b2c2dde9 Mon Sep 17 00:00:00 2001 From: Zachary Huang <33015448+zachary62@users.noreply.github.com> Date: Sun, 3 Aug 2025 16:42:25 -0700 Subject: [PATCH] Update call_llm.py --- utils/call_llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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