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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: igbot_base
-Version: 0.0.36
+Version: 0.0.38
 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/additional_data.py

@@ -0,0 +1,17 @@
+from pydantic.v1 import BaseModel
+from typing import Any, Dict, Optional
+
+class AdditionalData(BaseModel):
+    data: Dict[str, Any]
+
+    def get(self, key: str) -> Optional[Any]:
+        """
+        Get the value associated with the key, or return None if the key doesn't exist.
+
+        :param key: The key to look for in the dictionary.
+        :return: The value associated with the key, or None.
+        """
+        return self.data.get(key)
+
+
+EMPTY = AdditionalData(data={})

{igbot_base-0.0.36 → igbot_base-0.0.38}/igbot_base/agent.py

@@ -1,5 +1,6 @@
 from abc import ABC, abstractmethod
 
+from igbot_base.additional_data import AdditionalData, EMPTY
 from igbot_base.agent_response import AgentResponse
 
 from igbot_base.exception_handler import ExceptionHandler, ReturnFailedResponseGracefully
@@ -30,7 +31,7 @@ class Agent(ABC):
             return self.__ex_handler.handle(e)
 
     @abstractmethod
-    def _invoke(self, query, memory: LlmMemory) -> AgentResponse:
+    def _invoke(self, query, memory: LlmMemory, params: AdditionalData = EMPTY) -> AgentResponse:
         pass
 
     @abstractmethod

{igbot_base-0.0.36 → igbot_base-0.0.38}/igbot_base/llm.py

@@ -2,6 +2,7 @@ from abc import ABC, abstractmethod
 from builtins import dict
 from typing import Optional
 
+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
@@ -31,7 +32,7 @@ class Llm(ABC):
         return f"Llm({self._name} {self._model.value.get_name()})"
 
     @abstractmethod
-    def _call(self, user_query: str, history: LlmMemory, params: dict) -> str:
+    def _call(self, user_query: str, history: LlmMemory, params: dict, additonal_data: AdditionalData = EMPTY) -> str:
         pass
 
     @abstractmethod

{igbot_base-0.0.36 → igbot_base-0.0.38}/igbot_base.egg-info/PKG-INFO

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

{igbot_base-0.0.36 → igbot_base-0.0.38}/igbot_base.egg-info/SOURCES.txt

@@ -1,5 +1,6 @@
 pyproject.toml
 igbot_base/__init__.py
+igbot_base/additional_data.py
 igbot_base/agent.py
 igbot_base/agent_response.py
 igbot_base/base_exception.py

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

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