Skip to content

Commit 5c7a0dd

Browse files
committed
Refactor tool call processing and add provider mapping
1 parent d4a84b3 commit 5c7a0dd

File tree

3 files changed

+533
-535
lines changed

3 files changed

+533
-535
lines changed

defog/llm/models.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,13 @@
33
from typing import Optional, Union, Dict, Any, Literal
44

55

6+
class Provider(Enum):
7+
ANTHROPIC = "anthropic"
8+
OPENAI = "openai"
9+
GEMINI = "gemini"
10+
TOGETHER = "together"
11+
12+
613
class OpenAIFunctionSpecs(BaseModel):
714
name: str # name of the function to call
815
description: Optional[str] = None # description of the function

0 commit comments

Comments
 (0)