igbot-base 0.0.11__tar.gz → 0.0.12__tar.gz

{igbot_base-0.0.11 → igbot_base-0.0.12}/PKG-INFO

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

igbot_base-0.0.12/igbot_base/agent.py

@@ -0,0 +1,33 @@
+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):
+
+    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
+    def describe(self):
+        pass
+
+    def __str__(self):
+        return self.describe()

{igbot_base-0.0.11 → igbot_base-0.0.12}/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-0.0.11 → igbot_base-0.0.12}/igbot_base/llm.py

@@ -36,6 +36,7 @@ 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)

{igbot_base-0.0.11 → igbot_base-0.0.12}/igbot_base.egg-info/PKG-INFO

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

{igbot_base-0.0.11 → igbot_base-0.0.12}/pyproject.toml

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

igbot_base-0.0.11/igbot_base/agent.py

@@ -1,17 +0,0 @@
-from abc import ABC, abstractmethod
-
-from igbot_base.agent_response import AgentResponse
-
-
-class Agent(ABC):
-
-    @abstractmethod
-    def invoke(self, query) -> AgentResponse:
-        pass
-
-    @abstractmethod
-    def describe(self):
-        pass
-
-    def __str__(self):
-        return self.describe()