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

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/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/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/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/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/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/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/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/llm.py

@@ -25,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
@@ -52,3 +55,4 @@ class Llm(ABC):
             args["response_format"] = self._format
 
         return args
+

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/prompt_template.py

@@ -18,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):

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/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.10.dist-info → igbot_base-0.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: igbot_base
-Version: 0.0.10
+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.11.dist-info/RECORD

@@ -0,0 +1,25 @@
+igbot_base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+igbot_base/agent.py,sha256=xU1jDgoyfXfSJlXcdLl1Va7BX-Z5QLXz4ZpjOuz2z9M,301
+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/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
+igbot_base/persistable_memory.py,sha256=867Z4l_awD23QSTFDK3uN9TEttKfDpxx9g19c7ULG1Y,167
+igbot_base/prompt_template.py,sha256=q7bnU2Rp2Cvm4ixEmKHCyuolTPan6CHfYVAomT5kp10,1573
+igbot_base/response_formats.py,sha256=Sp679VfkcyxsR19ddFEHTOM7Ox0u2WBhdji_Y9pVc2M,94
+igbot_base/retriever.py,sha256=YHBHxiuPttsIAtjwwIz0fa9q2uFtsg0WoMRCorCt1PM,635
+igbot_base/tool.py,sha256=ghjuU3cK_VElHGStUl3IYf9dlRDOGM_aQucUQNUH3mU,431
+igbot_base/vectorstore.py,sha256=f-S9DmOuehC0rtVAPGz22Guo7ddb1jVw7BygzFJM624,928
+igbot_base/exception/agent_exception.py,sha256=qDnVYMpzaiwkhO4QI-zasySFewoA7T9ee2XIe-0E-DI,395
+igbot_base/exception/base_exception.py,sha256=Hlu_hNRW4Krx41CdUtYYaxI4VknKUw1Bs1stpm2XFTY,307
+igbot_base/exception/llm_exception.py,sha256=54oAekBvzjL6x_JckqbY34m54EAh3GKRAwJJQ8ZjkvQ,377
+igbot_base/exception/memory_exception.py,sha256=ZGKw3dN2tMf9XHacYaVTXgvYpWi14z7unJBFjrEUVyM,413
+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.10.dist-info/RECORD

@@ -1,18 +0,0 @@
-igbot_base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-igbot_base/agent.py,sha256=186kf599wrK4z41RVddeh0dH_iJ-_fCTqxM7J00P8IE,188
-igbot_base/agent_response.py,sha256=eAps47djL_aWYn8P_BMpVGkPwO6m8nhltWWyh8qSG6E,1042
-igbot_base/exception_handler.py,sha256=BMzQpgP4jnnRuTcEItx0oMljTZmKgMRCHzpybHn88Ig,341
-igbot_base/llm.py,sha256=Tuj8DJtHIDnG0ebbcQfeWHV0rheI-5MvfW3Vi-yytrA,1827
-igbot_base/llmmemory.py,sha256=KslOeTSZwvHVyFglbFhPcb6_GCCXZOqHz5YBTzJ2mew,977
-igbot_base/logging_adapter.py,sha256=kMuAaQPCba2luk0-WPxAbPNJaBL1yZsaff35ltKWgww,256
-igbot_base/models.py,sha256=15bitUZy4Z4JhZbLd_SMx5pD7m8YuZiIq-wYGumf0ws,622
-igbot_base/persistable_memory.py,sha256=867Z4l_awD23QSTFDK3uN9TEttKfDpxx9g19c7ULG1Y,167
-igbot_base/prompt_template.py,sha256=hw5oZDYRpjllso8J0TYf9MrsppnqkSfwBpuMIu9e6OU,1463
-igbot_base/response_formats.py,sha256=Sp679VfkcyxsR19ddFEHTOM7Ox0u2WBhdji_Y9pVc2M,94
-igbot_base/retriever.py,sha256=TeHWi6Sy5EvmmXOnNXuWLJO1Imlj0i6CEihv3WVN0SI,523
-igbot_base/tool.py,sha256=8QwzQuokkVSITCjZuZJeEbCvJU-n2yh9D-QauQex4Ko,319
-igbot_base/vectorstore.py,sha256=f-S9DmOuehC0rtVAPGz22Guo7ddb1jVw7BygzFJM624,928
-igbot_base-0.0.10.dist-info/METADATA,sha256=klA4DKfpsEshdzy5MAIwyLpGnYYwN4nJqb0bIp47oG4,211
-igbot_base-0.0.10.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-igbot_base-0.0.10.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
-igbot_base-0.0.10.dist-info/RECORD,,