9902ca3563
后端： 1. 阶段 1.5/1.6 收口 llm-service / rag-service，统一模型出口与检索基础设施入口，清退 backend/infra/llm 与 backend/infra/rag 旧实现； 2. 同步更新相关调用链与微服务迁移计划文档
39 lines
1.2 KiB
Go
39 lines
1.2 KiB
Go
package core
|
|
|
|
import "context"
|
|
|
|
// Chunker 负责文本切块。
|
|
type Chunker interface {
|
|
Chunk(ctx context.Context, doc SourceDocument, opt ChunkOption) ([]Chunk, error)
|
|
}
|
|
|
|
// Embedder 负责向量化。
|
|
type Embedder interface {
|
|
Embed(ctx context.Context, texts []string, action string) ([][]float32, error)
|
|
}
|
|
|
|
// Retriever 负责召回候选。
|
|
type Retriever interface {
|
|
Retrieve(ctx context.Context, req RetrieveRequest) ([]ScoredChunk, error)
|
|
}
|
|
|
|
// Reranker 负责重排候选。
|
|
type Reranker interface {
|
|
Rerank(ctx context.Context, query string, candidates []ScoredChunk, topK int) ([]ScoredChunk, error)
|
|
}
|
|
|
|
// VectorStore 负责向量库读写。
|
|
type VectorStore interface {
|
|
Upsert(ctx context.Context, rows []VectorRow) error
|
|
Search(ctx context.Context, req VectorSearchRequest) ([]ScoredVectorRow, error)
|
|
Delete(ctx context.Context, ids []string) error
|
|
Get(ctx context.Context, ids []string) ([]VectorRow, error)
|
|
}
|
|
|
|
// CorpusAdapter 负责把业务语料映射成统一文档/过滤条件。
|
|
type CorpusAdapter interface {
|
|
Name() string
|
|
BuildIngestDocuments(ctx context.Context, input any) ([]SourceDocument, error)
|
|
BuildRetrieveFilter(ctx context.Context, req any) (map[string]any, error)
|
|
}
|