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

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

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

@@ -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 → igbot_base-0.0.36}/pyproject.toml

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