51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package biz
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"geo/internal/collect"
|
|
"geo/internal/config"
|
|
|
|
"github.com/gofiber/fiber/v2/log"
|
|
)
|
|
|
|
// CollectBiz AI收集业务层
|
|
type CollectBiz struct {
|
|
manager *collect.CollectManager
|
|
config *config.Config
|
|
logger log.AllLogger
|
|
}
|
|
|
|
// NewCollectBiz 创建AI收集业务实例
|
|
func NewCollectBiz(ctx context.Context, cfg *config.Config, logger log.AllLogger) *CollectBiz {
|
|
manager := collect.NewCollectManager(ctx, cfg, logger)
|
|
return &CollectBiz{
|
|
manager: manager,
|
|
config: cfg,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
// AskAIQuestion 向指定AI平台提问
|
|
// platform: 平台类型 (wenxin, deepseek, doubao, qianwen)
|
|
// requestID: 请求ID
|
|
// question: 问题内容
|
|
// headless: 是否无头模式
|
|
func (b *CollectBiz) AskAIQuestion(platform string, requestID, question string, keywords []string, headless bool) (*collect.CollectResult, error) {
|
|
result, err := b.manager.AskQuestion(platform, question)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("向%s提问失败: %w", platform, err)
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
// WaitAILogin 等待AI平台登录 - 已废弃
|
|
// 注意：新架构中不再需要单独的登录方法，登录状态通过 Cookie 自动维持
|
|
|
|
// ListAIPlatforms 列出所有支持的AI平台
|
|
func (b *CollectBiz) ListAIPlatforms() []string {
|
|
platforms := []string{"wenxin", "deepseek", "doubao", "qianwen"}
|
|
return platforms
|
|
}
|