Update app.py
Browse files
app.py
CHANGED
@@ -187,7 +187,7 @@ class BasicAgent:
|
|
187 |
def __call__(self, prompt: str, **kwargs) -> str:
|
188 |
"""Directly call Gemini API for each prompt, returning a string."""
|
189 |
try:
|
190 |
-
response = client.generate_content(
|
191 |
model=model_id,
|
192 |
contents=[{"role": "user", "parts": [{"text": prompt}]}],
|
193 |
generation_config=generation_config
|
|
|
187 |
def __call__(self, prompt: str, **kwargs) -> str:
|
188 |
"""Directly call Gemini API for each prompt, returning a string."""
|
189 |
try:
|
190 |
+
response = client.models.generate_content(
|
191 |
model=model_id,
|
192 |
contents=[{"role": "user", "parts": [{"text": prompt}]}],
|
193 |
generation_config=generation_config
|