igbot-base 0.0.35__py3-none-any.whl → 0.0.39__py3-none-any.whl

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/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/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,11 @@ 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
+    def _add_llm_message(self, user_query: str, history: LlmMemory, params: dict) -> str:
         pass
 
     def _revert_memory(self, history: LlmMemory):
@@ -51,6 +56,20 @@ class Llm(ABC):
             self._exception_handler.handle(e)
             raise BaseLlmException(f"Exception occurred while calling llm api", self, e)
 
+    def add_llm_message(self, llm_message: str, history: LlmMemory, params: dict) -> str:
+        logger.debug("Call to %s with llm message: %s", self.__str__(), llm_message)
+        history.set_snapshot()
+        try:
+            response = self._add_llm_message(llm_message, history, params)
+            logger.debug("LLM %s %s responded: %s", self._name, self._model.value.get_name(), response)
+            return response
+        except Exception as e:
+            logger.error("Error occurred in LLM %s, Model %s at calling the API: %s",
+                         self._name, self._model.value.get_name(), e)
+            self._revert_memory(history)
+            self._exception_handler.handle(e)
+            raise BaseLlmException(f"Exception occurred while calling llm api", self, e)
+
     def get_additional_llm_args(self):
         args = {}
         if self._temperature is not None:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: igbot_base
-Version: 0.0.35
+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.35.dist-info → igbot_base-0.0.39.dist-info}/RECORD

@@ -1,9 +1,10 @@
 igbot_base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-igbot_base/agent.py,sha256=yrZ4XSrR50g0xNpkUCnkgxljA8lOZnp8XqRMtNAss1M,1483
+igbot_base/additional_data.py,sha256=CbuT4m4l4Kw_7yUVAyPpMMct0ED3yF91JVo79vqnXik,479
+igbot_base/agent.py,sha256=pdDPQIg60V9bpT_VQQjEQXZNoaxRlk_82HB-ffg24ro,1576
 igbot_base/agent_response.py,sha256=ujLaFkIw9he-X1YFIICTJvq_4pmzTvTF4txEfeZtKSA,1041
 igbot_base/base_exception.py,sha256=uRZXwYdsER2y_Rlah1TaSXwR6Kb9t2lNyngar0p5YFk,2146
 igbot_base/exception_handler.py,sha256=Er5LbNt2ga2-e0GSzkJw66E-NS1dN_2msTRyyac1qX8,536
-igbot_base/llm.py,sha256=0zul2Q3aiCpdhcLbTcpdSPkLdX7Dvwhhiv2V_KH7YRo,2157
+igbot_base/llm.py,sha256=0i0VVtvLafH_tqpvtNMvmULj4vkoKCrntXFkiimMifI,3172
 igbot_base/llmmemory.py,sha256=eMjkjFMKgyAqpM7-q3xzQAP6gDeNfkpPkb9MK1hN8iY,1157
 igbot_base/logging_adapter.py,sha256=kMuAaQPCba2luk0-WPxAbPNJaBL1yZsaff35ltKWgww,256
 igbot_base/models.py,sha256=kbscj0jTNkJu9cglEQILjdOWJuwD3pZ0Q1rxz2q7tg8,1370
@@ -13,9 +14,9 @@ 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=r0YV9KDVLyXyt2TnC4RLxK0r0llFPgDNwXw5TfhH_JA,634
 igbot_base/vectorstore.py,sha256=f-S9DmOuehC0rtVAPGz22Guo7ddb1jVw7BygzFJM624,928
-igbot_base-0.0.35.dist-info/METADATA,sha256=4yed4GWeLRSXW96vrlBHo7fyb5EbXtRmUMcZPxZm00Y,235
-igbot_base-0.0.35.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-igbot_base-0.0.35.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
-igbot_base-0.0.35.dist-info/RECORD,,
+igbot_base-0.0.39.dist-info/METADATA,sha256=0tx5AFH4T3wXbLg6hvVBiRtvZmeiA1sNt4MVeV7t43Y,235
+igbot_base-0.0.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+igbot_base-0.0.39.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
+igbot_base-0.0.39.dist-info/RECORD,,