Method GetQueryEmbedding

api/service/rag/embedding.go:960–1019

GetQueryEmbedding 获取查询文本的向量（使用 singleflight 防止并发重复请求）

(eid int64, query string, channelID int64, config *ChunkConfig)

// GetQueryEmbedding 获取查询文本的向量（使用 singleflight 防止并发重复请求）
func (s *EmbeddingService) GetQueryEmbedding(eid int64, query string, channelID int64, config *ChunkConfig) ([]float64, error) {
	// 验证输入
	if config == nil {
		return nil, fmt.Errorf("配置不能为空")
	}

	query = strings.TrimSpace(query)
	if query == "" {
		return nil, fmt.Errorf("查询内容不能为空")
	}

	modelName := ""
	if config.EmbeddingModelName != nil {
		modelName = *config.EmbeddingModelName
	}
	cacheKey := buildQueryEmbeddingCacheKey(eid, query, channelID, modelName)
	if cacheKey != "" {
		if cachedVector, hit := s.getCachedQueryEmbedding(cacheKey); hit {
			return cachedVector, nil
		}
	}

	// 使用 singleflight 防止并发时对相同查询重复调用 embedding API
	sfKey := fmt.Sprintf("eid:%d:ch:%d:m:%s:q:%s", eid, channelID, modelName, query)
	result, err := embeddingSingleflight.Do(sfKey, func() (interface{}, error) {
		// 再次检查缓存（可能在等待其他请求完成时已写入）
		if cacheKey != "" {
			if cachedVector, hit := s.getCachedQueryEmbedding(cacheKey); hit {
				return cachedVector, nil
			}
		}

		// 获取渠道配置
		channel, err := model.GetChannelByID(channelID)
		if err != nil {
			return nil, fmt.Errorf("获取渠道配置失败: %v", err)
		}

		if channel.Eid != eid {
			return nil, fmt.Errorf("渠道不属于当前企业")
		}

		// 调用embedding API
		vector, err := s.callEmbeddingAPI(query, channel, config, nil)
		if err != nil {
			return nil, err
		}

		if cacheKey != "" {
			s.setCachedQueryEmbedding(cacheKey, vector)
		}
		return vector, nil
	})

	if err != nil {
		return nil, err
	}