Roots 둘러보기

1import asyncio
2import sys
3import os
4from dotenv import load_dotenv
5from contextlib import AsyncExitStack
6
7from mcp_client import MCPClient
8from core.claude import Claude
9
10from core.cli_chat import CliChat
11from core.cli import CliApp
12
13load_dotenv()
14
15# Anthropic Config
16claude_model = os.getenv("CLAUDE_MODEL", "claude-sonnet-4-5")
17anthropic_api_key = os.getenv("ANTHROPIC_API_KEY", "")
18
19
20assert claude_model, "Error: CLAUDE_MODEL cannot be empty. Update .env"
21assert anthropic_api_key, (
22    "Error: ANTHROPIC_API_KEY cannot be empty. Update .env"
23)
24
25
26async def main():
27    claude_service = Claude(model=claude_model)
28
29    # Get root directories from command line arguments
30    root_paths = sys.argv[1:]
31    if not root_paths:
32        print("Usage: uv run main.py <root1> [root2] ...")
33        print("Example: uv run main.py /path/to/videos /another/path")
34        sys.exit(1)
35
36    clients = {}
37
38    async with AsyncExitStack() as stack:
39        # Create the MCP client with the provided root directories
40        doc_client = await stack.enter_async_context(
41            MCPClient(
42                command="uv", args=["run", "mcp_server.py"], roots=root_paths
43            )
44        )
45        clients["doc_client"] = doc_client
46
47        chat = CliChat(
48            doc_client=doc_client,
49            clients=clients,
50            claude_service=claude_service,
51        )
52
53        cli = CliApp(chat)
54        await cli.initialize()
55        await cli.run()
56
57
58if __name__ == "__main__":
59    if sys.platform == "win32":
60        asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
61    asyncio.run(main())
62