Spaces:
Sleeping
Sleeping
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
|