fix: Shared attribute values between class instances

Author: TunixR

agent_tools/hooks.py

@@ -78,10 +78,6 @@ def intercept_response(self, event: AfterToolCallEvent):
 
 
 class ToolLoggingHook(HookProvider):
-    agent_trace_id: UUID
-    tools: dict[str, UUID]
-    trace_id: UUID
-
     def __init__(self, agent_trace_id: UUID, tools: dict[str, UUID]):
         """Initializes the ToolLoggingHook.
 
@@ -91,6 +87,7 @@ def __init__(self, agent_trace_id: UUID, tools: dict[str, UUID]):
         """
         self.agent_trace_id = agent_trace_id
         self.tools = tools
+        self.trace_id: UUID
 
     @override
     def register_hooks(self, registry: HookRegistry, **kwargs) -> None:
@@ -147,14 +144,6 @@ def log_tool_response(self, event: AfterToolCallEvent) -> None:
 
 
 class AgentLoggingHook(HookProvider):
-    parent_trace_id: UUID | None
-    agent_id: UUID
-    agent_trace_id: UUID
-    invocation_state: dict
-    messages: list[dict] = []
-    is_gui_agent: bool = False
-    created: bool = False
-
     def __init__(
         self,
         agent_id: UUID,
@@ -177,6 +166,8 @@ def __init__(
         # We generate here the id for the agent trace so we can reference it outside
         self.agent_trace_id = uuid4()
         self.created = False
+        self.messages: list[dict] = []
+        self.created: bool = False
 
     @override
     def register_hooks(self, registry: HookRegistry, **kwargs) -> None: