igbot-base 0.0.11__py3-none-any.whl → 0.0.13__py3-none-any.whl

igbot_base/agent.py

@@ -2,11 +2,27 @@ from abc import ABC, abstractmethod
 
 from igbot_base.agent_response import AgentResponse
 
+from igbot_base.exception_handler import ExceptionHandler, ReturnFailedResponseGracefully
+from igbot_base.logging_adapter import get_logger
+
+logger = get_logger("application")
+
 
 class Agent(ABC):
 
-    @abstractmethod
+    def __init__(self, name, exception_handler: ExceptionHandler = ReturnFailedResponseGracefully):
+        self.__name = name
+        self.__ex_handler = exception_handler
+
     def invoke(self, query) -> AgentResponse:
+        try:
+            return self._invoke(query)
+        except Exception as e:
+            logger.exception("Exception occurred at %s for query %s: %s", self.describe(), query, e)
+            return self.__ex_handler.handle(e)
+
+    @abstractmethod
+    def _invoke(self, query) -> AgentResponse:
         pass
 
     @abstractmethod
@@ -14,4 +30,4 @@ class Agent(ABC):
         pass
 
     def __str__(self):
-        return self.describe()
+        return self.describe()

igbot_base/exception_handler.py

@@ -1,5 +1,7 @@
 from abc import ABC, abstractmethod
 
+from igbot_base.agent_response import AgentResponse
+
 
 class ExceptionHandler(ABC):
 
@@ -18,3 +20,9 @@ class PrintingExceptionHandler(ExceptionHandler):
 
     def handle(self, e: Exception):
         print(e)
+
+
+class ReturnFailedResponseGracefully(ExceptionHandler):
+
+    def handle(self, e: Exception):
+        return AgentResponse.error(str(e), e)

igbot_base/llm.py

@@ -26,7 +26,7 @@ class Llm(ABC):
         self._exception_handler = llm_exception_handler
 
     def __str__(self):
-        return f"Llm({self._name} {self._model.get_name()})"
+        return f"Llm({self._name} {self._model.value.get_name()})"
 
     @abstractmethod
     def _call(self, user_query: str, history: LlmMemory, params: dict) -> str:
@@ -36,14 +36,15 @@ class Llm(ABC):
         history.revert_to_snapshot()
 
     def call(self, user_query: str, history: LlmMemory, params: dict) -> str:
+        logger.debug("Call to %s with question: %s", self.__str__(), user_query)
         history.set_snapshot()
         try:
             response = self._call(user_query, history, params)
-            logger.debug("LLM %s %s responded: %s", self._name, self._model.get_name(), response)
+            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.get_name(), e)
+                         self._name, self._model.value.get_name(), e)
             self._revert_memory(history)
             self._exception_handler.handle(e)
 

{igbot_base-0.0.11.dist-info → igbot_base-0.0.13.dist-info}/METADATA

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

{igbot_base-0.0.11.dist-info → igbot_base-0.0.13.dist-info}/RECORD

@@ -1,8 +1,8 @@
 igbot_base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-igbot_base/agent.py,sha256=xU1jDgoyfXfSJlXcdLl1Va7BX-Z5QLXz4ZpjOuz2z9M,301
+igbot_base/agent.py,sha256=FwQ5nvk5rLm5F-QaxsghhqDZcus_at7AehXIWDIk8dw,932
 igbot_base/agent_response.py,sha256=eAps47djL_aWYn8P_BMpVGkPwO6m8nhltWWyh8qSG6E,1042
-igbot_base/exception_handler.py,sha256=BMzQpgP4jnnRuTcEItx0oMljTZmKgMRCHzpybHn88Ig,341
-igbot_base/llm.py,sha256=xwd-AyMXYtzIlGV9Hcshof_h84R_n5vcT57tjcCDI10,1913
+igbot_base/exception_handler.py,sha256=9DFAW0K0U7Y2vKGiUW-j1C0_zp48UCyO8jL3oRj9yJQ,535
+igbot_base/llm.py,sha256=_HVTrQcbIzUaGWfRWeD-Pgy1M7JkOybzogXiQSK-mH8,2012
 igbot_base/llmmemory.py,sha256=I6eSBPW1DK8UNiHhayon-DLbSbWSinO_EwM5S8C3tHA,1090
 igbot_base/logging_adapter.py,sha256=kMuAaQPCba2luk0-WPxAbPNJaBL1yZsaff35ltKWgww,256
 igbot_base/models.py,sha256=15bitUZy4Z4JhZbLd_SMx5pD7m8YuZiIq-wYGumf0ws,622
@@ -19,7 +19,7 @@ igbot_base/exception/memory_exception.py,sha256=ZGKw3dN2tMf9XHacYaVTXgvYpWi14z7u
 igbot_base/exception/prompt_exception.py,sha256=_u-O9E2Z1DK_OT4StyOmIDLfr_Mmk4k0GLe67kIoQNc,413
 igbot_base/exception/retriever_exception.py,sha256=A46sBV6_1N3jgBl-nFRw4KgEcz3oPb8d5Eb83YWDEPo,431
 igbot_base/exception/tool_exception.py,sha256=1EX7R90MplT4Vqv1-zzPslFrNvDwcrh7wgEWun2Ba6k,386
-igbot_base-0.0.11.dist-info/METADATA,sha256=qRbglwrx7br3mjX0RH6I_KjsJ-lyR6dxOL4I7ECvtTo,211
-igbot_base-0.0.11.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-igbot_base-0.0.11.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
-igbot_base-0.0.11.dist-info/RECORD,,
+igbot_base-0.0.13.dist-info/METADATA,sha256=uc9oNyCmuuP-qhhpEuqm8tYLLR6_aO1Y1oP7drjcnsA,211
+igbot_base-0.0.13.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+igbot_base-0.0.13.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
+igbot_base-0.0.13.dist-info/RECORD,,