dm-aioaiagent 0.1.3__tar.gz → 0.2.0__tar.gz

{dm_aioaiagent-0.1.3 → dm_aioaiagent-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dm-aioaiagent
-Version: 0.1.3
+Version: 0.2.0
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
@@ -43,7 +43,7 @@ async def main():
     # define a system message
     system_message = "Your custom system message with role, backstory and goal"
 
-    # define a list of tools, if you want to use them
+    # (optional) define a list of tools, if you want to use them
     tools = [...]
 
     # define a openai model, default is "gpt-4o-mini"
@@ -51,7 +51,39 @@ async def main():
 
     # create an agent
     ai_agent = DMAioAIAgent(system_message, tools, model=model_name)
-    # you can set input_output_logging=False, if you don't want to see the input and output messages from agent
+    # if you don't want to see the input and output messages from agent
+    # you can set input_output_logging=False
+
+    # define the conversation messages
+    messages = [
+       {"role": "user", "content": "Hello!"},
+       {"role": "ai", "content": "How can I help you?"},
+       {"role": "user", "content": "I want to know the weather in Kyiv"},
+    ]
+
+    # call an agent
+    answer = await ai_agent.run(messages)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+### Return context of the tools with answer
+```python
+import asyncio
+from dm_aioaiagent import DMAioAIAgent
+
+
+async def main():
+    # define a system message
+    system_message = "Your custom system message with role, backstory and goal"
+
+    # define a list of tools, if you want to use them
+    tools = [...]
+
+    # create an agent
+    ai_agent = DMAioAIAgent(system_message, tools, return_context=True)
 
     # define the conversation messages
     messages = [
@@ -65,7 +97,7 @@ async def main():
 
     # if you define tools, you can see the context of the tools
     answer = state["answer"]
-    print(state["context"])
+    context = state["context"]
 
 
 if __name__ == "__main__":
@@ -95,7 +127,7 @@ class MyLogger:
         print(message)
 
 
-# create agent
+# create an agent
 ai_agent = DMAioAIAgent()
 
 # set up custom logger for this agent

{dm_aioaiagent-0.1.3 → dm_aioaiagent-0.2.0}/README.md

@@ -20,7 +20,7 @@ async def main():
     # define a system message
     system_message = "Your custom system message with role, backstory and goal"
 
-    # define a list of tools, if you want to use them
+    # (optional) define a list of tools, if you want to use them
     tools = [...]
 
     # define a openai model, default is "gpt-4o-mini"
@@ -28,7 +28,39 @@ async def main():
 
     # create an agent
     ai_agent = DMAioAIAgent(system_message, tools, model=model_name)
-    # you can set input_output_logging=False, if you don't want to see the input and output messages from agent
+    # if you don't want to see the input and output messages from agent
+    # you can set input_output_logging=False
+
+    # define the conversation messages
+    messages = [
+       {"role": "user", "content": "Hello!"},
+       {"role": "ai", "content": "How can I help you?"},
+       {"role": "user", "content": "I want to know the weather in Kyiv"},
+    ]
+
+    # call an agent
+    answer = await ai_agent.run(messages)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+### Return context of the tools with answer
+```python
+import asyncio
+from dm_aioaiagent import DMAioAIAgent
+
+
+async def main():
+    # define a system message
+    system_message = "Your custom system message with role, backstory and goal"
+
+    # define a list of tools, if you want to use them
+    tools = [...]
+
+    # create an agent
+    ai_agent = DMAioAIAgent(system_message, tools, return_context=True)
 
     # define the conversation messages
     messages = [
@@ -42,7 +74,7 @@ async def main():
 
     # if you define tools, you can see the context of the tools
     answer = state["answer"]
-    print(state["context"])
+    context = state["context"]
 
 
 if __name__ == "__main__":
@@ -72,7 +104,7 @@ class MyLogger:
         print(message)
 
 
-# create agent
+# create an agent
 ai_agent = DMAioAIAgent()
 
 # set up custom logger for this agent

{dm_aioaiagent-0.1.3 → dm_aioaiagent-0.2.0}/dm_aioaiagent/ai_agent.py

@@ -1,6 +1,6 @@
 import json
 import os
-from typing import Optional, Literal
+from typing import Optional, Literal, Union
 from typing_extensions import TypedDict
 from pydantic import BaseModel, Field
 from threading import Thread
@@ -46,13 +46,15 @@ class DMAIAgent:
         model: str = "gpt-4o-mini",
         temperature: int = 1,
         agent_name: str = None,
-        input_output_logging: bool = True
+        input_output_logging: bool = True,
+        return_context: bool = False
     ):
         if not os.getenv("OPENAI_API_KEY"):
             raise EnvironmentError("OPENAI_API_KEY environment variable is not set!")
 
         self._logger = DMLogger(agent_name or self.agent_name)
         self._input_output_logging = input_output_logging
+        self._return_context = return_context
         self._is_tools_exists = bool(tools)
 
         prompt = ChatPromptTemplate.from_messages([SystemMessage(content=system_message),
@@ -78,8 +80,11 @@ class DMAIAgent:
         workflow.set_finish_point("Exit")
         self._graph = workflow.compile()
 
-    def run(self, messages: list[Message]) -> OutputState:
-        return self._graph.invoke({"messages": messages})
+    def run(self, messages: list[Message]) -> Union[str, OutputState]:
+        state = self._graph.invoke({"messages": messages})
+        if self._return_context:
+            return state
+        return state["answer"]
 
     def _prepare_messages_node(self, state: InputState) -> InputState:
         state.messages = state.messages or [{"role": "user", "content": "Привіт"}]

{dm_aioaiagent-0.1.3 → dm_aioaiagent-0.2.0}/dm_aioaiagent/async_ai_agent.py

@@ -1,21 +1,25 @@
 import json
 import sys
 import asyncio
+from typing import Union
 from langchain_core.messages import ToolMessage
 
-from .ai_agent import DMAIAgent, InputState, OutputState
-
-__all__ = ["DMAioAIAgent"]
+from .ai_agent import DMAIAgent, InputState, OutputState, Message
 
 if sys.platform == "win32":
     asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
 
+__all__ = ["DMAioAIAgent"]
+
 
 class DMAioAIAgent(DMAIAgent):
     agent_name = "AsyncAIAgent"
 
-    async def run(self, messages: list[dict[str, str]]) -> OutputState:
-        return await self._graph.ainvoke({"messages": messages})
+    async def run(self, messages: list[Message]) -> Union[str, OutputState]:
+        state = await self._graph.ainvoke({"messages": messages})
+        if self._return_context:
+            return state
+        return state["answer"]
 
     async def _invoke_llm_node(self, state: InputState) -> InputState:
         self._logger.debug("Run node: Invoke LLM")

{dm_aioaiagent-0.1.3 → dm_aioaiagent-0.2.0}/dm_aioaiagent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dm-aioaiagent
-Version: 0.1.3
+Version: 0.2.0
 Summary: This is my custom aioaiagent client
 Home-page: https://pypi.org/project/dm-aioaiagent
 Author: dimka4621
@@ -43,7 +43,7 @@ async def main():
     # define a system message
     system_message = "Your custom system message with role, backstory and goal"
 
-    # define a list of tools, if you want to use them
+    # (optional) define a list of tools, if you want to use them
     tools = [...]
 
     # define a openai model, default is "gpt-4o-mini"
@@ -51,7 +51,39 @@ async def main():
 
     # create an agent
     ai_agent = DMAioAIAgent(system_message, tools, model=model_name)
-    # you can set input_output_logging=False, if you don't want to see the input and output messages from agent
+    # if you don't want to see the input and output messages from agent
+    # you can set input_output_logging=False
+
+    # define the conversation messages
+    messages = [
+       {"role": "user", "content": "Hello!"},
+       {"role": "ai", "content": "How can I help you?"},
+       {"role": "user", "content": "I want to know the weather in Kyiv"},
+    ]
+
+    # call an agent
+    answer = await ai_agent.run(messages)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+### Return context of the tools with answer
+```python
+import asyncio
+from dm_aioaiagent import DMAioAIAgent
+
+
+async def main():
+    # define a system message
+    system_message = "Your custom system message with role, backstory and goal"
+
+    # define a list of tools, if you want to use them
+    tools = [...]
+
+    # create an agent
+    ai_agent = DMAioAIAgent(system_message, tools, return_context=True)
 
     # define the conversation messages
     messages = [
@@ -65,7 +97,7 @@ async def main():
 
     # if you define tools, you can see the context of the tools
     answer = state["answer"]
-    print(state["context"])
+    context = state["context"]
 
 
 if __name__ == "__main__":
@@ -95,7 +127,7 @@ class MyLogger:
         print(message)
 
 
-# create agent
+# create an agent
 ai_agent = DMAioAIAgent()
 
 # set up custom logger for this agent

{dm_aioaiagent-0.1.3 → dm_aioaiagent-0.2.0}/setup.py

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