30 lines
558 B
Python
30 lines
558 B
Python
from typing import Literal, Optional
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class LLMToolCall(BaseModel):
|
|
pass
|
|
|
|
|
|
class OpenAIToolCallFunction(BaseModel):
|
|
name: str
|
|
arguments: str
|
|
|
|
|
|
class OpenAIToolCall(LLMToolCall):
|
|
id: str
|
|
type: Literal["function"] = "function"
|
|
function: OpenAIToolCallFunction
|
|
|
|
|
|
class GoogleToolCall(LLMToolCall):
|
|
id: Optional[str] = None
|
|
name: str
|
|
arguments: Optional[dict] = None
|
|
|
|
|
|
class AnthropicToolCall(LLMToolCall):
|
|
type: Literal["tool_use"] = "tool_use"
|
|
id: str
|
|
name: str
|
|
input: object
|