Skip to content

Commit dc2990e

Browse files
authored
Update call_llm.py
1 parent ce0b14a commit dc2990e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

utils/call_llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,8 @@ def call_llm(prompt: str, use_cache: bool = True) -> str:
5858
client = genai.Client(
5959
api_key=os.getenv("GEMINI_API_KEY", ""),
6060
)
61-
model = os.getenv("GEMINI_MODEL", "gemini-2.5-pro-exp-03-25")
62-
# model = os.getenv("GEMINI_MODEL", "gemini-2.5-flash-preview-04-17")
61+
model = os.getenv("GEMINI_MODEL", "gemini-2.5-pro")
62+
# model = os.getenv("GEMINI_MODEL", "gemini-2.5-flash")
6363

6464
response = client.models.generate_content(model=model, contents=[prompt])
6565
response_text = response.text

0 commit comments

Comments
 (0)