OpenAI ChatKit Integration for Microsoft Agent Framework

import os
import asyncio
from chatkit import ChatKitClient
from chatkit.agents import BasicChatKitAgent
from chatkit.messages import TextMessage

async def main():
    # Ensure CHATKIT_API_KEY and OPENAI_API_KEY are set as environment variables
    chatkit_api_key = os.environ.get("CHATKIT_API_KEY", "")
    openai_api_key = os.environ.get("OPENAI_API_KEY", "")

    if not chatkit_api_key or not openai_api_key:
        print("Please set CHATKIT_API_KEY and OPENAI_API_KEY environment variables.")
        print("You can obtain these from your respective dashboards (OpenAI and ChatKit). Exiting.")
        return

    client = ChatKitClient(api_key=chatkit_api_key, openai_api_key=openai_api_key)
    agent_name = "my-chatkit-quickstart-agent"

    # Create or update a simple agent definition
    agent = BasicChatKitAgent(
        name=agent_name,
        model="gpt-4o", # Recommended to use a recent model
        instructions="You are a helpful, concise assistant. Respond briefly to questions."
    )
    # This method handles creation if the agent doesn't exist, or updates it if it does.
    await client.create_agent(agent)
    print(f"Agent '{agent_name}' is ready.")

    user_message = TextMessage(content="What is the capital of France?")
    print(f"\nUser: {user_message.content}")

    # Interact with the agent
    async for response_message in client.run_agent_message(
        agent_name=agent_name,
        message=user_message
    ):
        if isinstance(response_message, TextMessage):
            print(f"Agent: {response_message.content}")
        else:
            # Handle other message types like FunctionCallMessage if needed
            print(f"Agent sent a non-text message: {type(response_message).__name__}")

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