diff --git a/llm/models.py b/llm/models.py
index 5b2da9b..87b2e0e 100644
--- a/llm/models.py
+++ b/llm/models.py
@@ -171,6 +171,13 @@ class ToolCall:
     arguments: dict
 
 
+@dataclass
+class ToolResult:
+    name: str
+    output: str
+    tool_call_id: Optional[str] = None
+
+
 @dataclass
 class Prompt:
     _prompt: Optional[str]