dm-aioaiagent 0.4.1__tar.gz → 0.4.2__tar.gz

{dm_aioaiagent-0.4.1 → dm_aioaiagent-0.4.2}/PKG-INFO

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

{dm_aioaiagent-0.4.1 → dm_aioaiagent-0.4.2}/dm_aioaiagent/ai_agent.py

@@ -64,7 +64,7 @@ class DMAIAgent:
 
     def run_messages(
         self,
-        messages: InputMessagesType,
+        messages: list[InputMessage],
         *,
         ls_metadata: dict[str, Any] = None,
         ls_tags: list[str] = None,

{dm_aioaiagent-0.4.1 → dm_aioaiagent-0.4.2}/dm_aioaiagent/async_ai_agent.py

@@ -14,13 +14,13 @@ __all__ = ["DMAioAIAgent"]
 
 
 class DMAioAIAgent(DMAIAgent):
-    async def run(self, query: str, **kwargs) -> str:
-        new_messages = await self.run_messages(messages=[{"role": "user", "content": query}], **kwargs)
+    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
 
     async def run_messages(
         self,
-        messages: InputMessagesType,
+        messages: list[InputMessage],
         *,
         ls_metadata: dict[str, Any] = None,
         ls_tags: list[str] = None,

{dm_aioaiagent-0.4.1 → dm_aioaiagent-0.4.2}/dm_aioaiagent/types.py

@@ -3,7 +3,7 @@ from typing_extensions import TypedDict
 from langchain_core.messages import BaseMessage
 
 
-class ImageMessageTextMessage(TypedDict):
+class ImageMessageTextItem(TypedDict):
     type: Literal['text']
     text: str
 
@@ -15,7 +15,7 @@ class ImageMessageImageItem(TypedDict):
 
 class ImageMessage(TypedDict):
     role: Literal["user"]
-    content: list[Union[ImageMessageTextMessage, ImageMessageImageItem]]
+    content: list[Union[ImageMessageTextItem, ImageMessageImageItem]]
 
 
 class TextMessage(TypedDict):
@@ -23,8 +23,8 @@ class TextMessage(TypedDict):
     content: str
 
 
-InputMessagesType = list[Union[TextMessage, ImageMessage, BaseMessage]]
+InputMessage = Union[TextMessage, ImageMessage, BaseMessage]
 
 class State(TypedDict):
-    messages: InputMessagesType
+    messages: list[InputMessage]
     new_messages: list[BaseMessage]

{dm_aioaiagent-0.4.1 → dm_aioaiagent-0.4.2}/dm_aioaiagent.egg-info/PKG-INFO

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

{dm_aioaiagent-0.4.1 → dm_aioaiagent-0.4.2}/setup.py

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