igbot-base 0.0.38__tar.gz → 0.0.40__tar.gz

{igbot_base-0.0.38 → igbot_base-0.0.40}/PKG-INFO

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

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

@@ -16,6 +16,8 @@ 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/vectorstore.py
 igbot_base.egg-info/PKG-INFO
 igbot_base.egg-info/SOURCES.txt

{igbot_base-0.0.38 → igbot_base-0.0.40}/pyproject.toml

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