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

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

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: dm-aioaiagent
-Version: 0.4.7
+Version: 0.4.9
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
@@ -12,7 +12,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: dm-logger~=0.5.2
+Requires-Dist: dm-logger~=0.6.2
 Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: pydantic<3.0.0,>=2.9.2
 Requires-Dist: langchain~=0.3.0
@@ -48,6 +48,16 @@ Dynamic: summary
 
 Analogue to `DMAioAIAgent` is the synchronous client `DMAIAgent`.
 
+### Windows Setup
+
+```python
+import asyncio
+import sys
+
+if sys.platform == "win32":
+    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+```
+
 ### Use agent *with* inner memory and run *single* message
 
 By default, agent use inner memory to store the conversation history.
@@ -172,30 +182,21 @@ if __name__ == "__main__":
 
 ### Set custom logger
 
-_If you want set up custom logger_
-
 ```python
-from dm_aioaiagent import DMAioAIAgent
-
-
-# create custom logger
-class MyLogger:
-    def debug(self, message):
-        pass
-
-    def info(self, message):
-        pass
-
-    def warning(self, message):
-        print(message)
-
-    def error(self, message):
-        print(message)
-
+from dm_aioaiagent import DMAIAgent
+from dm_logger import FormatterConfig
+
+
+# set up custom logger for all clients
+DMAIAgent.set_logger_params(
+   {
+      "name": "my_name",
+      "formatter_config": FormatterConfig(
+         show_datetime=False,
+      )
+   }
+)
+```
 
-# create an agent
-ai_agent = DMAioAIAgent()
+See more about DMLogger [here](https://github.com/MykhLibs/dm-logger)
 
-# set up custom logger for this agent
-ai_agent.set_logger(MyLogger())
-```

{dm_aioaiagent-0.4.7 → dm_aioaiagent-0.4.9}/README.md

@@ -11,6 +11,16 @@
 
 Analogue to `DMAioAIAgent` is the synchronous client `DMAIAgent`.
 
+### Windows Setup
+
+```python
+import asyncio
+import sys
+
+if sys.platform == "win32":
+    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+```
+
 ### Use agent *with* inner memory and run *single* message
 
 By default, agent use inner memory to store the conversation history.
@@ -135,30 +145,21 @@ if __name__ == "__main__":
 
 ### Set custom logger
 
-_If you want set up custom logger_
-
 ```python
-from dm_aioaiagent import DMAioAIAgent
-
-
-# create custom logger
-class MyLogger:
-    def debug(self, message):
-        pass
-
-    def info(self, message):
-        pass
-
-    def warning(self, message):
-        print(message)
-
-    def error(self, message):
-        print(message)
-
+from dm_aioaiagent import DMAIAgent
+from dm_logger import FormatterConfig
+
+
+# set up custom logger for all clients
+DMAIAgent.set_logger_params(
+   {
+      "name": "my_name",
+      "formatter_config": FormatterConfig(
+         show_datetime=False,
+      )
+   }
+)
+```
 
-# create an agent
-ai_agent = DMAioAIAgent()
+See more about DMLogger [here](https://github.com/MykhLibs/dm-logger)
 
-# set up custom logger for this agent
-ai_agent.set_logger(MyLogger())
-```

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

@@ -18,6 +18,7 @@ __all__ = ["DMAIAgent"]
 class DMAIAgent:
     MAX_MEMORY_MESSAGES = 20  # Only INT greater than 0
     _ALLOWED_ROLES = ("user", "ai")
+    _logger_params = None
 
     def __init__(
         self,
@@ -25,8 +26,8 @@ class DMAIAgent:
         tools: list[BaseTool] = None,
         *,
         model: str = "gpt-4o-mini",
-        temperature: int = 1,
-        parallel_tool_calls: bool = True,
+        temperature: float = None,
+        parallel_tool_calls: bool = None,
         agent_name: str = "AIAgent",
         input_output_logging: bool = True,
         is_memory_enabled: bool = True,
@@ -37,15 +38,15 @@ class DMAIAgent:
         response_if_request_fail: str = "I can't provide a response right now. Please try again later.",
         response_if_invalid_image: str = "The image is unavailable or the link is incorrect."
     ):
-        self._logger = DMLogger(agent_name)
+        self._set_logger(agent_name)
         self._input_output_logging = bool(input_output_logging)
 
         self._system_message = str(system_message)
         self._tools = tools or []
         self._is_tools_exists = bool(tools)
         self._model = str(model)
-        self._temperature = int(temperature)
-        self._parallel_tool_calls = bool(parallel_tool_calls)
+        self._temperature = temperature
+        self._parallel_tool_calls = parallel_tool_calls
         self._llm_provider_api_key = str(llm_provider_api_key)
         self._llm_provider_base_url = str(llm_provider_base_url)
 
@@ -193,27 +194,28 @@ class DMAIAgent:
         return route
 
     def _init_agent(self) -> None:
-        base_kwargs = {
-            "model": self._model,
-            "temperature": self._temperature,
-            "base_url": self._llm_provider_base_url if self._llm_provider_base_url else None
-        }
+        base_kwargs = {"model": self._model}
+        if isinstance(self._temperature, float):
+            base_kwargs["temperature"] = self._temperature
         if self._llm_provider_api_key:
             base_kwargs["api_key"] = SecretStr(self._llm_provider_api_key)
+        if self._llm_provider_base_url:
+            base_kwargs["base_url"] = self._llm_provider_base_url
 
         if self._model.startswith("claude"):
             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
-            }}
+            bind_tool_kwargs = {"tool_choice": {"type": "auto"}}
+            if isinstance(self._parallel_tool_calls, bool):
+                bind_tool_kwargs["tool_choice"]["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}
+            bind_tool_kwargs = {}
+            if isinstance(self._parallel_tool_calls, bool):
+                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}
@@ -264,13 +266,13 @@ class DMAIAgent:
         except Exception as e:
             self._logger.error(e)
 
