Skip to content

Conversation

tblinka1996
Copy link
Owner

功能概述

添加网络搜索功能支持。用户可以通过在模型名后添加 -search 后缀来启用 You.com 的网络搜索功能。

主要变更

  • 🔍 支持搜索模型(如 gpt-4o-searchclaude-3.5-sonnet-search 等)
  • 📋 模型列表API自动包含所有搜索模型版本(总数从33个扩展到66个)
  • 🔧 智能检测搜索模型并启用相应的You.com搜索参数
  • 🔄 保持完整的OpenAI API兼容性
  • 📚 完整的使用文档和示例

技术实现

  • 新增 isSearchModel()getBaseModelName() 函数自动检测搜索模型
  • 修改模型映射逻辑,搜索模型映射到对应的基础模型
  • 在You.com API请求中添加搜索参数:use_search=true, search_focus=web, enable_web_results=true
  • 响应中正确返回原始搜索模型名称

使用示例

# 获取包含搜索模型的完整模型列表
curl http://localhost:8080/v1/models

# 普通聊天
curl -X POST http://localhost:8080/v1/chat/completions \
  -H "Authorization: Bearer YOUR_DS_TOKEN" \
  -d '{"model": "gpt-4o", "messages": [{"role": "user", "content": "什么是AI?"}]}'

# 启用搜索功能
curl -X POST http://localhost:8080/v1/chat/completions \
  -H "Authorization: Bearer YOUR_DS_TOKEN" \
  -d '{"model": "gpt-4o-search", "messages": [{"role": "user", "content": "今天的最新科技新闻"}]}'

测试验证

  • ✅ 所有现有功能保持兼容
  • ✅ 搜索功能正常工作,能获取实时网络信息
  • ✅ 模型列表正确返回66个模型(33个基础+33个搜索)
  • ✅ 错误处理和边缘情况处理完善
  • ✅ 支持流式和非流式响应

文件变更

  • api/main.go - 核心搜索功能实现
  • SEARCH_FEATURE.md - 用户使用文档

用户现在可以直接从模型列表中选择搜索模型,无需手动添加后缀,获得增强的AI回答体验。

@tblinka1996 can click here to continue refining the PR

- Add support for search models (e.g., gpt-4o-search, claude-3.5-sonnet-search)
- Automatically detect -search suffix and enable You.com web search features
- Extend model list API to include all search model variants
- Maintain full OpenAI API compatibility
- Return original search model name in responses

Users can now select search models directly from the model list at /v1/models
and get AI responses enhanced with real-time web search results.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants