This commit is contained in:
gantnocap 2025-03-18 00:42:04 +08:00
parent 0072174023
commit f6b2250e95

View File

@ -6,9 +6,11 @@ from contextlib import AsyncExitStack
from typing import Optional from typing import Optional
from colorama import Fore, init from colorama import Fore, init
from openai import AsyncOpenAI
from mcp import ClientSession, StdioServerParameters from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client from mcp.client.stdio import stdio_client
from openai import AsyncOpenAI
# Add current directory to Python path # Add current directory to Python path
current_dir = os.path.dirname(os.path.abspath(__file__)) current_dir = os.path.dirname(os.path.abspath(__file__))
@ -21,10 +23,12 @@ root_dir = os.path.dirname(parent_dir)
sys.path.insert(0, root_dir) sys.path.insert(0, root_dir)
from app.config import config from app.config import config
# Initialize colorama # Initialize colorama
def init_colorama(): def init_colorama():
init(autoreset=True) init(autoreset=True)
class OpenManusClient: class OpenManusClient:
def __init__(self): def __init__(self):
# Load configuration # Load configuration