rag_service

type AppBriefList struct {
	RagInfos []*common.AppBrief `protobuf:"bytes,1,rep,name=ragInfos,proto3" json:"ragInfos,omitempty"`
	// contains filtered or unexported fields
}

type ChatRagReq struct {
	RagId    string         `protobuf:"bytes,1,opt,name=ragId,proto3" json:"ragId,omitempty"`
	Question string         `protobuf:"bytes,2,opt,name=question,proto3" json:"question,omitempty"`
	Identity *Identity      `protobuf:"bytes,3,opt,name=identity,proto3" json:"identity,omitempty"`
	History  []*HistoryItem `protobuf:"bytes,4,rep,name=history,proto3" json:"history,omitempty"`
	// contains filtered or unexported fields
}

type ChatRagResp struct {
	Content string `protobuf:"bytes,1,opt,name=content,proto3" json:"content,omitempty"`
	// contains filtered or unexported fields
}

type CopyRagReq struct {
	RagId    string    `protobuf:"bytes,1,opt,name=ragId,proto3" json:"ragId,omitempty"`
	Identity *Identity `protobuf:"bytes,2,opt,name=identity,proto3" json:"identity,omitempty"`
	// contains filtered or unexported fields
}

type CreateRagReq struct {
	AppBrief *common.AppBriefConfig `protobuf:"bytes,1,opt,name=appBrief,proto3" json:"appBrief,omitempty"`
	Identity *Identity              `protobuf:"bytes,2,opt,name=identity,proto3" json:"identity,omitempty"`
	// contains filtered or unexported fields
}

type CreateRagResp struct {
	RagId string `protobuf:"bytes,1,opt,name=ragId,proto3" json:"ragId,omitempty"`
	// contains filtered or unexported fields
}

type GetRagByIdsReq struct {
	RagIdList []string `protobuf:"bytes,1,rep,name=ragIdList,proto3" json:"ragIdList,omitempty"`
	// contains filtered or unexported fields
}

type HistoryItem struct {
	Query       string `protobuf:"bytes,1,opt,name=query,proto3" json:"query,omitempty"`
	Response    string `protobuf:"bytes,2,opt,name=response,proto3" json:"response,omitempty"`
	NeedHistory bool   `protobuf:"varint,3,opt,name=needHistory,proto3" json:"needHistory,omitempty"`
	// contains filtered or unexported fields
}

type Identity struct {
	UserId string `protobuf:"bytes,1,opt,name=userId,proto3" json:"userId,omitempty"`
	OrgId  string `protobuf:"bytes,2,opt,name=orgId,proto3" json:"orgId,omitempty"`
	// contains filtered or unexported fields
}

type RagDeleteReq struct {
	RagId string `protobuf:"bytes,1,opt,name=ragId,proto3" json:"ragId,omitempty"`
	// contains filtered or unexported fields
}

type RagDetailReq struct {
	RagId string `protobuf:"bytes,1,opt,name=ragId,proto3" json:"ragId,omitempty"`
	// contains filtered or unexported fields
}

type RagGlobalConfig struct {
	MaxHistory        int32   `protobuf:"varint,1,opt,name=maxHistory,proto3" json:"maxHistory,omitempty"`                // 最长上下文
	Threshold         float32 `protobuf:"fixed32,2,opt,name=threshold,proto3" json:"threshold,omitempty"`                 // 过滤阈值
	TopK              int32   `protobuf:"varint,3,opt,name=topK,proto3" json:"topK,omitempty"`                            // topK
	MatchType         string  `protobuf:"bytes,4,opt,name=matchType,proto3" json:"matchType,omitempty"`                   //matchType：vector（向量检索）、text（文本检索）、mix（混合检索：向量+文本）
	KeywordPriority   float32 `protobuf:"fixed32,5,opt,name=keywordPriority,proto3" json:"keywordPriority,omitempty"`     // 关键词权重
	PriorityMatch     int32   `protobuf:"varint,6,opt,name=priorityMatch,proto3" json:"priorityMatch,omitempty"`          // 权重匹配，只有在混合检索模式下，选择权重设置后，这个才设置为1
	SemanticsPriority float32 `protobuf:"fixed32,7,opt,name=semanticsPriority,proto3" json:"semanticsPriority,omitempty"` // 语义权重
	TermWeight        float32 `protobuf:"fixed32,8,opt,name=termWeight,proto3" json:"termWeight,omitempty"`               // 关键词系数，默认为1
	TermWeightEnable  bool    `protobuf:"varint,9,opt,name=termWeightEnable,proto3" json:"termWeightEnable,omitempty"`    // 关键词系数开关
	UseGraph          bool    `protobuf:"varint,10,opt,name=useGraph,proto3" json:"useGraph,omitempty"`                   // 知识图谱开关
	// contains filtered or unexported fields
}

