Fix type signature for open ai function_call, turns out you need a di…

…ct to force a function call
rogeriochaves · Jul 30, 2023 · 09cc56a · 09cc56a
1 parent a8e8b9e
commit 09cc56a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litechain/contrib/llms/open_ai.py b/litechain/contrib/llms/open_ai.py
@@ -275,7 +275,7 @@ def __init__(
         ],
         model: str,
         functions: Optional[List[Dict[str, Any]]] = None,
-        function_call: Optional[Union[Literal["none", "auto"], str]] = None,
+        function_call: Optional[Union[Literal["none", "auto"], Dict[str, Any]]] = None,
         temperature: Optional[float] = 0,
         max_tokens: Optional[int] = None,
         timeout: int = 5,