dm-aioaiagent 0.4.5__tar.gz → 0.4.7__tar.gz

{dm_aioaiagent-0.4.5 → dm_aioaiagent-0.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: dm-aioaiagent
-Version: 0.4.5
+Version: 0.4.7
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621

{dm_aioaiagent-0.4.5 → dm_aioaiagent-0.4.7}/dm_aioaiagent/ai_agent.py

@@ -205,18 +205,22 @@ class DMAIAgent:
             from langchain_anthropic import ChatAnthropic
 
             llm = ChatAnthropic(**base_kwargs)
+            bind_tool_kwargs = {"tool_choice": {
+                "type": "auto",
+                "disable_parallel_tool_use": not self._parallel_tool_calls
+            }}
         else:
             from langchain_openai import ChatOpenAI
 
             llm = ChatOpenAI(**base_kwargs)
+            bind_tool_kwargs = {"parallel_tool_calls": self._parallel_tool_calls}
 
         if self._is_tools_exists:
             self._tool_map = {t.name: t for t in self._tools}
-            llm = llm.bind_tools(self._tools, parallel_tool_calls=self._parallel_tool_calls)
+            llm = llm.bind_tools(self._tools, **bind_tool_kwargs)
 
         prompt = ChatPromptTemplate.from_messages([SystemMessage(content=self._system_message),
                                                    MessagesPlaceholder(variable_name="messages")])
-
         self._agent = prompt | llm
 
     def _init_graph(self) -> None:

{dm_aioaiagent-0.4.5 → dm_aioaiagent-0.4.7}/dm_aioaiagent/async_ai_agent.py

@@ -1,4 +1,3 @@
-import sys
 import uuid
 import asyncio
 from typing import Any
@@ -7,9 +6,6 @@ from langchain_core.messages import AIMessage, ToolMessage
 from .ai_agent import DMAIAgent
 from .types import *
 
-if sys.platform == "win32":
-    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
-
 __all__ = ["DMAioAIAgent"]
 
 

{dm_aioaiagent-0.4.5 → dm_aioaiagent-0.4.7}/dm_aioaiagent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: dm-aioaiagent
-Version: 0.4.5
+Version: 0.4.7
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621

{dm_aioaiagent-0.4.5 → dm_aioaiagent-0.4.7}/setup.py

@@ -8,7 +8,7 @@ def readme():
 
 setup(
     name='dm-aioaiagent',
-    version='v0.4.5',
+    version='v0.4.7',
     author='dimka4621',
     author_email='mismartconfig@gmail.com',
     description='This is my custom aioaiagent client',