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

{dm_aioaiagent-0.4.8 → 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.8
+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.8 → 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.8 → 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,
@@ -37,7 +38,7 @@ 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)
@@ -265,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.8 → 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.8 → 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.8
+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.8 → 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.8 → dm_aioaiagent-0.4.9}/setup.py

@@ -8,7 +8,7 @@ def readme():
 
 setup(
     name='dm-aioaiagent',
-    version='v0.4.8',
+    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',