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

make the way to count abstractions more robust

Browse Source
This commit is contained in:
zachary62
2025-05-02 09:53:40 -04:00
parent c765bff257
commit c238dba586
2 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/call_llm.py
+3 -3
View File
@@ -47,7 +47,7 @@ def call_llm(prompt: str, use_cache: bool = True) -> str:
logger.info(f"RESPONSE: {cache[prompt]}")
return cache[prompt]
# Call the LLM if not in cache or cache disabled
# # Call the LLM if not in cache or cache disabled
# client = genai.Client(
# vertexai=True,
# # TODO: change to your own project id and location
@@ -59,8 +59,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.0-flash-exp")
model = os.getenv("GEMINI_MODEL", "gemini-2.5-pro-exp-03-25")
# model = os.getenv("GEMINI_MODEL", "gemini-2.5-flash-preview-04-17")
response = client.models.generate_content(model=model, contents=[prompt])
response_text = response.text