Skip to content

Commit 22fe860

Browse files
committed
fix: Increase timeout for tool requests to improve reliability
1 parent f001ecd commit 22fe860

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/upsonicai/server/tools/server/function_tools.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ class ToolRequest(BaseModel):
8080

8181

8282
@app.post(f"{prefix}/tools")
83-
@timeout(10.0)
83+
@timeout(30.0)
8484
async def list_tools():
8585
print("Listing tools...")
8686

@@ -102,7 +102,7 @@ async def list_tools():
102102

103103

104104
@app.post(f"{prefix}/call_tool")
105-
@timeout(10.0)
105+
@timeout(30.0)
106106
async def call_tool(request: ToolRequest):
107107
print(f"Received tool call request: {request}")
108108

src/upsonicai/server/tools/server/mcp_tools.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ async def get_session(command: str, args: list, env: dict):
4747

4848

4949
@app.post(f"{prefix}/tools")
50-
@timeout(10.0)
50+
@timeout(30.0)
5151
async def list_tools(request: ListToolsRequest):
5252
print("Listing tools...")
5353

@@ -62,7 +62,7 @@ async def list_tools(request: ListToolsRequest):
6262

6363

6464
@app.post(f"{prefix}/call_tool")
65-
@timeout(10.0)
65+
@timeout(30.0)
6666
async def call_tool(request: ToolRequest):
6767
print(f"Received tool call request: {request}")
6868

0 commit comments

Comments
 (0)