refactor, date context

models/data_types.py

@@ -0,0 +1,15 @@
+from dataclasses import dataclass
+from typing import Optional, Deque
+from models.tool_definitions import ToolsData
+
+
+@dataclass
+class ToolWorkflowParams:
+    conversation_summary: Optional[str] = None
+    prompt_queue: Optional[Deque[str]] = None
+
+
+@dataclass
+class CombinedInput:
+    tool_params: ToolWorkflowParams
+    tools_data: ToolsData

models/tool_definitions.py

@@ -0,0 +1,21 @@
+from dataclasses import dataclass
+from typing import List
+
+
+@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]