mirror of
https://git.mirrors.martin98.com/https://github.com/bytedance/deer-flow
synced 2025-08-18 06:55:56 +08:00
fix: unify naming
This commit is contained in:
parent
867a158c6b
commit
daf2e57f14
5
main.py
5
main.py
@ -2,15 +2,16 @@
|
|||||||
# SPDX-License-Identifier: MIT
|
# SPDX-License-Identifier: MIT
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Entry point script for the Deer project.
|
Entry point script for the DeerFlow project.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
import asyncio
|
import asyncio
|
||||||
|
|
||||||
from InquirerPy import inquirer
|
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.config.questions import BUILT_IN_QUESTIONS, BUILT_IN_QUESTIONS_ZH_CN
|
||||||
|
from src.workflow import run_agent_workflow_async
|
||||||
|
|
||||||
|
|
||||||
def ask(
|
def ask(
|
||||||
|
@ -2,7 +2,7 @@
|
|||||||
# SPDX-License-Identifier: MIT
|
# SPDX-License-Identifier: MIT
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Server script for running the Deer API.
|
Server script for running the DeerFlow API.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
@ -20,7 +20,7 @@ logger = logging.getLogger(__name__)
|
|||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
# Parse command line arguments
|
# 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(
|
parser.add_argument(
|
||||||
"--reload",
|
"--reload",
|
||||||
action="store_true",
|
action="store_true",
|
||||||
@ -55,7 +55,7 @@ if __name__ == "__main__":
|
|||||||
if args.reload:
|
if args.reload:
|
||||||
reload = True
|
reload = True
|
||||||
|
|
||||||
logger.info("Starting Deer API server")
|
logger.info("Starting DeerFlow API server")
|
||||||
uvicorn.run(
|
uvicorn.run(
|
||||||
"src.server:app",
|
"src.server:app",
|
||||||
host=args.host,
|
host=args.host,
|
||||||
|
@ -33,7 +33,7 @@ from src.tools import VolcengineTTS
|
|||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
app = FastAPI(
|
app = FastAPI(
|
||||||
title="Deer API",
|
title="DeerFlow API",
|
||||||
description="API for Deer",
|
description="API for Deer",
|
||||||
version="0.1.0",
|
version="0.1.0",
|
||||||
)
|
)
|
||||||
@ -285,11 +285,11 @@ async def mcp_server_metadata(request: MCPServerMetadataRequest):
|
|||||||
try:
|
try:
|
||||||
# Set default timeout with a longer value for this endpoint
|
# Set default timeout with a longer value for this endpoint
|
||||||
timeout = 300 # Default to 300 seconds for this endpoint
|
timeout = 300 # Default to 300 seconds for this endpoint
|
||||||
|
|
||||||
# Use custom timeout from request if provided
|
# Use custom timeout from request if provided
|
||||||
if request.timeout_seconds is not None:
|
if request.timeout_seconds is not None:
|
||||||
timeout = request.timeout_seconds
|
timeout = request.timeout_seconds
|
||||||
|
|
||||||
# Load tools from the MCP server using the utility function
|
# Load tools from the MCP server using the utility function
|
||||||
tools = await load_mcp_tools(
|
tools = await load_mcp_tools(
|
||||||
server_type=request.transport,
|
server_type=request.transport,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user