igbot-base 0.0.39__tar.gz → 0.0.41__tar.gz

{igbot_base-0.0.39 → igbot_base-0.0.41}/PKG-INFO

@@ -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 → igbot_base-0.0.41}/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-0.0.41/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-0.0.41/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.41/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-0.0.41/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 → igbot_base-0.0.41}/igbot_base.egg-info/PKG-INFO

@@ -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 → igbot_base-0.0.41}/igbot_base.egg-info/SOURCES.txt

@@ -16,6 +16,10 @@ igbot_base/response_formats.py
 igbot_base/retriever.py
 igbot_base/tokenizer.py
 igbot_base/tool.py
+igbot_base/tool_merger.py
+igbot_base/tool_merger_chain.py
+igbot_base/tool_pre_merger.py
+igbot_base/tool_pre_merger_chain.py
 igbot_base/vectorstore.py
 igbot_base.egg-info/PKG-INFO
 igbot_base.egg-info/SOURCES.txt

{igbot_base-0.0.39 → igbot_base-0.0.41}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "igbot_base"
-version = "0.0.39"
+version = "0.0.41"
 description = "Base classes for igbot"
 authors = [{name = "Igor Kopeć", email = "igor.kopec95@gmail.com"}]
 license={text="LGPL-3.0-or-later"}