igbot-base 0.0.40__py3-none-any.whl → 0.0.42__py3-none-any.whl

igbot_base/models.py

@@ -35,4 +35,12 @@ class Model(Enum):
     OPENAI_GPT_4o_MINI_JSON = ModelInfo("gpt-4o-mini-2024-07-18", lambda: OpenAI(),
                                         OpenAiTokenizer("gpt-4o"), 128_000)
     OPENAI_GPT_4_1_NANO = ModelInfo("gpt-4.1-nano", lambda: OpenAI(),
-                                    OpenAiTokenizer("gpt-4o"), 1_047_576)
+                                    OpenAiTokenizer("gpt-4o"), 1_047_576),
+    OPENAI_GPT_5 = ModelInfo("gpt-5", lambda: OpenAI(),
+                                    OpenAiTokenizer("gpt-4o"), 200_000),
+    OPENAI_GPT_5_MINI = ModelInfo("gpt-5-mini", lambda: OpenAI(),
+                             OpenAiTokenizer("gpt-4o"), 200_000),
+    OPENAI_GPT_5_NANO = ModelInfo("gpt-5-nano", lambda: OpenAI(),
+                             OpenAiTokenizer("gpt-4o"), 200_000),
+    OPENAI_GPT_5_THINKING_NANO = ModelInfo("gpt-5-thinking-nano", lambda: OpenAI(),
+                                  OpenAiTokenizer("gpt-4o"), 150_000)

igbot_base/tool.py

@@ -9,6 +9,10 @@ class ToolResponse(BaseModel):
     data: Optional[Any] = None
     tool_name: str
 
+class ToolCall(BaseModel):
+    args: Optional[dict] = None
+    tool_name: str
+
 class Tool(ABC):
 
     def __init__(

igbot_base/tool_pre_merger.py

@@ -0,0 +1,16 @@
+from abc import ABC, abstractmethod
+
+from igbot_base.tool import ToolCall
+
+
+class ToolPreMerger(ABC):
+    def __init__(self):
+        pass
+
+    @abstractmethod
+    def supports_tools(self) -> list[str]:
+        pass
+
+    @abstractmethod
+    def process_tool_results(self, tool_calls: list[ToolCall]):
+        pass

igbot_base/tool_pre_merger_chain.py

@@ -0,0 +1,14 @@
+from igbot_base.tool_pre_merger import ToolPreMerger
+from igbot_base.tool import ToolCall
+
+
+class ToolPreMergerChain:
+    def __init__(self, mergers: list[ToolPreMerger]):
+        self.__mergers = mergers
+
+    def process(self, tool_responses: list[ToolCall]):
+        for merger in self.__mergers:
+            supported_tools = merger.supports_tools()
+            matching_responses = [r for r in tool_responses if r.tool_name in supported_tools]
+            if len(matching_responses) != 0:
+                merger.process_tool_results(matching_responses)

{igbot_base-0.0.40.dist-info → igbot_base-0.0.42.dist-info}/METADATA

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

{igbot_base-0.0.40.dist-info → igbot_base-0.0.42.dist-info}/RECORD

@@ -7,18 +7,20 @@ igbot_base/exception_handler.py,sha256=Er5LbNt2ga2-e0GSzkJw66E-NS1dN_2msTRyyac1q
 igbot_base/llm.py,sha256=0i0VVtvLafH_tqpvtNMvmULj4vkoKCrntXFkiimMifI,3172
 igbot_base/llmmemory.py,sha256=eMjkjFMKgyAqpM7-q3xzQAP6gDeNfkpPkb9MK1hN8iY,1157
 igbot_base/logging_adapter.py,sha256=kMuAaQPCba2luk0-WPxAbPNJaBL1yZsaff35ltKWgww,256
-igbot_base/models.py,sha256=kbscj0jTNkJu9cglEQILjdOWJuwD3pZ0Q1rxz2q7tg8,1370
+igbot_base/models.py,sha256=wbwdj58b9GgDtPQIrtGUUPl5tK28WzQTyevZpm4av1M,1918
 igbot_base/notifier.py,sha256=-uBInGqdPBKML9ryxGnIAeB498sZGSQRga33vbSIkE0,244
 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/tokenizer.py,sha256=WSR0GFc1eoMmyL58RAxtpb2iWCrqu9vQ2Ui-hqs4plw,1032
-igbot_base/tool.py,sha256=Eb-w7Wiqykop1EoFcVlTvOoVmdZshS-D0F0MbVtE7sg,653
+igbot_base/tool.py,sha256=Nc491Dp78g2uQLKqjHjIlFV_gzbtr9k1FM36LaUE2f8,732
 igbot_base/tool_merger.py,sha256=_skJwyz8yuYjJ6oIvIp8QQlcgjdId3HtofUGLZgp-6s,322
 igbot_base/tool_merger_chain.py,sha256=Qhi0IUyb-TXv3tv3ZBMQb2-fAmCnHGBNKwWEzZSFjRY,552
+igbot_base/tool_pre_merger.py,sha256=wgBy-Wny8cxsNJR5biaVdYjBjAC91C9owbLaGddKD7g,313
+igbot_base/tool_pre_merger_chain.py,sha256=qNuQHRZp6r-5j8GA8FOucTsUJ7G6IqVisGW3TaqdAm4,557
 igbot_base/vectorstore.py,sha256=f-S9DmOuehC0rtVAPGz22Guo7ddb1jVw7BygzFJM624,928
-igbot_base-0.0.40.dist-info/METADATA,sha256=6DFqe0svQtiEVr-tiqKQr_vH-GC1VmBnJWtzaijx6X8,235
-igbot_base-0.0.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-igbot_base-0.0.40.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
-igbot_base-0.0.40.dist-info/RECORD,,
+igbot_base-0.0.42.dist-info/METADATA,sha256=A1CRDLehStRuKt1EdA9jK_z6tcTPYP-YlfqBOl_vJRk,235
+igbot_base-0.0.42.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+igbot_base-0.0.42.dist-info/top_level.txt,sha256=RGpEN0pLsnfNLoLfpw1D_nLSyZwi3ggSaDq07o4ZEAI,11
+igbot_base-0.0.42.dist-info/RECORD,,