From ecb70cb6f77e7a8f9058e185d7d02ff20375ea81 Mon Sep 17 00:00:00 2001
From: Soulter <905617992@qq.com>
Date: Tue, 27 May 2025 15:05:42 +0800
Subject: [PATCH] feat: add support for custom headers in SSE client
 configuration

fixes: #1659
---
 astrbot/core/provider/func_tool_manager.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/astrbot/core/provider/func_tool_manager.py b/astrbot/core/provider/func_tool_manager.py
index 7059a00f7..416e74386 100644
--- a/astrbot/core/provider/func_tool_manager.py
+++ b/astrbot/core/provider/func_tool_manager.py
@@ -109,7 +109,12 @@ class MCPClient:
 
         if "url" in cfg:
             # SSE transport method
-            self._streams_context = sse_client(url=cfg["url"])
+            self._streams_context = sse_client(
+                url=cfg["url"],
+                headers=cfg.get("headers", {}),
+                timeout=cfg.get("timeout", 5),
+                sse_read_timeout=cfg.get("sse_read_timeout", 60 * 5),
+            )
             streams = await self._streams_context.__aenter__()
 
             # Create a new client session