Chatbot session

import asyncio

from vectorshift import Chatbot
from vectorshift.events import SessionEventType


async def main() -> None:
    # Fetch an existing chatbot (must already exist)
    chatbot = Chatbot.fetch(name="My Chatbot")
    print(f"Fetched chatbot: {chatbot.name} (id={chatbot.id})")

    # Open a session - same API as agent.create_session()
    async with await chatbot.create_session() as session:
        print(f"Session connected: {session.session_id}")

        await session.send("Hello! What can you help me with?")

        print("\nAssistant: ", end="", flush=True)
        async for event in session.listen(
            event_types=[
                SessionEventType.MESSAGE_DELTA,
                SessionEventType.MESSAGE_COMPLETE,
            ]
        ):
            if event.delta:
                print(event.delta, end="", flush=True)
            if event.is_complete:
                break
        print()

    print("\nDone.")


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

Fetched chatbot: My Chatbot (id=...)
Session connected: ...

Assistant: Hi there! I can help you with...

Done.