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

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

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

{igbot_base-0.0.9 → igbot_base-0.0.11}/igbot_base/agent.py

@@ -7,4 +7,11 @@ class Agent(ABC):
 
     @abstractmethod
     def invoke(self, query) -> AgentResponse:
-        pass
+        pass
+
+    @abstractmethod
+    def describe(self):
+        pass
+
+    def __str__(self):
+        return self.describe()

igbot_base-0.0.11/igbot_base/exception/agent_exception.py

@@ -0,0 +1,14 @@
+from igbot_base.agent import Agent
+from igbot_base.exception.base_exception import IgBotBaseException
+
+
+class BaseAgentException(IgBotBaseException):
+
+    def __init__(self, message, agent: Agent, cause: Exception = None):
+        super().__init__(message, cause)
+        self.agent = agent
+
+    def __str__(self):
+        result = super().__str__()
+        result += f" at agent {self.agent}"
+

igbot_base-0.0.11/igbot_base/exception/base_exception.py

@@ -0,0 +1,12 @@
+class IgBotBaseException(Exception):
+
+    def __init__(self, message, cause: Exception = None):
+        super().__init__(message)
+        self.cause = cause
+
+    def __str__(self):
+        result = self.args[0]
+        if self.cause:
+            result += f" Caused by: {self.cause}"
+
+        return result

igbot_base-0.0.11/igbot_base/exception/llm_exception.py

@@ -0,0 +1,13 @@
+from igbot_base.exception.base_exception import IgBotBaseException
+from igbot_base.llm import Llm
+
+
+class BaseLlmException(IgBotBaseException):
+
+    def __init__(self, message, llm: Llm,  cause: Exception = None):
+        super().__init__(message, cause)
+        self.llm = llm
+
+    def __str__(self):
+        result = super().__str__()
+        result += f" at llm {self.llm}"

igbot_base-0.0.11/igbot_base/exception/memory_exception.py

@@ -0,0 +1,14 @@
+from igbot_base.exception.base_exception import IgBotBaseException
+from igbot_base.llmmemory import LlmMemory
+
+
+class BaseMemoryException(IgBotBaseException):
+
+    def __init__(self, message, memory: LlmMemory, cause: Exception = None):
+        super().__init__(message, cause)
+        self.memory = memory
+
+    def __str__(self):
+        result = super().__str__()
+        result += f" at memory {self.memory}"
+

igbot_base-0.0.11/igbot_base/exception/prompt_exception.py

@@ -0,0 +1,14 @@
+from igbot_base.exception.base_exception import IgBotBaseException
+from igbot_base.prompt_template import Prompt
+
+
+class BasePromptException(IgBotBaseException):
+
+    def __init__(self, message, prompt: Prompt, cause: Exception = None):
+        super().__init__(message, cause)
+        self.prompt = prompt
+
+    def __str__(self):
+        result = super().__str__()
+        result += f" at prompt {self.prompt}"
+

igbot_base-0.0.11/igbot_base/exception/retriever_exception.py

@@ -0,0 +1,14 @@
+from igbot_base.exception.base_exception import IgBotBaseException
+from igbot_base.retriever import Retriever
+
+
+class BaseRetrieverException(IgBotBaseException):
+
+    def __init__(self, message, retriever: Retriever, cause: Exception = None):
+        super().__init__(message, cause)
+        self.retriever = retriever
+
+    def __str__(self):
+        result = super().__str__()
+        result += f" at retriever {self.retriever}"
+

igbot_base-0.0.11/igbot_base/exception/tool_exception.py

@@ -0,0 +1,14 @@
+from igbot_base.exception.base_exception import IgBotBaseException
+from igbot_base.tool import Tool
+
+
+class BaseToolException(IgBotBaseException):
+
+    def __init__(self, message, tool: Tool, cause: Exception = None):
+        super().__init__(message, cause)
+        self.tool = tool
+
+    def __str__(self):
+        result = super().__str__()
+        result += f" at tool {self.tool}"
+

{igbot_base-0.0.9 → igbot_base-0.0.11}/igbot_base/llm.py

@@ -5,6 +5,9 @@ from typing import Optional
 from igbot_base.exception_handler import ExceptionHandler, NoopExceptionHandler
 from igbot_base.llmmemory import LlmMemory
 from igbot_base.models import Model
+from igbot_base.logging_adapter import get_logger
+
+logger = get_logger("application")
 
 
 class Llm(ABC):
@@ -22,6 +25,9 @@ class Llm(ABC):
         self._format = response_format
         self._exception_handler = llm_exception_handler
 
+    def __str__(self):
+        return f"Llm({self._name} {self._model.get_name()})"
+
     @abstractmethod
     def _call(self, user_query: str, history: LlmMemory, params: dict) -> str:
         pass
