igbot-base 0.0.39__py3-none-any.whl → 0.0.41__py3-none-any.whl

igbot_base/tool.py

@@ -7,6 +7,11 @@ from pydantic import BaseModel
 class ToolResponse(BaseModel):
     message: Optional[str] = None
     data: Optional[Any] = None
+    tool_name: str
+
+class ToolCall(BaseModel):
+    args: Optional[dict] = None
+    tool_name: str
 
 class Tool(ABC):
 

igbot_base/tool_merger.py

@@ -0,0 +1,16 @@
+from abc import ABC, abstractmethod
+
+from igbot_base.tool import ToolResponse
+
+
+class ToolMerger(ABC):
+    def __init__(self):
+        pass
+
+    @abstractmethod
+    def supports_tools(self) -> list[str]:
+        pass
+
+    @abstractmethod
+    def process_tool_results(self, tool_responses: list[ToolResponse]):
+        pass

igbot_base/tool_merger_chain.py

@@ -0,0 +1,14 @@
+from igbot_base.tool_merger import ToolMerger
+from igbot_base.tool import ToolResponse
+
+
+class ToolMergerChain:
+    def __init__(self, mergers: list[ToolMerger]):
+        self.__mergers = mergers
+
+    def process(self, tool_responses: list[ToolResponse]):
+        for merger in self.__mergers:
+            supported_tools = merger.supports_tools()
+            matching_responses = [r for r in tool_responses if r.tool_name in supported_tools]
+            if len(matching_responses) != 0:
+                merger.process_tool_results(matching_responses)

igbot_base/tool_pre_merger.py

@@ -0,0 +1,16 @@
+from abc import ABC, abstractmethod
+
+from igbot_base.tool import ToolCall
+
+
+class ToolPreMerger(ABC):
+    def __init__(self):
+        pass
+
+    @abstractmethod
+    def supports_tools(self) -> list[str]:
+        pass
+
+    @abstractmethod
+    def process_tool_results(self, tool_calls: list[ToolCall]):
+        pass

igbot_base/tool_pre_merger_chain.py

@@ -0,0 +1,14 @@
+from igbot_base.tool_pre_merger import ToolPreMerger
+from igbot_base.tool import ToolCall
+
+
+class ToolPreMergerChain:
+    def __init__(self, mergers: list[ToolPreMerger]):
+        self.__mergers = mergers
+
+    def process(self, tool_responses: list[ToolCall]):
+        for merger in self.__mergers:
+            supported_tools = merger.supports_tools()
+            matching_responses = [r for r in tool_responses if r.tool_name in supported_tools]
+            if len(matching_responses) != 0:
+                merger.process_tool_results(matching_responses)

{igbot_base-0.0.39.dist-info → igbot_base-0.0.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: igbot_base
-Version: 0.0.39
+Version: 0.0.41
 Summary: Base classes for igbot
 Author-email: Igor Kopeć <igor.kopec95@gmail.com>
 License: LGPL-3.0-or-later

{igbot_base-0.0.39.dist-info → igbot_base-0.0.41.dist-info}/RECORD

@@ -14,9 +14,13 @@ igbot_base/prompt_template.py,sha256=q7bnU2Rp2Cvm4ixEmKHCyuolTPan6CHfYVAomT5kp10
 igbot_base/response_formats.py,sha256=Sp679VfkcyxsR19ddFEHTOM7Ox0u2WBhdji_Y9pVc2M,94
 igbot_base/retriever.py,sha256=YHBHxiuPttsIAtjwwIz0fa9q2uFtsg0WoMRCorCt1PM,635
 igbot_base/tokenizer.py,sha256=WSR0GFc1eoMmyL58RAxtpb2iWCrqu9vQ2Ui-hqs4plw,1032
-igbot_base/tool.py,sha256=r0YV9KDVLyXyt2TnC4RLxK0r0llFPgDNwXw5TfhH_JA,634
+igbot_base/tool.py,sha256=Nc491Dp78g2uQLKqjHjIlFV_gzbtr9k1FM36LaUE2f8,732
+igbot_base/tool_merger.py,sha256=_skJwyz8yuYjJ6oIvIp8QQlcgjdId3HtofUGLZgp-6s,322
+igbot_base/tool_merger_chain.py,sha256=Qhi0IUyb-TXv3tv3ZBMQb2-fAmCnHGBNKwWEzZSFjRY,552
+igbot_base/tool_pre_merger.py,sha256=wgBy-Wny8cxsNJR5biaVdYjBjAC91C9owbLaGddKD7g,313
+igbot_base/tool_pre_merger_chain.py,sha256=qNuQHRZp6r-5j8GA8FOucTsUJ7G6IqVisGW3TaqdAm4,557
 igbot_base/vectorstore.py,sha256=f-S9DmOuehC0rtVAPGz22Guo7ddb1jVw7BygzFJM624,928
-igbot_base-0.0.39.dist-info/METADATA,sha256=0tx5AFH4T3wXbLg6hvVBiRtvZmeiA1sNt4MVeV7t43Y,235
-igbot_base-0.0.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-igbot_base-0.0.39.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
-igbot_base-0.0.39.dist-info/RECORD,,
+igbot_base-0.0.41.dist-info/METADATA,sha256=eoOCg_EsGPzNxUP_pzxN6vgnSrTNMZ_UYyMGzwOLQLE,235
+igbot_base-0.0.41.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+igbot_base-0.0.41.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
+igbot_base-0.0.41.dist-info/RECORD,,