dm-aioaiagent 0.3.0__tar.gz → 0.3.2__tar.gz

{dm_aioaiagent-0.3.0 → dm_aioaiagent-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dm-aioaiagent
-Version: 0.3.0
+Version: 0.3.2
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
@@ -12,14 +12,14 @@ 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: python-dotenv==1.0.1
-Requires-Dist: pydantic==2.9.2
-Requires-Dist: langchain==0.3.0
-Requires-Dist: langchain-core==0.3.5
-Requires-Dist: langgraph==0.2.23
-Requires-Dist: langchain-community==0.3.0
-Requires-Dist: langchain-openai==0.2.0
+Requires-Dist: dm-logger~=0.5.2
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: pydantic<3.0.0,>=2.9.2
+Requires-Dist: langchain~=0.3.0
+Requires-Dist: langchain-core~=0.3.5
+Requires-Dist: langgraph~=0.2.23
+Requires-Dist: langchain-community~=0.3.0
+Requires-Dist: langchain-openai~=0.2.0
 
 # DM-aioaiagent
 

{dm_aioaiagent-0.3.0 → dm_aioaiagent-0.3.2}/dm_aioaiagent/ai_agent.py

@@ -16,7 +16,7 @@ __all__ = ["DMAIAgent"]
 class DMAIAgent:
     agent_name = "AIAgent"
     _allowed_roles = ("user", "ai")
-    MAX_MEMORY_MESSAGES = 20 # Only INT greater than 0
+    MAX_MEMORY_MESSAGES = 20  # Only INT greater than 0
 
     def __init__(
         self,
@@ -67,8 +67,19 @@ class DMAIAgent:
         state = self._graph.invoke({"input_messages": input_messages, "memory_id": memory_id})
         return state["response"]
 
-    def get_memory_messages(self, memory_id: str = None) -> list[BaseMessage]:
-        return self._memory.get(self._validate_memory_id(memory_id), [])
+    def get_memory_messages(
+        self,
+        memory_id: str = None,
+        *,
+        without_tool_m: bool = False,
+        return_str_m: bool = False
+    ) -> Union[list[BaseMessage], list[str]]:
+        messages = self._memory.get(self._validate_memory_id(memory_id), [])
+        if without_tool_m:
+            messages = [m for m in messages if not (m.type == "tool" or (m.type == "ai" and m.tool_calls))]
+        if return_str_m:
+            messages = [m.content for m in messages]
+        return messages
 
     def clear_memory(self, memory_id: str = None) -> None:
         self._memory[self._validate_memory_id(memory_id)] = []

{dm_aioaiagent-0.3.0 → dm_aioaiagent-0.3.2}/dm_aioaiagent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dm-aioaiagent
-Version: 0.3.0
+Version: 0.3.2
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
@@ -12,14 +12,14 @@ 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: python-dotenv==1.0.1
-Requires-Dist: pydantic==2.9.2
-Requires-Dist: langchain==0.3.0
-Requires-Dist: langchain-core==0.3.5
-Requires-Dist: langgraph==0.2.23
-Requires-Dist: langchain-community==0.3.0
-Requires-Dist: langchain-openai==0.2.0
+Requires-Dist: dm-logger~=0.5.2
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: pydantic<3.0.0,>=2.9.2
+Requires-Dist: langchain~=0.3.0
+Requires-Dist: langchain-core~=0.3.5
+Requires-Dist: langgraph~=0.2.23
+Requires-Dist: langchain-community~=0.3.0
+Requires-Dist: langchain-openai~=0.2.0
 
 # DM-aioaiagent
 

dm_aioaiagent-0.3.2/dm_aioaiagent.egg-info/requires.txt

@@ -0,0 +1,8 @@
+dm-logger~=0.5.2
+python-dotenv>=1.0.0
+pydantic<3.0.0,>=2.9.2
+langchain~=0.3.0
+langchain-core~=0.3.5
+langgraph~=0.2.23
+langchain-community~=0.3.0
+langchain-openai~=0.2.0

{dm_aioaiagent-0.3.0 → dm_aioaiagent-0.3.2}/setup.py

@@ -8,7 +8,7 @@ def readme():
 
 setup(
     name='dm-aioaiagent',
-    version='v0.3.0',
+    version='v0.3.2',
     author='dimka4621',
     author_email='mismartconfig@gmail.com',
     description='This is my custom aioaiagent client',
@@ -17,14 +17,14 @@ setup(
     url='https://pypi.org/project/dm-aioaiagent',
     packages=find_packages(),
     install_requires=[
-        'dm-logger==0.5.2',
-        'python-dotenv==1.0.1',
-        'pydantic==2.9.2',
-        'langchain==0.3.0',
-        'langchain-core==0.3.5',
-        'langgraph==0.2.23',
-        'langchain-community==0.3.0',
-        'langchain-openai==0.2.0',
+        'dm-logger~=0.5.2',
+        'python-dotenv>=1.0.0',
+        'pydantic>=2.9.2, < 3.0.0',
+        'langchain~=0.3.0',
+        'langchain-core~=0.3.5',
+        'langgraph~=0.2.23',
+        'langchain-community~=0.3.0',
+        'langchain-openai~=0.2.0',
     ],
     classifiers=[
         'Programming Language :: Python :: 3.8',

dm_aioaiagent-0.3.0/dm_aioaiagent.egg-info/requires.txt

@@ -1,8 +0,0 @@
-dm-logger==0.5.2
-python-dotenv==1.0.1
-pydantic==2.9.2
-langchain==0.3.0
-langchain-core==0.3.5
-langgraph==0.2.23
-langchain-community==0.3.0
-langchain-openai==0.2.0