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

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

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

{igbot_base-0.0.37 → igbot_base-0.0.39}/igbot_base/agent.py

@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 
-from additional_data import AdditionalData, EMPTY
+from igbot_base.additional_data import AdditionalData, EMPTY
 from igbot_base.agent_response import AgentResponse
 
 from igbot_base.exception_handler import ExceptionHandler, ReturnFailedResponseGracefully

{igbot_base-0.0.37 → igbot_base-0.0.39}/igbot_base/llm.py

@@ -2,7 +2,7 @@ from abc import ABC, abstractmethod
 from builtins import dict
 from typing import Optional
 
-from additional_data import AdditionalData, EMPTY
+from igbot_base.additional_data import AdditionalData, EMPTY
 from igbot_base.exception_handler import ExceptionHandler, NoopExceptionHandler
 from igbot_base.llmmemory import LlmMemory
 from igbot_base.models import Model

{igbot_base-0.0.37 → igbot_base-0.0.39}/igbot_base/tool.py

@@ -1,5 +1,12 @@
 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
 
 class Tool(ABC):
 
@@ -12,7 +19,7 @@ class Tool(ABC):
         return self._name
 
     @abstractmethod
-    def get_function(self):
+    def get_function(self) -> Callable[..., ToolResponse]:
         pass
 
     @abstractmethod

{igbot_base-0.0.37 → igbot_base-0.0.39}/igbot_base.egg-info/PKG-INFO

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

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

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