Merge pull request #1048 from winsonluk/fix-gemini

Fix Gemini bug when system role is provided
assafelovic · Dec 29, 2024 · eae2cde · eae2cde
2 parents 8643231 + d8afbe8
commit eae2cde
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/gpt_researcher/actions/report_generation.py b/gpt_researcher/actions/report_generation.py
@@ -245,7 +245,22 @@ async def generate_report(
             llm_kwargs=cfg.llm_kwargs,
             cost_callback=cost_callback,
         )
-    except Exception as e:
-        print(f"Error in generate_report: {e}")
+    except:
+        try:
+            report = await create_chat_completion(
+                model=cfg.smart_llm_model,
+                messages=[
+                    {"role": "user", "content": f"{agent_role_prompt}\n\n{content}"},
+                ],
+                temperature=0.35,
+                llm_provider=cfg.smart_llm_provider,
+                stream=True,
+                websocket=websocket,
+                max_tokens=cfg.smart_token_limit,
+                llm_kwargs=cfg.llm_kwargs,
+                cost_callback=cost_callback,
+            )
+        except Exception as e:
+            print(f"Error in generate_report: {e}")
 
     return report