Skip to main content

Basic Chat

import asyncio
from upsonic import Agent, Chat


async def main():
    agent = Agent("openai/gpt-4o")

    chat = Chat(
        session_id="example_session",
        user_id="example_user",
        agent=agent
    )

    response1 = await chat.invoke("Hello, my name is Alice")
    print(f"Assistant: {response1}")

    response2 = await chat.invoke("What is my name?")
    print(f"Assistant: {response2}")

    print(f"\nMessages: {len(chat.all_messages)}")
    print(f"Cost: ${chat.total_cost:.4f}")


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

With Task Objects

import asyncio
from upsonic import Agent, Task, Chat


async def main():
    agent = Agent("openai/gpt-4o")
    chat = Chat(session_id="session1", user_id="user1", agent=agent)

    task = Task(description="Explain quantum computing in simple terms")
    response = await chat.invoke(task)
    print(response)


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

Streaming

import asyncio
from upsonic import Agent, Chat


async def main():
    agent = Agent("openai/gpt-4o")
    chat = Chat(session_id="session1", user_id="user1", agent=agent)

    print("Assistant: ", end="")
    async for chunk in chat.stream("Tell me a short joke"):
        print(chunk, end="", flush=True)
    print()


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