igbot-base 0.0.34__py3-none-any.whl → 0.0.36__py3-none-any.whl

igbot_base/agent.py

@@ -22,10 +22,21 @@ class Agent(ABC):
             logger.exception("Exception occurred at %s for query %s: %s", self.describe(), query, e)
             return self.__ex_handler.handle(e)
 
+    def add_llm_message(self, llm_message, memory: LlmMemory) -> AgentResponse:
+        try:
+            return self._add_llm_message(llm_message, memory)
+        except Exception as e:
+            logger.exception("Exception occurred at %s for llm_message %s: %s", self.describe(), llm_message, e)
+            return self.__ex_handler.handle(e)
+
     @abstractmethod
     def _invoke(self, query, memory: LlmMemory) -> AgentResponse:
         pass
 
+    @abstractmethod
+    def _add_llm_message(self, llm_message, memory: LlmMemory) -> AgentResponse:
+        pass
+
     @abstractmethod
     def describe(self):
         pass

igbot_base/llm.py

@@ -34,6 +34,10 @@ class Llm(ABC):
     def _call(self, user_query: str, history: LlmMemory, params: dict) -> str:
         pass
 
+    @abstractmethod
+    def _add_llm_message(self, user_query: str, history: LlmMemory, params: dict) -> str:
+        pass
+
     def _revert_memory(self, history: LlmMemory):
         history.revert_to_snapshot()
 
@@ -51,6 +55,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-0.0.34.dist-info → igbot_base-0.0.36.dist-info}/METADATA

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

{igbot_base-0.0.34.dist-info → igbot_base-0.0.36.dist-info}/RECORD

@@ -1,9 +1,9 @@
 igbot_base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-igbot_base/agent.py,sha256=pDpJ6DkzgAAli9T53bDSqzo6bWB6IxczfZYi9LjkDzM,1021
+igbot_base/agent.py,sha256=yrZ4XSrR50g0xNpkUCnkgxljA8lOZnp8XqRMtNAss1M,1483
 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=dBZo7utmmy1BJM34SMujP8iyidJnU6c1iAjLNhwYz2E,3071
 igbot_base/llmmemory.py,sha256=eMjkjFMKgyAqpM7-q3xzQAP6gDeNfkpPkb9MK1hN8iY,1157
 igbot_base/logging_adapter.py,sha256=kMuAaQPCba2luk0-WPxAbPNJaBL1yZsaff35ltKWgww,256
 igbot_base/models.py,sha256=kbscj0jTNkJu9cglEQILjdOWJuwD3pZ0Q1rxz2q7tg8,1370
@@ -15,7 +15,7 @@ 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/vectorstore.py,sha256=f-S9DmOuehC0rtVAPGz22Guo7ddb1jVw7BygzFJM624,928
-igbot_base-0.0.34.dist-info/METADATA,sha256=rHU6pBorvGfY7FgKB89HX3zIuE5cK4bxPh6Ol9s0OSc,235
-igbot_base-0.0.34.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-igbot_base-0.0.34.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
-igbot_base-0.0.34.dist-info/RECORD,,
+igbot_base-0.0.36.dist-info/METADATA,sha256=NC8x8EdzFLTvGUqSbv3L_odPpTIZOfl7GwuwjCXmfVQ,235
+igbot_base-0.0.36.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+igbot_base-0.0.36.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
+igbot_base-0.0.36.dist-info/RECORD,,

{igbot_base-0.0.34.dist-info → igbot_base-0.0.36.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.3.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any