Spaces:
Running
Running
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)
|