Tool Call Improvements

defog/llm/models.py

@@ -1,12 +1,6 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 from enum import Enum
-from typing import Optional, Union, Dict, Any
-
-
-class OpenAIToolChoice(Enum):
-    AUTO = "auto"  # default if not provided. calls 0, 1, or multiple functions
-    REQUIRED = "required"  # calls at least 1 function
-    NONE = "none"  # calls no functions
+from typing import Optional, Union, Dict, Any, Literal
 
 
 class OpenAIFunctionSpecs(BaseModel):
@@ -17,26 +11,9 @@ class OpenAIFunctionSpecs(BaseModel):
     )
 
 
-class OpenAIForcedFunction(BaseModel):
-    # a forced function call - forces a call to one specific function
-    type: str = "function"
-    function: OpenAIFunctionSpecs
-
-
-class AnthropicToolChoice(Enum):
-    AUTO = "auto"  # default if not provided. calls 0, 1, or multiple functions
-    REQUIRED = "required"  # calls at least 1 function
-
-
 class AnthropicFunctionSpecs(BaseModel):
     name: str  # name of the function to call
     description: Optional[str] = None  # description of the function
     input_schema: Optional[Union[str, Dict[str, Any]]] = (
         None  # parameters of the function
     )
-
-
-class AnthropicForcedFunction(BaseModel):
-    # a forced function call - forces a call to one specific function
-    type: str = "function"
-    function: AnthropicFunctionSpecs