瀏覽代碼

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