Spaces:
Sleeping
Sleeping
Update orchestrator/gemini.py
Browse files- 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 |
-
|
|
|
|
|
|
| 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)
|