diff --git a/models/tool_definitions.py b/models/tool_definitions.py index 739b77c..d667203 100644 --- a/models/tool_definitions.py +++ b/models/tool_definitions.py @@ -19,14 +19,3 @@ class ToolDefinition: @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) diff --git a/prompts/agent_prompt_generators.py b/prompts/agent_prompt_generators.py index bdf7c25..afee05b 100644 --- a/prompts/agent_prompt_generators.py +++ b/prompts/agent_prompt_generators.py @@ -127,7 +127,7 @@ def generate_genai_prompt( ) prompt_lines.append( """ - Example JSON: + Example JSON for question: { "args": { "dateDepart": "2025-03-26",