knowledgebase_service

type CreateKnowledgeReq 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"`
	Name               string              `protobuf:"bytes,3,opt,name=name,proto3" json:"name,omitempty"`
	Description        string              `protobuf:"bytes,4,opt,name=description,proto3" json:"description,omitempty"`
	EmbeddingModelInfo *EmbeddingModelInfo `protobuf:"bytes,5,opt,name=embeddingModelInfo,proto3" json:"embeddingModelInfo,omitempty"`
	// contains filtered or unexported fields
}

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

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

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

type KnowledgeBaseServiceClient interface {
	// 获取知识库列表
	SelectKnowledgeList(ctx context.Context, in *KnowledgeSelectReq, opts ...grpc.CallOption) (*KnowledgeSelectListResp, error)
	// 获取知识库详情
	SelectKnowledgeDetailById(ctx context.Context, in *KnowledgeDetailSelectReq, opts ...grpc.CallOption) (*KnowledgeInfo, error)
	// 获取知识库详情列表
	SelectKnowledgeDetailByIdList(ctx context.Context, in *KnowledgeDetailSelectListReq, opts ...grpc.CallOption) (*KnowledgeDetailSelectListResp, error)
	// 获取知识库详情
	SelectKnowledgeDetailByName(ctx context.Context, in *KnowledgeDetailSelectReq, opts ...grpc.CallOption) (*KnowledgeInfo, error)
	// 新增知识库
	CreateKnowledge(ctx context.Context, in *CreateKnowledgeReq, opts ...grpc.CallOption) (*CreateKnowledgeResp, error)
	// 修改知识库
	UpdateKnowledge(ctx context.Context, in *UpdateKnowledgeReq, opts ...grpc.CallOption) (*emptypb.Empty, error)
	// 删除文档知识分类
	DeleteKnowledge(ctx context.Context, in *DeleteKnowledgeReq, opts ...grpc.CallOption) (*emptypb.Empty, error)
	// 知识库命中测试
	KnowledgeHit(ctx context.Context, in *KnowledgeHitReq, opts ...grpc.CallOption) (*KnowledgeHitResp, error)
}

type KnowledgeBaseServiceServer interface {
	// 获取知识库列表
	SelectKnowledgeList(context.Context, *KnowledgeSelectReq) (*KnowledgeSelectListResp, error)
	// 获取知识库详情
	SelectKnowledgeDetailById(context.Context, *KnowledgeDetailSelectReq) (*KnowledgeInfo, error)
	// 获取知识库详情列表
	SelectKnowledgeDetailByIdList(context.Context, *KnowledgeDetailSelectListReq) (*KnowledgeDetailSelectListResp, error)
	// 获取知识库详情
	SelectKnowledgeDetailByName(context.Context, *KnowledgeDetailSelectReq) (*KnowledgeInfo, error)
	// 新增知识库
	CreateKnowledge(context.Context, *CreateKnowledgeReq) (*CreateKnowledgeResp, error)
	// 修改知识库
	UpdateKnowledge(context.Context, *UpdateKnowledgeReq) (*emptypb.Empty, error)
	// 删除文档知识分类
	DeleteKnowledge(context.Context, *DeleteKnowledgeReq) (*emptypb.Empty, error)
	// 知识库命中测试
	KnowledgeHit(context.Context, *KnowledgeHitReq) (*KnowledgeHitResp, error)
	// contains filtered or unexported methods
}

type KnowledgeDetailSelectListReq 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"`
	KnowledgeIds []string `protobuf:"bytes,3,rep,name=knowledgeIds,proto3" json:"knowledgeIds,omitempty"` //知识库id列表
	// contains filtered or unexported fields
}

type KnowledgeDetailSelectListResp struct {
	List  []*KnowledgeInfo `protobuf:"bytes,1,rep,name=list,proto3" json:"list,omitempty"` // 知识库详情列表
	Total int32            `protobuf:"varint,2,opt,name=total,proto3" json:"total,omitempty"`
	// contains filtered or unexported fields
}

type KnowledgeDetailSelectReq 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"`
	KnowledgeId   string `protobuf:"bytes,3,opt,name=knowledgeId,proto3" json:"knowledgeId,omitempty"`     //知识库id
	KnowledgeName string `protobuf:"bytes,4,opt,name=knowledgeName,proto3" json:"knowledgeName,omitempty"` // 知识库名称
	// contains filtered or unexported fields
}

type KnowledgeHitReq struct {
	OrgId                string                `protobuf:"bytes,1,opt,name=orgId,proto3" json:"orgId,omitempty"`
	UserId               string                `protobuf:"bytes,2,opt,name=userId,proto3" json:"userId,omitempty"`
	Question             string                `protobuf:"bytes,3,opt,name=question,proto3" json:"question,omitempty"`                         //问题
	KnowledgeIdList      []string              `protobuf:"bytes,4,rep,name=knowledgeIdList,proto3" json:"knowledgeIdList,omitempty"`           //知识库id 列表
	KnowledgeMatchParams *KnowledgeMatchParams `protobuf:"bytes,5,opt,name=knowledgeMatchParams,proto3" json:"knowledgeMatchParams,omitempty"` //模型匹配方式
	// contains filtered or unexported fields
}

