Skip to content

Commit b722b09

Browse files
authored
Merge pull request #70 from jd-opensource/feature-fix-sse-client-timout
feat: Added the `timeout=self.timeout` parameter to the `sse_client` …
2 parents 2cd0cd6 + 879a023 commit b722b09

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

oxygent/oxy/mcp_tools/sse_mcp_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ async def init(self, is_fetch_tools=True) -> None:
7676
raise Exception(f"Server {self.name} error")
7777

7878
async def call_tool(self, tool_name, arguments, headers=None):
79-
async with sse_client(build_url(self.sse_url), headers=headers) as streams:
79+
async with sse_client(build_url(self.sse_url), headers=headers, timeout=self.timeout) as streams:
8080
async with ClientSession(*streams) as session:
8181
await session.initialize()
8282
return await session.call_tool(tool_name, arguments)

0 commit comments

Comments
 (0)