saberbud/PocketFlow-Tutorial-Codebase-Knowledge
1
0
Fork 0
mirror of https://github.com/The-Pocket/PocketFlow-Tutorial-Codebase-Knowledge.git synced 2026-08-29 08:34:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update call_llm.py

Browse Source
This commit is contained in:
Zachary Huang
2025-08-03 16:42:25 -07:00
committed by GitHub
parent ce0b14a8ef
commit dc2990e552
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/call_llm.py
+2 -2
View File
@@ -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