dm-aioaiagent 0.1.0

This is my custom aioaiagent client

DM-aioaiagent

Urls

• PyPI
• GitHub

* Package contains both asynchronous and synchronous clients

Usage

Example of using DMAioAIAgent without tools

Analogue to DMAioAIAgent is the synchronous client DMAIAgent.

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 = [...]

    # define a openai model, default is "gpt-4o-mini"
    model_name = "gpt-4o"

    # create an agent
    ai_agent = DMAioAIAgent(system_message, tools, model=model_name)

    # 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"},
    ]

    # start the agent
    state = await ai_agent.run(messages)

    # print answer
    print(state["answer"])

    # if you define tools, you can see the context of the tools
    print(state["answer"])


if __name__ == "__main__":
    asyncio.run(main())

Set custom logger

If you want set up custom logger

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)


# create agent
ai_agent = DMAioAIAgent()

# set up custom logger for this agent
ai_agent.set_logger(MyLogger())