diff --git a/main.py b/main.py index 38a7460..aee1500 100644 --- a/main.py +++ b/main.py @@ -2,15 +2,16 @@ # SPDX-License-Identifier: MIT """ -Entry point script for the Deer project. +Entry point script for the DeerFlow project. """ import argparse import asyncio + from InquirerPy import inquirer -from src.workflow import run_agent_workflow_async from src.config.questions import BUILT_IN_QUESTIONS, BUILT_IN_QUESTIONS_ZH_CN +from src.workflow import run_agent_workflow_async def ask( diff --git a/server.py b/server.py index 86cf6bc..97ebd51 100644 --- a/server.py +++ b/server.py @@ -2,7 +2,7 @@ # SPDX-License-Identifier: MIT """ -Server script for running the Deer API. +Server script for running the DeerFlow API. """ import argparse @@ -20,7 +20,7 @@ logger = logging.getLogger(__name__) if __name__ == "__main__": # Parse command line arguments - parser = argparse.ArgumentParser(description="Run the Deer API server") + parser = argparse.ArgumentParser(description="Run the DeerFlow API server") parser.add_argument( "--reload", action="store_true", @@ -55,7 +55,7 @@ if __name__ == "__main__": if args.reload: reload = True - logger.info("Starting Deer API server") + logger.info("Starting DeerFlow API server") uvicorn.run( "src.server:app", host=args.host, diff --git a/src/server/app.py b/src/server/app.py index f18c70b..9c97a0b 100644 --- a/src/server/app.py +++ b/src/server/app.py @@ -33,7 +33,7 @@ from src.tools import VolcengineTTS logger = logging.getLogger(__name__) app = FastAPI( - title="Deer API", + title="DeerFlow API", description="API for Deer", version="0.1.0", ) @@ -285,11 +285,11 @@ async def mcp_server_metadata(request: MCPServerMetadataRequest): try: # Set default timeout with a longer value for this endpoint timeout = 300 # Default to 300 seconds for this endpoint - + # Use custom timeout from request if provided if request.timeout_seconds is not None: timeout = request.timeout_seconds - + # Load tools from the MCP server using the utility function tools = await load_mcp_tools( server_type=request.transport,