✨ 增加自定义User-Agent支持,允许在请求API时设置User-Agent
Some checks failed
Pyright Lint / Pyright Lint (push) Has been cancelled
Ruff Lint / Ruff Lint (push) Has been cancelled

This commit is contained in:
FuQuan233 2026-08-24 11:23:53 +08:00
parent 3320758cf9
commit 0f33591f4d
3 changed files with 4 additions and 0 deletions

View file

@ -159,6 +159,7 @@ _✨ 支持多API预设、MCP协议、内置工具、联网搜索、视觉模型
| max_tokens | 否 | 2048 | 最大响应token数 |
| temperature | 否 | 0.7 | 生成温度 |
| proxy | 否 | 无 | 请求API时使用的HTTP代理 |
| user_agent | 否 | 无 | 请求API时使用的自定义User-Agent |
| support_mcp | 否 | False | 是否支持MCP协议 |
| support_image | 否 | False | 是否支持图片输入 |
| stream | 否 | False | 是否启用流式响应;普通行完成后立即发送,三反引号内容框闭合后整体发送 |

View file

@ -11,6 +11,7 @@ class PresetConfig(BaseModel):
max_tokens: int = Field(default=2048, description="最大响应token数")
temperature: float = Field(default=0.7, description="生成温度(0-2]")
proxy: str = Field(default="", description="HTTP代理服务器")
user_agent: str = Field(default="", description="请求API时使用的自定义User-Agent")
support_mcp: bool = Field(default=False, description="是否支持MCP")
support_image: bool = Field(default=False, description="是否支持图片输入")
stream: bool = Field(default=False, description="是否使用流式响应")

View file

@ -67,6 +67,8 @@ class ConversationService:
"api_key": preset.api_key,
"timeout": self.config.request_timeout,
}
if preset.user_agent:
options["default_headers"] = {"User-Agent": preset.user_agent}
if preset.proxy:
options["http_client"] = httpx.AsyncClient(proxy=preset.proxy)
return AsyncOpenAI(**options)