type KnowledgeHitResp struct {
	Prompt     string                 `protobuf:"bytes,1,opt,name=prompt,proto3" json:"prompt,omitempty"`
	Score      []float64              `protobuf:"fixed64,2,rep,packed,name=score,proto3" json:"score,omitempty"`
	SearchList []*KnowledgeSearchInfo `protobuf:"bytes,3,rep,name=searchList,proto3" json:"searchList,omitempty"`
	// contains filtered or unexported fields
}

type KnowledgeInfo struct {
	KnowledgeId          string              `protobuf:"bytes,1,opt,name=knowledgeId,proto3" json:"knowledgeId,omitempty"` //知识库id
	Name                 string              `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`               //知识库名称
	DocCount             int32               `protobuf:"varint,3,opt,name=docCount,proto3" json:"docCount,omitempty"`
	Description          string              `protobuf:"bytes,4,opt,name=description,proto3" json:"description,omitempty"`
	EmbeddingModelInfo   *EmbeddingModelInfo `protobuf:"bytes,5,opt,name=embeddingModelInfo,proto3" json:"embeddingModelInfo,omitempty"`
	KnowledgeTagInfoList []*KnowledgeTagInfo `protobuf:"bytes,6,rep,name=knowledgeTagInfoList,proto3" json:"knowledgeTagInfoList,omitempty"`
	CreatedAt            string              `protobuf:"bytes,7,opt,name=createdAt,proto3" json:"createdAt,omitempty"`
	// contains filtered or unexported fields
}

type KnowledgeMatchParams struct {
	MatchType         string  `protobuf:"bytes,1,opt,name=matchType,proto3" json:"matchType,omitempty"`                   //matchType：vector（向量检索）、text（文本检索）、mix（混合检索：向量+文本）
	RerankModelId     string  `protobuf:"bytes,2,opt,name=rerankModelId,proto3" json:"rerankModelId,omitempty"`           //rerank模型id
	PriorityMatch     int32   `protobuf:"varint,3,opt,name=priorityMatch,proto3" json:"priorityMatch,omitempty"`          // 权重匹配，只有在混合检索模式下，选择权重设置后，这个才设置为1
	SemanticsPriority float32 `protobuf:"fixed32,4,opt,name=semanticsPriority,proto3" json:"semanticsPriority,omitempty"` // 语义权重
	KeywordPriority   float32 `protobuf:"fixed32,5,opt,name=keywordPriority,proto3" json:"keywordPriority,omitempty"`     // 关键词权重
	TopK              int32   `protobuf:"varint,6,opt,name=topK,proto3" json:"topK,omitempty"`                            //topK 获取最高的几行
	Score             float32 `protobuf:"fixed32,7,opt,name=score,proto3" json:"score,omitempty"`                         //score 过滤分数阈值
	TermWeight        float32 `protobuf:"fixed32,8,opt,name=termWeight,proto3" json:"termWeight,omitempty"`               // 关键词系数
	TermWeightEnable  bool    `protobuf:"varint,9,opt,name=termWeightEnable,proto3" json:"termWeightEnable,omitempty"`    // 关键词系数开关
	// contains filtered or unexported fields
}

type KnowledgeSearchInfo struct {
	Title         string `protobuf:"bytes,1,opt,name=title,proto3" json:"title,omitempty"`
	Snippet       string `protobuf:"bytes,2,opt,name=snippet,proto3" json:"snippet,omitempty"`
	KnowledgeName string `protobuf:"bytes,3,opt,name=knowledgeName,proto3" json:"knowledgeName,omitempty"`
	// contains filtered or unexported fields
}

type KnowledgeSelectListResp struct {
	KnowledgeList []*KnowledgeInfo `protobuf:"bytes,1,rep,name=knowledgeList,proto3" json:"knowledgeList,omitempty"` //知识库列表
	// contains filtered or unexported fields
}

type KnowledgeSelectReq 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"`
	Name      string   `protobuf:"bytes,3,opt,name=name,proto3" json:"name,omitempty"`           //知识库名称，支持模糊搜索
	TagIdList []string `protobuf:"bytes,4,rep,name=tagIdList,proto3" json:"tagIdList,omitempty"` //知识库id 列表
	// contains filtered or unexported fields
}

type KnowledgeTagInfo struct {
	TagId   string `protobuf:"bytes,1,opt,name=tagId,proto3" json:"tagId,omitempty"`     //知识库标签id
	TagName string `protobuf:"bytes,2,opt,name=tagName,proto3" json:"tagName,omitempty"` //知识库标签名称
	// contains filtered or unexported fields
}

type UnimplementedKnowledgeBaseServiceServer struct{}

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

type UpdateKnowledgeReq struct {
	KnowledgeId string `protobuf:"bytes,1,opt,name=knowledgeId,proto3" json:"knowledgeId,omitempty"`
	Name        string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`
	Description string `protobuf:"bytes,3,opt,name=description,proto3" json:"description,omitempty"`
	UserId      string `protobuf:"bytes,4,opt,name=userId,proto3" json:"userId,omitempty"`
	OrgId       string `protobuf:"bytes,5,opt,name=orgId,proto3" json:"orgId,omitempty"`
	// contains filtered or unexported fields
}