type RagInfo struct {
	RagId               string                  `protobuf:"bytes,1,opt,name=ragId,proto3" json:"ragId,omitempty"`
	BriefConfig         *common.AppBriefConfig  `protobuf:"bytes,2,opt,name=briefConfig,proto3" json:"briefConfig,omitempty"`
	ModelConfig         *common.AppModelConfig  `protobuf:"bytes,3,opt,name=modelConfig,proto3" json:"modelConfig,omitempty"`                 // 模型
	RerankConfig        *common.AppModelConfig  `protobuf:"bytes,4,opt,name=rerankConfig,proto3" json:"rerankConfig,omitempty"`               // Rerank模型
	KnowledgeBaseConfig *RagKnowledgeBaseConfig `protobuf:"bytes,5,opt,name=knowledgeBaseConfig,proto3" json:"knowledgeBaseConfig,omitempty"` // 知识库
	SensitiveConfig     *RagSensitiveConfig     `protobuf:"bytes,6,opt,name=sensitiveConfig,proto3" json:"sensitiveConfig,omitempty"`         // 安全护栏
	Identity            *Identity               `protobuf:"bytes,7,opt,name=identity,proto3" json:"identity,omitempty"`
	// contains filtered or unexported fields
}

type RagKnowledgeBaseConfig struct {
	PerKnowledgeConfigs []*RagPerKnowledgeConfig `protobuf:"bytes,1,rep,name=perKnowledgeConfigs,proto3" json:"perKnowledgeConfigs,omitempty"`
	GlobalConfig        *RagGlobalConfig         `protobuf:"bytes,2,opt,name=globalConfig,proto3" json:"globalConfig,omitempty"`
	// contains filtered or unexported fields
}

