Merge pull request #283 from JTCorrin/bugfix/openai_provider_tool_choice

removed hardcoded tool choice in openai provider in favour of a "get"…
MadcowD · Oct 4, 2024 · a899193 · a899193
2 parents 0a92701 + c0d9dd6
commit a899193
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ell/providers/openai.py b/src/ell/providers/openai.py
@@ -39,7 +39,7 @@ def translate_to_provider(self, ell_call : EllCallParams) -> Dict[str, Any]:
                 final_call_params.pop("stream_options", None)
             if ell_call.tools:
                 final_call_params.update(
-                    tool_choice="auto",
+                    tool_choice=final_call_params.get("tool_choice", "auto"),
                     tools=[  
                         dict(
                             type="function",