-    def set_logger(self, logger) -> None:
-        if (
-            hasattr(logger, "debug") and callable(logger.debug) and
-            hasattr(logger, "info") and callable(logger.info) and
-            hasattr(logger, "warning") and callable(logger.warning) and
-            hasattr(logger, "error") and callable(logger.error)
-        ):
-            self._logger = logger
-        else:
-            print("Invalid logger")
+    def _set_logger(self, agent_name: str) -> None:
+        params = {"name": agent_name}
+        if isinstance(self._logger_params, dict):
+            params.update(self._logger_params)
+        self._logger = DMLogger(**params)
+
+    @classmethod
+    def set_logger_params(cls, extra_params = None) -> None:
+        if isinstance(extra_params, dict) or extra_params is None:
+            cls._logger_params = extra_params

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

@@ -10,6 +10,17 @@ __all__ = ["DMAioAIAgent"]
 
 
 class DMAioAIAgent(DMAIAgent):
+    _logger_params = None
+
+    def __init__(self, *args, agent_name: str = "AioAIAgent", **kwargs):
+        super().__init__(*args, **kwargs)
+
+        if not isinstance(self._logger_params, dict):
+            self._logger_params = {}
+        if "name" not in self._logger_params:
+            self._logger_params["name"] = agent_name
+        self._set_logger(agent_name)
+
     async def run(self, query: str, *args, **kwargs) -> str:
         new_messages = await self.run_messages(messages=[{"role": "user", "content": query}], *args, **kwargs)
         return new_messages[-1].content

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

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: dm-aioaiagent
-Version: 0.4.7
+Version: 0.4.9
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
@@ -12,7 +12,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: dm-logger~=0.5.2
+Requires-Dist: dm-logger~=0.6.2
 Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: pydantic<3.0.0,>=2.9.2
 Requires-Dist: langchain~=0.3.0
@@ -48,6 +48,16 @@ Dynamic: summary
 
 Analogue to `DMAioAIAgent` is the synchronous client `DMAIAgent`.
 
+### Windows Setup
+
+```python
+import asyncio
+import sys
+
+if sys.platform == "win32":
+    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+```
+
 ### Use agent *with* inner memory and run *single* message
 
 By default, agent use inner memory to store the conversation history.
@@ -172,30 +182,21 @@ if __name__ == "__main__":
 
 ### Set custom logger
 
-_If you want set up custom logger_
-
 ```python
-from dm_aioaiagent import DMAioAIAgent
-
-
-# create custom logger
-class MyLogger:
-    def debug(self, message):
-        pass
-
-    def info(self, message):
-        pass
-
-    def warning(self, message):
-        print(message)
-
-    def error(self, message):
-        print(message)
-
+from dm_aioaiagent import DMAIAgent
+from dm_logger import FormatterConfig
+
+
+# set up custom logger for all clients
+DMAIAgent.set_logger_params(
+   {
+      "name": "my_name",
+      "formatter_config": FormatterConfig(
+         show_datetime=False,
+      )
+   }
+)
+```
 
-# create an agent
-ai_agent = DMAioAIAgent()
+See more about DMLogger [here](https://github.com/MykhLibs/dm-logger)
 
-# set up custom logger for this agent
-ai_agent.set_logger(MyLogger())
-```

{dm_aioaiagent-0.4.7 → dm_aioaiagent-0.4.9}/dm_aioaiagent.egg-info/requires.txt

@@ -1,4 +1,4 @@
-dm-logger~=0.5.2
+dm-logger~=0.6.2
 python-dotenv>=1.0.0
 pydantic<3.0.0,>=2.9.2
 langchain~=0.3.0

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

@@ -8,7 +8,7 @@ def readme():
 
 setup(
     name='dm-aioaiagent',
-    version='v0.4.7',
+    version='v0.4.9',
     author='dimka4621',
     author_email='mismartconfig@gmail.com',
     description='This is my custom aioaiagent client',
@@ -17,7 +17,7 @@ setup(
     url='https://pypi.org/project/dm-aioaiagent',
     packages=find_packages(),
     install_requires=[
-        'dm-logger~=0.5.2',
+        'dm-logger~=0.6.2',
         'python-dotenv>=1.0.0',
         'pydantic>=2.9.2, < 3.0.0',
         'langchain~=0.3.0',