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

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):
@@ -32,8 +35,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)
 

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/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:
@@ -30,9 +33,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/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.dist-info → igbot_base-0.0.10.dist-info}/METADATA

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

{igbot_base-0.0.9.dist-info → igbot_base-0.0.10.dist-info}/RECORD

@@ -2,16 +2,17 @@ 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=XNd_HyR2jK_7XlDW-2rdSaSCmVjRLAl_fdVTC2sW0HU,1461
+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=YHJDbe-33Zs202DMh6gLY_lLDejcyqB9xmek6xnyE0Q,1106
+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=Q1ZRlsJ45ckgFZuPPQVMWuYl3W_L-HiX9VJIJ0wnjU8,842
-igbot_base-0.0.9.dist-info/METADATA,sha256=Vp252wgk01DCuFYmC6KxzXzpu1_h0QRVDSXNVNsLZh8,210
-igbot_base-0.0.9.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-igbot_base-0.0.9.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
-igbot_base-0.0.9.dist-info/RECORD,,
+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,,