mgbam commited on
Commit
0925637
·
verified ·
1 Parent(s): c1b748e

Update orchestrator/gemini.py

Browse files
Files changed (1) hide show
  1. orchestrator/gemini.py +2 -1
orchestrator/gemini.py CHANGED
@@ -21,4 +21,5 @@ def gemini_generate(prompt, model="gemini-1.5-flash"):
21
  genai.configure(api_key=api_key)
22
  model_obj = genai.GenerativeModel(model)
23
  response = model_obj.generate_content(prompt)
24
- return response.text if hasattr(response, \"text\") else str(response)
 
 
21
  genai.configure(api_key=api_key)
22
  model_obj = genai.GenerativeModel(model)
23
  response = model_obj.generate_content(prompt)
24
+ # Correct line:
25
+ return response.text if hasattr(response, "text") else str(response)