9902ca3563
后端： 1. 阶段 1.5/1.6 收口 llm-service / rag-service，统一模型出口与检索基础设施入口，清退 backend/infra/llm 与 backend/infra/rag 旧实现； 2. 同步更新相关调用链与微服务迁移计划文档
112 lines
2.9 KiB
Go
112 lines
2.9 KiB
Go
package rag
|
|
|
|
import (
|
|
"context"
|
|
|
|
ragconfig "github.com/LoveLosita/smartflow/backend/services/rag/config"
|
|
)
|
|
|
|
// Options 描述 rag-service 需要持有的底层运行时。
|
|
type Options struct {
|
|
Runtime Runtime
|
|
}
|
|
|
|
// Service 是 rag-service 对外暴露的统一入口。
|
|
//
|
|
// 职责边界：
|
|
// 1. 负责持有运行时，并把 memory / web 两条能力线统一收口到服务层。
|
|
// 2. 负责在服务入口内完成基于配置的运行时装配。
|
|
// 3. 不直接承载 chunk / embed / store 的实现细节，这些细节下沉到服务树内部子包。
|
|
type Service struct {
|
|
runtime Runtime
|
|
}
|
|
|
|
// New 使用调用方传入的运行时构造服务。
|
|
func New(opts Options) *Service {
|
|
return &Service{runtime: opts.Runtime}
|
|
}
|
|
|
|
// NewFromConfig 基于服务树内的配置与工厂能力构造自给自足的 RAG 服务。
|
|
func NewFromConfig(ctx context.Context, cfg ragconfig.Config, deps FactoryDeps) (*Service, error) {
|
|
if !cfg.Enabled {
|
|
return New(Options{}), nil
|
|
}
|
|
runtime, err := NewRuntimeFromConfig(ctx, cfg, deps)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewWithRuntime(runtime), nil
|
|
}
|
|
|
|
// Runtime 返回当前服务持有的运行时。
|
|
func (s *Service) Runtime() Runtime {
|
|
if s == nil {
|
|
return nil
|
|
}
|
|
return s.runtime
|
|
}
|
|
|
|
// IngestMemory 写入记忆语料。
|
|
func (s *Service) IngestMemory(ctx context.Context, req MemoryIngestRequest) (*IngestResult, error) {
|
|
if s == nil || s.runtime == nil {
|
|
return nil, nil
|
|
}
|
|
return s.runtime.IngestMemory(ctx, req)
|
|
}
|
|
|
|
// RetrieveMemory 检索记忆语料。
|
|
func (s *Service) RetrieveMemory(ctx context.Context, req MemoryRetrieveRequest) (*RetrieveResult, error) {
|
|
if s == nil || s.runtime == nil {
|
|
return nil, nil
|
|
}
|
|
return s.runtime.RetrieveMemory(ctx, req)
|
|
}
|
|
|
|
// DeleteMemory 删除指定记忆文档。
|
|
func (s *Service) DeleteMemory(ctx context.Context, documentIDs []string) error {
|
|
if s == nil || s.runtime == nil {
|
|
return nil
|
|
}
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
return s.runtime.DeleteMemory(ctx, documentIDs)
|
|
}
|
|
|
|
// IngestWeb 写入网页语料。
|
|
func (s *Service) IngestWeb(ctx context.Context, req WebIngestRequest) (*IngestResult, error) {
|
|
if s == nil || s.runtime == nil {
|
|
return nil, nil
|
|
}
|
|
return s.runtime.IngestWeb(ctx, req)
|
|
}
|
|
|
|
// RetrieveWeb 检索网页语料。
|
|
func (s *Service) RetrieveWeb(ctx context.Context, req WebRetrieveRequest) (*RetrieveResult, error) {
|
|
if s == nil || s.runtime == nil {
|
|
return nil, nil
|
|
}
|
|
return s.runtime.RetrieveWeb(ctx, req)
|
|
}
|
|
|
|
// EnsureRuntime 返回一个可继续向下传递的运行时引用。
|
|
func (s *Service) EnsureRuntime() Runtime {
|
|
if s == nil {
|
|
return nil
|
|
}
|
|
return s.runtime
|
|
}
|
|
|
|
// SetRuntime 允许在装配阶段延迟注入运行时。
|
|
func (s *Service) SetRuntime(runtime Runtime) {
|
|
if s == nil {
|
|
return
|
|
}
|
|
s.runtime = runtime
|
|
}
|
|
|
|
// NewWithRuntime 用显式运行时构造服务。
|
|
func NewWithRuntime(runtime Runtime) *Service {
|
|
return New(Options{Runtime: runtime})
|
|
}
|