307 lines
7.7 KiB
Markdown
307 lines
7.7 KiB
Markdown
# AI平台收集功能使用说明
|
||
|
||
## 概述
|
||
|
||
`internal/collect` 模块提供了访问多个AI平台并进行问答的功能，目前支持以下平台：
|
||
|
||
- **文心一言** (wenxin) - 百度AI助手
|
||
- **DeepSeek** (deepseek) - 深度求索AI
|
||
- **豆包** (doubao) - 字节跳动AI助手
|
||
- **通义千问** (qianwen) - 阿里云AI助手
|
||
|
||
## 架构设计
|
||
|
||
### 核心组件
|
||
|
||
1. **CollectorInterface** - 收集器接口
|
||
- `WaitLogin() (bool, string)` - 等待登录
|
||
- `AskQuestion(question string) (string, error)` - 提问并获取答案
|
||
|
||
2. **BaseCollector** - 基础收集器
|
||
- 浏览器驱动管理
|
||
- Cookie管理（保存/加载）
|
||
- 页面操作工具方法
|
||
|
||
3. **CollectManager** - 收集管理器
|
||
- 统一管理不同平台的收集器
|
||
- 提供便捷的API调用
|
||
|
||
4. **平台实现**
|
||
- `WenxinCollector` - 文心一言实现
|
||
- `DeepseekCollector` - DeepSeek实现
|
||
- `DoubaoCollector` - 豆包实现
|
||
- `QianwenCollector` - 通义千问实现
|
||
|
||
## 快速开始
|
||
|
||
### 1. 基本使用
|
||
|
||
```go
|
||
package main
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"geo/internal/collect"
|
||
"geo/internal/config"
|
||
"log"
|
||
"os"
|
||
)
|
||
|
||
func main() {
|
||
// 加载配置
|
||
cfg := &config.Config{
|
||
Sys: config.SysConfig{
|
||
ChromePath: "chrome/chrome.exe", // Chrome浏览器路径
|
||
ChromeDataDir: "chrome_data", // Chrome数据目录
|
||
CookiesDir: "cookies", // Cookie存储目录
|
||
LogsDir: "logs", // 日志目录
|
||
},
|
||
}
|
||
|
||
ctx := context.Background()
|
||
logger := log.New(os.Stdout, "", log.LstdFlags)
|
||
|
||
// 创建管理器
|
||
manager := collect.NewCollectManager(ctx, cfg, logger)
|
||
|
||
// 设置参数
|
||
params := &collect.CollectParams{
|
||
Headless: false, // 是否无头模式（false显示浏览器窗口）
|
||
UserIndex: "user_001", // 用户索引
|
||
PlatIndex: "wenxin", // 平台索引
|
||
RequestID: "req_001", // 请求ID
|
||
Platform: "wenxin", // 平台类型
|
||
}
|
||
|
||
// 向文心一言提问
|
||
question := "请介绍一下Go语言的特点"
|
||
answer, err := manager.AskQuestion("wenxin", params, question)
|
||
if err != nil {
|
||
fmt.Printf("错误: %v\n", err)
|
||
return
|
||
}
|
||
|
||
fmt.Printf("问题: %s\n", question)
|
||
fmt.Printf("答案: %s\n", answer)
|
||
}
|
||
```
|
||
|
||
### 2. 多平台对比
|
||
|
||
```go
|
||
// 向多个AI平台提问同一个问题
|
||
platforms := []string{"wenxin", "deepseek", "doubao", "qianwen"}
|
||
question := "什么是人工智能？"
|
||
|
||
for _, platform := range platforms {
|
||
params := &collect.CollectParams{
|
||
Headless: true,
|
||
UserIndex: "user_001",
|
||
PlatIndex: platform,
|
||
RequestID: fmt.Sprintf("req_%s", platform),
|
||
Platform: platform,
|
||
}
|
||
|
||
answer, err := manager.AskQuestion(platform, params, question)
|
||
if err != nil {
|
||
fmt.Printf("[%s] 错误: %v\n", platform, err)
|
||
continue
|
||
}
|
||
|
||
fmt.Printf("[%s] 答案: %s\n\n", platform, answer)
|
||
}
|
||
```
|
||
|
||
### 3. 登录管理
|
||
|
||
```go
|
||
// 首次使用时需要登录
|
||
params := &collect.CollectParams{
|
||
Headless: false, // 显示浏览器窗口以便扫码登录
|
||
UserIndex: "user_001",
|
||
PlatIndex: "wenxin",
|
||
RequestID: "login_req",
|
||
Platform: "wenxin",
|
||
}
|
||
|
||
// 等待登录（会打开浏览器窗口，需要手动扫码或输入账号密码）
|
||
success, msg := manager.WaitLogin("wenxin", params)
|
||
if success {
|
||
fmt.Println("登录成功！Cookie已保存")
|
||
} else {
|
||
fmt.Printf("登录失败: %s\n", msg)
|
||
}
|
||
|
||
// 后续使用会自动加载Cookie，无需重复登录
|
||
params.Headless = true // 可以切换到无头模式
|
||
answer, _ := manager.AskQuestion("wenxin", params, "你好")
|
||
```
|
||
|
||
### 4. 列出支持的平台
|
||
|
||
```go
|
||
platforms := manager.ListPlatforms()
|
||
fmt.Printf("支持的平台: %v\n", platforms)
|
||
// 输出: 支持的平台: [wenxin deepseek doubao qianwen]
|
||
```
|
||
|
||
## 配置说明
|
||
|
||
### 必需的配置项
|
||
|
||
```go
|
||
type SysConfig struct {
|
||
ChromePath string // Chrome浏览器可执行文件路径
|
||
ChromeDataDir string // Chrome用户数据目录
|
||
CookiesDir string // Cookie存储目录
|
||
LogsDir string // 日志文件目录
|
||
}
|
||
```
|
||
|
||
### 示例配置
|
||
|
||
```go
|
||
cfg := &config.Config{
|
||
Sys: config.SysConfig{
|
||
ChromePath: "/usr/bin/google-chrome", // Linux
|
||
// ChromePath: "C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe", // Windows
|
||
ChromeDataDir: "./chrome_data",
|
||
CookiesDir: "./cookies",
|
||
LogsDir: "./logs",
|
||
},
|
||
}
|
||
```
|
||
|
||
## 工作流程
|
||
|
||
1. **初始化浏览器** - 启动Chrome浏览器实例
|
||
2. **加载Cookie** - 从本地文件加载之前的登录状态
|
||
3. **检查登录** - 验证是否已登录
|
||
4. **导航到聊天页面** - 打开AI平台的对话界面
|
||
5. **输入问题** - 在输入框中输入问题
|
||
6. **点击发送** - 触发AI回答
|
||
7. **等待回答** - 等待AI生成完整答案
|
||
8. **提取答案** - 从页面中提取回答内容
|
||
9. **返回结果** - 将答案返回给调用者
|
||
|
||
## 注意事项
|
||
|
||
### 1. 首次使用需要登录
|
||
|
||
每个平台首次使用时需要手动登录：
|
||
- 设置 `Headless: false` 显示浏览器窗口
|
||
- 调用 `WaitLogin()` 方法
|
||
- 在浏览器中完成登录操作（扫码或输入账号密码）
|
||
- 登录成功后Cookie会自动保存
|
||
|
||
### 2. Cookie管理
|
||
|
||
- Cookie保存在 `cookies/{UserIndex}/{PlatIndex}.json`
|
||
- 下次使用会自动加载Cookie，无需重复登录
|
||
- 如果登录失效，重新调用 `WaitLogin()` 即可
|
||
|
||
### 3. 选择器适配
|
||
|
||
由于AI平台的页面结构可能会更新，如果遇到问题可能需要调整CSS选择器：
|
||
- 在对应的Collector文件中修改 `inputSelectors`、`sendSelectors`、`answerSelectors`
|
||
- 可以通过浏览器的开发者工具查看最新的元素选择器
|
||
|
||
### 4. 超时设置
|
||
|
||
- 登录超时: 180-300秒
|
||
- 回答超时: 120秒
|
||
- 可根据实际情况在代码中调整
|
||
|
||
### 5. 无头模式
|
||
|
||
- 开发调试时建议设置 `Headless: false`
|
||
- 生产环境可以设置 `Headless: true` 节省资源
|
||
|
||
## 扩展新平台
|
||
|
||
如果要添加新的AI平台，需要：
|
||
|
||
1. 创建新的Collector文件，如 `newplatform.go`
|
||
2. 实现 `CollectorInterface` 接口
|
||
3. 继承 `BaseCollector` 基础结构
|
||
4. 在 `interface.go` 的 `CollectorMap` 中注册
|
||
|
||
示例：
|
||
|
||
```go
|
||
package collect
|
||
|
||
import (
|
||
"context"
|
||
"geo/internal/config"
|
||
"log"
|
||
)
|
||
|
||
type NewPlatformCollector struct {
|
||
*BaseCollector
|
||
}
|
||
|
||
func NewNewPlatformCollector(ctx context.Context, params *CollectParams, cfg *config.Config, logger *log.Logger) CollectorInterface {
|
||
collector := &NewPlatformCollector{
|
||
BaseCollector: NewBaseCollector(ctx, params, cfg, logger),
|
||
}
|
||
collector.LoginURL = "https://example.com/login"
|
||
collector.ChatURL = "https://example.com/chat"
|
||
return collector
|
||
}
|
||
|
||
// 实现 CheckLoginStatus、WaitLogin、AskQuestion 等方法
|
||
// ...
|
||
```
|
||
|
||
然后在 `interface.go` 中注册：
|
||
|
||
```go
|
||
var CollectorMap = map[string]*CollectorValue{
|
||
// ... 其他平台
|
||
"newplatform": {
|
||
Name: "新平台",
|
||
InitMethod: NewNewPlatformCollector,
|
||
Platform: "newplatform",
|
||
},
|
||
}
|
||
```
|
||
|
||
## 故障排查
|
||
|
||
### 1. 浏览器启动失败
|
||
|
||
- 检查 `ChromePath` 是否正确
|
||
- 确认Chrome版本是否兼容
|
||
- 查看日志文件了解详细错误
|
||
|
||
### 2. 找不到输入框或发送按钮
|
||
|
||
- 页面结构可能已更新
|
||
- 打开浏览器（Headless: false）查看实际DOM结构
|
||
- 更新对应的选择器
|
||
|
||
### 3. 登录状态失效
|
||
|
||
- 删除对应的Cookie文件
|
||
- 重新调用 `WaitLogin()` 登录
|
||
- 检查账号是否正常
|
||
|
||
### 4. 获取不到答案
|
||
|
||
- 增加超时时间
|
||
- 检查网络连接
|
||
- 查看页面是否有验证码或其他拦截
|
||
|
||
## 技术栈
|
||
|
||
- **go-rod**: Chrome DevTools Protocol的Go语言封装
|
||
- **Chrome/Chromium**: 浏览器引擎
|
||
- **Context**: Go上下文管理
|
||
- **JSON**: Cookie序列化
|
||
|
||
## 许可证
|
||
|
||
与项目主许可证保持一致。
|