iframe-proxy | Sunbelt Computer Software

53 lines (42 loc) · 1.35 KB

import asyncio
from copilot import CopilotClient
from copilot.session import PermissionHandler
from copilot.session_events import (
    AssistantMessageData,
    AssistantReasoningData,
    ToolExecutionStartData,
BLUE = "\033[34m"
RESET = "\033[0m"
async def main():
    client = CopilotClient()
    await client.start()
    session = await client.create_session(on_permission_request=PermissionHandler.approve_all)
    def on_event(event):
        output = None
        match event.data:
            case AssistantReasoningData() as data:
                output = f"[reasoning: {data.content}]"
            case ToolExecutionStartData() as data:
                output = f"[tool: {data.tool_name}]"
        if output:
            print(f"{BLUE}{output}{RESET}")
    session.on(on_event)
    print("Chat with Copilot (Ctrl+C to exit)\n")
    while True:
        user_input = input("You: ").strip()
        if not user_input:
            continue
        print()
        reply = await session.send_and_wait(user_input)
        assistant_output = None
        if reply:
            match reply.data:
                case AssistantMessageData() as data:
                    assistant_output = data.content
        print(f"\nAssistant: {assistant_output}\n")
if __name__ == "__main__":
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\nBye!")

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

chat.py

Latest commit

History

chat.py

File metadata and controls