lukmanaj commited on
Commit
282cfb9
·
verified ·
1 Parent(s): 3c81963

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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