igbot-base 0.0.38__py3-none-any.whl → 0.0.40__py3-none-any.whl

igbot_base/tool.py

@@ -1,5 +1,13 @@
 from abc import ABC, abstractmethod
+from typing import Any, Optional, Callable
 
+from pydantic import BaseModel
+
+
+class ToolResponse(BaseModel):
+    message: Optional[str] = None
+    data: Optional[Any] = None
+    tool_name: str
 
 class Tool(ABC):
 
@@ -12,7 +20,7 @@ class Tool(ABC):
         return self._name
 
     @abstractmethod
-    def get_function(self):
+    def get_function(self) -> Callable[..., ToolResponse]:
         pass
 
     @abstractmethod

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-0.0.38.dist-info → igbot_base-0.0.40.dist-info}/METADATA

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

{igbot_base-0.0.38.dist-info → igbot_base-0.0.40.dist-info}/RECORD

@@ -14,9 +14,11 @@ 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=ghjuU3cK_VElHGStUl3IYf9dlRDOGM_aQucUQNUH3mU,431
+igbot_base/tool.py,sha256=Eb-w7Wiqykop1EoFcVlTvOoVmdZshS-D0F0MbVtE7sg,653
+igbot_base/tool_merger.py,sha256=_skJwyz8yuYjJ6oIvIp8QQlcgjdId3HtofUGLZgp-6s,322
+igbot_base/tool_merger_chain.py,sha256=Qhi0IUyb-TXv3tv3ZBMQb2-fAmCnHGBNKwWEzZSFjRY,552
 igbot_base/vectorstore.py,sha256=f-S9DmOuehC0rtVAPGz22Guo7ddb1jVw7BygzFJM624,928
-igbot_base-0.0.38.dist-info/METADATA,sha256=6z_Qjyudq-i-IKUZ9Pz-vB_WNvHjUVFqbSZ_2OgShwQ,235
-igbot_base-0.0.38.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-igbot_base-0.0.38.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
-igbot_base-0.0.38.dist-info/RECORD,,
+igbot_base-0.0.40.dist-info/METADATA,sha256=6DFqe0svQtiEVr-tiqKQr_vH-GC1VmBnJWtzaijx6X8,235
+igbot_base-0.0.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+igbot_base-0.0.40.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
+igbot_base-0.0.40.dist-info/RECORD,,