type RagListReq struct {
	Name     string    `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Identity *Identity `protobuf:"bytes,2,opt,name=identity,proto3" json:"identity,omitempty"`
	// contains filtered or unexported fields
}

type RagListResp struct {
	RagInfos []*common.AppBrief `protobuf:"bytes,1,rep,name=ragInfos,proto3" json:"ragInfos,omitempty"`
	Total    int64              `protobuf:"varint,2,opt,name=total,proto3" json:"total,omitempty"`
	// contains filtered or unexported fields
}

type RagMetaFilter struct {
	FilterEnable    bool                 `protobuf:"varint,1,opt,name=filterEnable,proto3" json:"filterEnable,omitempty"`
	FilterLogicType string               `protobuf:"bytes,2,opt,name=filterLogicType,proto3" json:"filterLogicType,omitempty"`
	FilterItems     []*RagMetaFilterItem `protobuf:"bytes,3,rep,name=filterItems,proto3" json:"filterItems,omitempty"`
	// contains filtered or unexported fields
}

type RagMetaFilterItem struct {
	Key       string `protobuf:"bytes,1,opt,name=key,proto3" json:"key,omitempty"`
	Type      string `protobuf:"bytes,2,opt,name=type,proto3" json:"type,omitempty"`
	Value     string `protobuf:"bytes,3,opt,name=value,proto3" json:"value,omitempty"`
	Condition string `protobuf:"bytes,4,opt,name=condition,proto3" json:"condition,omitempty"`
	// contains filtered or unexported fields
}

type RagPerKnowledgeConfig struct {
	KnowledgeId   string         `protobuf:"bytes,1,opt,name=knowledgeId,proto3" json:"knowledgeId,omitempty"`
	GraphSwitch   int32          `protobuf:"varint,2,opt,name=graphSwitch,proto3" json:"graphSwitch,omitempty"`
	RagMetaFilter *RagMetaFilter `protobuf:"bytes,3,opt,name=ragMetaFilter,proto3" json:"ragMetaFilter,omitempty"`
	// contains filtered or unexported fields
}

type RagSensitiveConfig struct {
	Enable   bool     `protobuf:"varint,1,opt,name=enable,proto3" json:"enable,omitempty"`    // 是否启用安全护栏
	TableIds []string `protobuf:"bytes,2,rep,name=tableIds,proto3" json:"tableIds,omitempty"` // 敏感词表id
	// contains filtered or unexported fields
}

type RagServiceClient interface {
	// 流式对话
	ChatRag(ctx context.Context, in *ChatRagReq, opts ...grpc.CallOption) (grpc.ServerStreamingClient[ChatRagResp], error)
	// 创建 rag
	CreateRag(ctx context.Context, in *CreateRagReq, opts ...grpc.CallOption) (*CreateRagResp, error)
	// 更新 rag 基本信息
	UpdateRag(ctx context.Context, in *UpdateRagReq, opts ...grpc.CallOption) (*emptypb.Empty, error)
	// 更新 rag 配置信息
	UpdateRagConfig(ctx context.Context, in *UpdateRagConfigReq, opts ...grpc.CallOption) (*emptypb.Empty, error)
	// 删除 rag
	DeleteRag(ctx context.Context, in *RagDeleteReq, opts ...grpc.CallOption) (*emptypb.Empty, error)
	// 获取 rag
	GetRagDetail(ctx context.Context, in *RagDetailReq, opts ...grpc.CallOption) (*RagInfo, error)
	// 获取 rag 列表
	ListRag(ctx context.Context, in *RagListReq, opts ...grpc.CallOption) (*RagListResp, error)
	// 根据 ragIds 获取 rag 列表
	GetRagByIds(ctx context.Context, in *GetRagByIdsReq, opts ...grpc.CallOption) (*AppBriefList, error)
	// 复制 rag
	CopyRag(ctx context.Context, in *CopyRagReq, opts ...grpc.CallOption) (*CreateRagResp, error)
}

type RagServiceServer interface {
	// 流式对话
	ChatRag(*ChatRagReq, grpc.ServerStreamingServer[ChatRagResp]) error
	// 创建 rag
	CreateRag(context.Context, *CreateRagReq) (*CreateRagResp, error)
	// 更新 rag 基本信息
	UpdateRag(context.Context, *UpdateRagReq) (*emptypb.Empty, error)
	// 更新 rag 配置信息
	UpdateRagConfig(context.Context, *UpdateRagConfigReq) (*emptypb.Empty, error)
	// 删除 rag
	DeleteRag(context.Context, *RagDeleteReq) (*emptypb.Empty, error)
	// 获取 rag
	GetRagDetail(context.Context, *RagDetailReq) (*RagInfo, error)
	// 获取 rag 列表
	ListRag(context.Context, *RagListReq) (*RagListResp, error)
	// 根据 ragIds 获取 rag 列表
	GetRagByIds(context.Context, *GetRagByIdsReq) (*AppBriefList, error)
	// 复制 rag
	CopyRag(context.Context, *CopyRagReq) (*CreateRagResp, error)
	// contains filtered or unexported methods
}

type RagService_ChatRagClient = grpc.ServerStreamingClient[ChatRagResp]

type RagService_ChatRagServer = grpc.ServerStreamingServer[ChatRagResp]

type UnimplementedRagServiceServer struct{}

type UnsafeRagServiceServer interface {
	// contains filtered or unexported methods
}

type UpdateRagConfigReq struct {
	RagId               string                  `protobuf:"bytes,1,opt,name=ragId,proto3" json:"ragId,omitempty"`
	ModelConfig         *common.AppModelConfig  `protobuf:"bytes,2,opt,name=modelConfig,proto3" json:"modelConfig,omitempty"`                 // 模型
	RerankConfig        *common.AppModelConfig  `protobuf:"bytes,3,opt,name=rerankConfig,proto3" json:"rerankConfig,omitempty"`               // Rerank模型
	KnowledgeBaseConfig *RagKnowledgeBaseConfig `protobuf:"bytes,4,opt,name=knowledgeBaseConfig,proto3" json:"knowledgeBaseConfig,omitempty"` // 知识库
	SensitiveConfig     *RagSensitiveConfig     `protobuf:"bytes,5,opt,name=sensitiveConfig,proto3" json:"sensitiveConfig,omitempty"`         // 安全护栏
	// contains filtered or unexported fields
}

type UpdateRagReq struct {
	RagId    string                 `protobuf:"bytes,1,opt,name=ragId,proto3" json:"ragId,omitempty"`
	AppBrief *common.AppBriefConfig `protobuf:"bytes,2,opt,name=appBrief,proto3" json:"appBrief,omitempty"`
	Identity *Identity              `protobuf:"bytes,3,opt,name=identity,proto3" json:"identity,omitempty"`
	// contains filtered or unexported fields
}