igbot-base 0.0.18__py3-none-any.whl → 0.0.19__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
igbot_base/agent.py CHANGED
@@ -4,6 +4,7 @@ from igbot_base.agent_response import AgentResponse
4
4
 
5
5
  from igbot_base.exception_handler import ExceptionHandler, ReturnFailedResponseGracefully
6
6
  from igbot_base.logging_adapter import get_logger
7
+ from llmmemory import LlmMemory
7
8
 
8
9
  logger = get_logger("application")
9
10
 
@@ -14,15 +15,15 @@ class Agent(ABC):
14
15
  self.__name = name
15
16
  self.__ex_handler = exception_handler
16
17
 
17
- def invoke(self, query) -> AgentResponse:
18
+ def invoke(self, query, memory: LlmMemory) -> AgentResponse:
18
19
  try:
19
- return self._invoke(query)
20
+ return self._invoke(query, memory)
20
21
  except Exception as e:
21
22
  logger.exception("Exception occurred at %s for query %s: %s", self.describe(), query, e)
22
23
  return self.__ex_handler.handle(e)
23
24
 
24
25
  @abstractmethod
25
- def _invoke(self, query) -> AgentResponse:
26
+ def _invoke(self, query, memory: LlmMemory) -> AgentResponse:
26
27
  pass
27
28
 
28
29
  @abstractmethod
@@ -1,7 +1,6 @@
1
1
  from enum import Enum
2
2
  from typing import Optional
3
3
 
4
-
5
4
  class Type(Enum):
6
5
  SUCCESS = 1,
7
6
  NO_CONTENT = 2,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: igbot_base
3
- Version: 0.0.18
3
+ Version: 0.0.19
4
4
  Summary: Base classes for igbot
5
5
  Author-email: Igor Kopeć <igor.kopec95@gmail.com>
6
6
  License: LGPL-3.0-or-later
@@ -1,6 +1,6 @@
1
1
  igbot_base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
- igbot_base/agent.py,sha256=FwQ5nvk5rLm5F-QaxsghhqDZcus_at7AehXIWDIk8dw,932
3
- igbot_base/agent_response.py,sha256=eAps47djL_aWYn8P_BMpVGkPwO6m8nhltWWyh8qSG6E,1042
2
+ igbot_base/agent.py,sha256=MJX6gzS3Ch6bHZO_qQmfquBXUNhyUxsFaEmlkxj11Ms,1010
3
+ igbot_base/agent_response.py,sha256=fmZLaPlpnolXWrPVtNah20nI3Cble21DMQybPJxMvrs,1041
4
4
  igbot_base/base_exception.py,sha256=uRZXwYdsER2y_Rlah1TaSXwR6Kb9t2lNyngar0p5YFk,2146
5
5
  igbot_base/exception_handler.py,sha256=Er5LbNt2ga2-e0GSzkJw66E-NS1dN_2msTRyyac1qX8,536
6
6
  igbot_base/llm.py,sha256=0zul2Q3aiCpdhcLbTcpdSPkLdX7Dvwhhiv2V_KH7YRo,2157
@@ -13,7 +13,7 @@ igbot_base/response_formats.py,sha256=Sp679VfkcyxsR19ddFEHTOM7Ox0u2WBhdji_Y9pVc2
13
13
  igbot_base/retriever.py,sha256=YHBHxiuPttsIAtjwwIz0fa9q2uFtsg0WoMRCorCt1PM,635
14
14
  igbot_base/tool.py,sha256=ghjuU3cK_VElHGStUl3IYf9dlRDOGM_aQucUQNUH3mU,431
15
15
  igbot_base/vectorstore.py,sha256=f-S9DmOuehC0rtVAPGz22Guo7ddb1jVw7BygzFJM624,928
16
- igbot_base-0.0.18.dist-info/METADATA,sha256=Zmc1RwJdLrk6fRN8ZC3k7YejC071HuQDMUr8TXJtQXU,211
17
- igbot_base-0.0.18.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
18
- igbot_base-0.0.18.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
19
- igbot_base-0.0.18.dist-info/RECORD,,
16
+ igbot_base-0.0.19.dist-info/METADATA,sha256=hoqGiJOM-oEgXfXMtmdsH5N22MVDJICpcZT35CUMASY,211
17
+ igbot_base-0.0.19.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
18
+ igbot_base-0.0.19.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
19
+ igbot_base-0.0.19.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (78.1.0)
2
+ Generator: setuptools (78.1.1)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5