dm-aioaiagent 0.5.5__tar.gz → 0.5.7__tar.gz

{dm_aioaiagent-0.5.5 → dm_aioaiagent-0.5.7}/PKG-INFO

@@ -1,19 +1,19 @@
 Metadata-Version: 2.4
 Name: dm-aioaiagent
-Version: 0.5.5
+Version: 0.5.7
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
 Author-email: mismartconfig@gmail.com
 Project-URL: GitHub, https://github.com/MykhLibs/dm-aioaiagent
 Keywords: dm aioaiagent
-Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
 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.7.0,>=0.6.6
-Requires-Dist: python-dotenv<1.1.0,>=1.0.0
+Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: pydantic<3.0.0,>=2.9.2
 Requires-Dist: langchain<0.4.0,>=0.3.0
 Requires-Dist: langchain-core<0.4.0,>=0.3.5

{dm_aioaiagent-0.5.5 → dm_aioaiagent-0.5.7}/dm_aioaiagent/ai_agent.py

@@ -36,6 +36,7 @@ class DMAIAgent:
         max_memory_messages: int = MAX_MEMORY_MESSAGES,
         # other
         input_output_logging: bool = True,
+        node_execution_logging: bool = True,
         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.",
         # llm provider
@@ -66,6 +67,7 @@ class DMAIAgent:
         self._max_memory_messages = self._validate_max_memory_messages(max_memory_messages)
         # other
         self._input_output_logging = bool(input_output_logging)
+        self._node_execution_logging = bool(node_execution_logging)
         self._response_if_request_fail = str(response_if_request_fail)
         self._response_if_invalid_image = str(response_if_invalid_image)
         # llm provider
@@ -142,7 +144,8 @@ class DMAIAgent:
         return state
 
     def _invoke_llm_node(self, state: State, second_attempt: bool = False) -> State:
-        self._logger.debug("Run node: Invoke LLM")
+        if self._node_execution_logging:
+            self._logger.debug("Run node: Invoke LLM")
         try:
             ai_response = self._agent.invoke({"messages": state["messages"]})
         except Exception as e:
@@ -165,7 +168,8 @@ class DMAIAgent:
         return state
 
     def _execute_tool_node(self, state: State) -> State:
-        self._logger.debug("Run node: Execute tool")
+        if self._node_execution_logging:
+            self._logger.debug("Run node: Execute tool")
         threads = []
 
         for tool_call in state["messages"][-1].tool_calls:

{dm_aioaiagent-0.5.5 → dm_aioaiagent-0.5.7}/dm_aioaiagent/async_ai_agent.py

@@ -58,7 +58,8 @@ class DMAioAIAgent(DMAIAgent):
         return state["new_messages"]
 
     async def _invoke_llm_node(self, state: State, second_attempt: bool = False) -> State:
-        self._logger.debug("Run node: Invoke LLM")
+        if self._node_execution_logging:
+            self._logger.debug("Run node: Invoke LLM")
         try:
             ai_response = await self._agent.ainvoke({"messages": state["messages"]})
         except Exception as e:
@@ -81,7 +82,8 @@ class DMAioAIAgent(DMAIAgent):
         return state
 
     async def _execute_tool_node(self, state: State) -> State:
-        self._logger.debug("Run node: Execute tool")
+        if self._node_execution_logging:
+            self._logger.debug("Run node: Execute tool")
         tasks = []
 
         for tool_call in state["messages"][-1].tool_calls:

{dm_aioaiagent-0.5.5 → dm_aioaiagent-0.5.7}/dm_aioaiagent.egg-info/PKG-INFO

@@ -1,19 +1,19 @@
 Metadata-Version: 2.4
 Name: dm-aioaiagent
-Version: 0.5.5
+Version: 0.5.7
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
 Author-email: mismartconfig@gmail.com
 Project-URL: GitHub, https://github.com/MykhLibs/dm-aioaiagent
 Keywords: dm aioaiagent
-Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
 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.7.0,>=0.6.6
-Requires-Dist: python-dotenv<1.1.0,>=1.0.0
+Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: pydantic<3.0.0,>=2.9.2
 Requires-Dist: langchain<0.4.0,>=0.3.0
 Requires-Dist: langchain-core<0.4.0,>=0.3.5

{dm_aioaiagent-0.5.5 → dm_aioaiagent-0.5.7}/dm_aioaiagent.egg-info/requires.txt

@@ -1,5 +1,5 @@
 dm-logger<0.7.0,>=0.6.6
-python-dotenv<1.1.0,>=1.0.0
+python-dotenv>=1.0.0
 pydantic<3.0.0,>=2.9.2
 langchain<0.4.0,>=0.3.0
 langchain-core<0.4.0,>=0.3.5

{dm_aioaiagent-0.5.5 → dm_aioaiagent-0.5.7}/setup.py

@@ -8,7 +8,7 @@ def readme():
 
 setup(
     name='dm-aioaiagent',
-    version='v0.5.5',
+    version='v0.5.7',
     author='dimka4621',
     author_email='mismartconfig@gmail.com',
     description='This is my custom aioaiagent client',
@@ -18,7 +18,7 @@ setup(
     packages=find_packages(),
     install_requires=[
         'dm-logger>=0.6.6, <0.7.0',
-        'python-dotenv>=1.0.0, <1.1.0',
+        'python-dotenv>=1.0.0',
         'pydantic>=2.9.2, <3.0.0',
         'langchain>=0.3.0, <0.4.0',
         'langchain-core>=0.3.5, <0.4.0',
@@ -30,7 +30,7 @@ setup(
         'grandalf>=0.8.0, <0.9.0',
     ],
     classifiers=[
-        'Programming Language :: Python :: 3.8',
+        'Programming Language :: Python :: 3.9',
         'License :: OSI Approved :: MIT License',
         'Operating System :: OS Independent'
     ],