فهرست منبع

Don't send gemini settings when the llm is not gemini (#3940)

Engel Nyst 1 سال پیش
والد
کامیت
47f60b8275
1فایلهای تغییر یافته به همراه10 افزوده شده و 2 حذف شده
  1. 10 2
      openhands/llm/llm.py

+ 10 - 2
openhands/llm/llm.py

@@ -157,7 +157,11 @@ class LLM:
             timeout=self.config.timeout,
             temperature=self.config.temperature,
             top_p=self.config.top_p,
-            safety_settings=gemini_safety_settings,
+            **(
+                {'safety_settings': gemini_safety_settings}
+                if gemini_safety_settings is not None
+                else {}
+            ),
         )
 
         if self.vision_is_active():
@@ -264,7 +268,11 @@ class LLM:
             temperature=self.config.temperature,
             top_p=self.config.top_p,
             drop_params=True,
-            safety_settings=gemini_safety_settings,
+            **(
+                {'safety_settings': gemini_safety_settings}
+                if gemini_safety_settings is not None
+                else {}
+            ),
         )
 
         async_completion_unwrapped = self._async_completion