Fix shared state initialization when testing tools separately #165

VRSEN · Aug 15, 2024 · 89a60a2 · 89a60a2
1 parent 7f2e891
commit 89a60a2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/agency_swarm/tools/BaseTool.py b/agency_swarm/tools/BaseTool.py
@@ -12,7 +12,12 @@ class BaseTool(OpenAISchema, ABC):
  caller_agent: Any = None
  event_handler: Any = None
  one_call_at_a_time: bool = False
-
+
+ def __init__(self, **kwargs):
+ super().__init__(**kwargs)
+ if not self.shared_state:
+ self.shared_state = SharedState()
+
  class ToolConfig:
  strict: bool = False