Files
temporal-ai-agent/models/tool_definitions.py
2025-01-01 16:57:08 -08:00

33 lines
509 B
Python

from dataclasses import dataclass, field
from typing import List, Dict, Any
@dataclass
class ToolArgument:
name: str
type: str
description: str
@dataclass
class ToolDefinition:
name: str
description: str
arguments: List[ToolArgument]
@dataclass
class ToolsData:
tools: List[ToolDefinition]
@dataclass
class ToolInvocation:
tool: str
args: Dict[str, Any]
@dataclass
class MultiToolSequence:
tool_invocations: List[ToolInvocation] = field(default_factory=list)