We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ce0b14a commit dc2990eCopy full SHA for dc2990e
utils/call_llm.py
@@ -58,8 +58,8 @@ def call_llm(prompt: str, use_cache: bool = True) -> str:
58
client = genai.Client(
59
api_key=os.getenv("GEMINI_API_KEY", ""),
60
)
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")
+ model = os.getenv("GEMINI_MODEL", "gemini-2.5-pro")
+ # model = os.getenv("GEMINI_MODEL", "gemini-2.5-flash")
63
64
response = client.models.generate_content(model=model, contents=[prompt])
65
response_text = response.text
0 commit comments