igbot-base 0.0.17__tar.gz → 0.0.19__tar.gz

{igbot_base-0.0.17 → igbot_base-0.0.19}/PKG-INFO

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

{igbot_base-0.0.17 → igbot_base-0.0.19}/igbot_base/agent.py

@@ -4,6 +4,7 @@ from igbot_base.agent_response import AgentResponse
 
 from igbot_base.exception_handler import ExceptionHandler, ReturnFailedResponseGracefully
 from igbot_base.logging_adapter import get_logger
+from llmmemory import LlmMemory
 
 logger = get_logger("application")
 
@@ -14,15 +15,15 @@ class Agent(ABC):
         self.__name = name
         self.__ex_handler = exception_handler
 
-    def invoke(self, query) -> AgentResponse:
+    def invoke(self, query, memory: LlmMemory) -> AgentResponse:
         try:
-            return self._invoke(query)
+            return self._invoke(query, memory)
         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:
+    def _invoke(self, query, memory: LlmMemory) -> AgentResponse:
         pass
 
     @abstractmethod

{igbot_base-0.0.17 → igbot_base-0.0.19}/igbot_base/agent_response.py

@@ -1,7 +1,6 @@
 from enum import Enum
 from typing import Optional
 
-
 class Type(Enum):
     SUCCESS = 1,
     NO_CONTENT = 2,

{igbot_base-0.0.17 → igbot_base-0.0.19}/igbot_base/base_exception.py

@@ -1,9 +1,4 @@
-from igbot_base.agent import Agent
-from igbot_base.llm import Llm
-from igbot_base.llmmemory import LlmMemory
-from igbot_base.prompt_template import Prompt
-from igbot_base.retriever import Retriever
-from igbot_base.tool import Tool
+
 
 
 class IgBotBaseException(Exception):
@@ -22,9 +17,9 @@ class IgBotBaseException(Exception):
 
 class BaseAgentException(IgBotBaseException):
 
-    def __init__(self, message, agent: Agent, cause: Exception = None):
+    def __init__(self, message, agent: object, cause: Exception = None):
         super().__init__(message, cause)
-        self.agent = agent
+        self.agent = agent.__str__()
 
     def __str__(self):
         result = super().__str__()
@@ -33,9 +28,9 @@ class BaseAgentException(IgBotBaseException):
 
 class BaseLlmException(IgBotBaseException):
 
-    def __init__(self, message, llm: Llm, cause: Exception = None):
+    def __init__(self, message, llm: object, cause: Exception = None):
         super().__init__(message, cause)
-        self.llm = llm
+        self.llm = llm.__str__()
 
     def __str__(self):
         result = super().__str__()
@@ -44,9 +39,9 @@ class BaseLlmException(IgBotBaseException):
 
 class BaseMemoryException(IgBotBaseException):
 
-    def __init__(self, message, memory: LlmMemory, cause: Exception = None):
+    def __init__(self, message, memory: object, cause: Exception = None):
         super().__init__(message, cause)
-        self.memory = memory
+        self.memory = memory.__str__()
 
     def __str__(self):
         result = super().__str__()
@@ -55,9 +50,9 @@ class BaseMemoryException(IgBotBaseException):
 
 class BasePromptException(IgBotBaseException):
 
-    def __init__(self, message, prompt: Prompt, cause: Exception = None):
+    def __init__(self, message, prompt: object, cause: Exception = None):
         super().__init__(message, cause)
-        self.prompt = prompt
+        self.prompt = prompt.__str__()
 
     def __str__(self):
         result = super().__str__()
@@ -66,9 +61,9 @@ class BasePromptException(IgBotBaseException):
 
 class BaseRetrieverException(IgBotBaseException):
 
-    def __init__(self, message, retriever: Retriever, cause: Exception = None):
+    def __init__(self, message, retriever: object, cause: Exception = None):
         super().__init__(message, cause)
-        self.retriever = retriever
+        self.retriever = retriever.__str__()
 
     def __str__(self):
         result = super().__str__()
@@ -77,9 +72,9 @@ class BaseRetrieverException(IgBotBaseException):
 
 class BaseToolException(IgBotBaseException):
 
-    def __init__(self, message, tool: Tool, cause: Exception = None):
+    def __init__(self, message, tool: object, cause: Exception = None):
         super().__init__(message, cause)
-        self.tool = tool
+        self.tool = tool.__str__()
 
     def __str__(self):
         result = super().__str__()

{igbot_base-0.0.17 → igbot_base-0.0.19}/igbot_base.egg-info/PKG-INFO

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

{igbot_base-0.0.17 → igbot_base-0.0.19}/pyproject.toml

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