From bdbb27d4a990ff812cee86e7f22db287749e7fec Mon Sep 17 00:00:00 2001
From: Steve Androulakis <steve.androulakis@gmail.com>
Date: Thu, 2 Jan 2025 10:22:35 -0800
Subject: [PATCH] cleanup

---
 models/tool_definitions.py         | 11 -----------
 prompts/agent_prompt_generators.py |  2 +-
 2 files changed, 1 insertion(+), 12 deletions(-)

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",