mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-05-21 07:26:50 +00:00
feat: support dify in rag module (#550)
Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
This commit is contained in:
@@ -97,7 +97,8 @@ async def load_mcp_tools(
|
||||
)
|
||||
|
||||
return await _get_tools_from_client_session(
|
||||
sse_client(url=url, headers=headers, timeout=timeout_seconds), timeout_seconds
|
||||
sse_client(url=url, headers=headers, timeout=timeout_seconds),
|
||||
timeout_seconds,
|
||||
)
|
||||
|
||||
elif server_type == "streamable_http":
|
||||
@@ -107,7 +108,10 @@ async def load_mcp_tools(
|
||||
)
|
||||
|
||||
return await _get_tools_from_client_session(
|
||||
streamablehttp_client(url=url, headers=headers, timeout=timeout_seconds), timeout_seconds,
|
||||
streamablehttp_client(
|
||||
url=url, headers=headers, timeout=timeout_seconds
|
||||
),
|
||||
timeout_seconds,
|
||||
)
|
||||
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user