@@ -32,8 +38,12 @@ class Llm(ABC):
     def call(self, user_query: str, history: LlmMemory, params: dict) -> str:
         history.set_snapshot()
         try:
-            return self._call(user_query, history, params)
+            response = self._call(user_query, history, params)
+            logger.debug("LLM %s %s responded: %s", self._name, self._model.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._revert_memory(history)
             self._exception_handler.handle(e)
 
@@ -45,3 +55,4 @@ class Llm(ABC):
             args["response_format"] = self._format
 
         return args
+

{igbot_base-0.0.9 → igbot_base-0.0.11}/igbot_base/llmmemory.py

@@ -50,3 +50,10 @@ class LlmMemory(ABC):
     @abstractmethod
     def set_snapshot(self):
         pass
+
+    @abstractmethod
+    def describe(self):
+        pass
+
+    def __str__(self):
+        return self.describe()

igbot_base-0.0.11/igbot_base/logging_adapter.py

@@ -0,0 +1,15 @@
+import logging
+
+try:
+    import structlog
+
+    USE_STRUCTLOG = True
+except ImportError:
+    USE_STRUCTLOG = False
+
+
+def get_logger(name: str):
+    if USE_STRUCTLOG:
+        return structlog.get_logger(name)
+    else:
+        return logging.getLogger(name)

{igbot_base-0.0.9 → igbot_base-0.0.11}/igbot_base/prompt_template.py

@@ -2,6 +2,9 @@ import re
 from typing import Optional
 
 from igbot_base.exception_handler import ExceptionHandler, PrintingExceptionHandler
+from igbot_base.logging_adapter import get_logger
+
+logger = get_logger("application")
 
 
 class Prompt:
@@ -15,6 +18,9 @@ class Prompt:
         self.__variables = variables
         self.__exception_handler = exception_handler
 
+    def __str__(self):
+        return f"Prompt({self.__get_content_short()} ({",".join(self.__variables)}))"
+
     @staticmethod
     def replace_placeholders(text: str, values: dict) -> str:
         def replacer(match):
@@ -30,9 +36,16 @@ class Prompt:
         try:
             return self.replace_placeholders(self.__content, params)
         except KeyError as e:
+            logger.exception("Exception occurred when parsing prompt %s... %s", self.__get_content_short(), e)
             self.__exception_handler.handle(e)
 
         raise Exception(f"Error while substituting values for prompt")
 
     def get_content(self) -> str:
         return self.__content
+
+    def __get_content_short(self):
+        if len(self.__content) > 15:
+            return self.__content[:15]
+        else:
+            return self.__content

{igbot_base-0.0.9 → igbot_base-0.0.11}/igbot_base/retriever.py

@@ -22,3 +22,10 @@ class Retriever(ABC):
     @abstractmethod
     def get_relevant_data(self, query: str) -> RetrieverResponse:
         pass
+
+    @abstractmethod
+    def describe(self):
+        pass
+
+    def __str__(self):
+        return self.describe()

{igbot_base-0.0.9 → igbot_base-0.0.11}/igbot_base/tool.py

@@ -18,3 +18,10 @@ class Tool(ABC):
     @abstractmethod
     def get_definition(self):
         pass
+
+    @abstractmethod
+    def describe(self):
+        pass
+
+    def __str__(self):
+        return self.describe()

{igbot_base-0.0.9 → igbot_base-0.0.11}/igbot_base/vectorstore.py

@@ -1,6 +1,9 @@
 from abc import ABC, abstractmethod
 
 from igbot_base.retriever import Retriever
+from igbot_base.logging_adapter import get_logger
+
+logger = get_logger("application")
 
 
 class Metadata:

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

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

{igbot_base-0.0.9 → igbot_base-0.0.11}/igbot_base.egg-info/SOURCES.txt

@@ -5,6 +5,7 @@ igbot_base/agent_response.py
 igbot_base/exception_handler.py
 igbot_base/llm.py
 igbot_base/llmmemory.py
+igbot_base/logging_adapter.py
 igbot_base/models.py
 igbot_base/persistable_memory.py
 igbot_base/prompt_template.py
@@ -16,4 +17,11 @@ igbot_base.egg-info/PKG-INFO
 igbot_base.egg-info/SOURCES.txt
 igbot_base.egg-info/dependency_links.txt
 igbot_base.egg-info/requires.txt
-igbot_base.egg-info/top_level.txt
+igbot_base.egg-info/top_level.txt
+igbot_base/exception/agent_exception.py
+igbot_base/exception/base_exception.py
+igbot_base/exception/llm_exception.py
+igbot_base/exception/memory_exception.py
+igbot_base/exception/prompt_exception.py
+igbot_base/exception/retriever_exception.py
+igbot_base/exception/tool_exception.py

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

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