yourproject

type CreateKBRequest struct {
	Uid            string `protobuf:"bytes,1,opt,name=uid,proto3" json:"uid,omitempty"`                                             // 用户ID
	KbName         string `protobuf:"bytes,2,opt,name=kb_name,json=kbName,proto3" json:"kb_name,omitempty"`                         // 知识库名称
	EmbeddingModel string `protobuf:"bytes,3,opt,name=embedding_model,json=embeddingModel,proto3" json:"embedding_model,omitempty"` // (可选) 指定Embedding模型, 留空则使用默认
	// contains filtered or unexported fields
}

type CreateKBResponse struct {
	Success bool   `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"`      // 是否成功
	KbId    string `protobuf:"bytes,2,opt,name=kb_id,json=kbId,proto3" json:"kb_id,omitempty"` // 知识库ID
	Message string `protobuf:"bytes,3,opt,name=message,proto3" json:"message,omitempty"`       // 响应消息
	// contains filtered or unexported fields
}

type DeleteDocRequest struct {
	Uid   string `protobuf:"bytes,1,opt,name=uid,proto3" json:"uid,omitempty"`                  // 用户ID
	KbId  string `protobuf:"bytes,2,opt,name=kb_id,json=kbId,proto3" json:"kb_id,omitempty"`    // 知识库ID
	DocId string `protobuf:"bytes,3,opt,name=doc_id,json=docId,proto3" json:"doc_id,omitempty"` // 文档ID
	// contains filtered or unexported fields
}

type DeleteDocResponse struct {
	Success bool   `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"` // 是否成功
	Message string `protobuf:"bytes,2,opt,name=message,proto3" json:"message,omitempty"`  // 响应消息
	// contains filtered or unexported fields
}

type DeleteKBRequest struct {
	Uid  string `protobuf:"bytes,1,opt,name=uid,proto3" json:"uid,omitempty"`               // 用户ID
	KbId string `protobuf:"bytes,2,opt,name=kb_id,json=kbId,proto3" json:"kb_id,omitempty"` // 知识库ID
	// contains filtered or unexported fields
}

type DeleteKBResponse struct {
	Success bool   `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"` // 是否成功
	Message string `protobuf:"bytes,2,opt,name=message,proto3" json:"message,omitempty"`  // 响应消息
	// contains filtered or unexported fields
}

type DocumentInfo struct {
	DocId     string `protobuf:"bytes,1,opt,name=doc_id,json=docId,proto3" json:"doc_id,omitempty"`             // 文档ID
	KbId      string `protobuf:"bytes,2,opt,name=kb_id,json=kbId,proto3" json:"kb_id,omitempty"`                // 知识库ID
	DocName   string `protobuf:"bytes,3,opt,name=doc_name,json=docName,proto3" json:"doc_name,omitempty"`       // 文档名称
	FileType  string `protobuf:"bytes,4,opt,name=file_type,json=fileType,proto3" json:"file_type,omitempty"`    // 文件类型
	CreatedAt string `protobuf:"bytes,5,opt,name=created_at,json=createdAt,proto3" json:"created_at,omitempty"` // 创建时间
	// contains filtered or unexported fields
}

type KnowledgeBaseInfo struct {
	KbId           string `protobuf:"bytes,1,opt,name=kb_id,json=kbId,proto3" json:"kb_id,omitempty"`                               // 知识库ID
	KbName         string `protobuf:"bytes,2,opt,name=kb_name,json=kbName,proto3" json:"kb_name,omitempty"`                         // 知识库名称
	EmbeddingModel string `protobuf:"bytes,3,opt,name=embedding_model,json=embeddingModel,proto3" json:"embedding_model,omitempty"` // 使用的Embedding模型
	// contains filtered or unexported fields
}

type KnowledgeBaseServiceClient interface {
	// 创建知识库
	CreateKnowledgeBase(ctx context.Context, in *CreateKBRequest, opts ...grpc.CallOption) (*CreateKBResponse, error)
	// 获取用户的所有知识库
	ListKnowledgeBases(ctx context.Context, in *ListKBsRequest, opts ...grpc.CallOption) (*ListKBsResponse, error)
	// 上传并处理文档
	UploadDocument(ctx context.Context, in *UploadDocRequest, opts ...grpc.CallOption) (*UploadDocResponse, error)
	// 从知识库检索信息
	RetrieveInfo(ctx context.Context, in *RetrieveRequest, opts ...grpc.CallOption) (*RetrieveResponse, error)
	// 获取支持的Embedding模型列表
	ListEmbeddingModels(ctx context.Context, in *ListEmbeddingModelsRequest, opts ...grpc.CallOption) (*ListEmbeddingModelsResponse, error)
	// 获取知识库中的文档列表
	ListDocuments(ctx context.Context, in *ListDocsRequest, opts ...grpc.CallOption) (*ListDocsResponse, error)
	// 删除知识库
	DeleteKnowledgeBase(ctx context.Context, in *DeleteKBRequest, opts ...grpc.CallOption) (*DeleteKBResponse, error)
	// 删除文档
	DeleteDocument(ctx context.Context, in *DeleteDocRequest, opts ...grpc.CallOption) (*DeleteDocResponse, error)
}

type KnowledgeBaseServiceServer interface {
	// 创建知识库
	CreateKnowledgeBase(context.Context, *CreateKBRequest) (*CreateKBResponse, error)
	// 获取用户的所有知识库
	ListKnowledgeBases(context.Context, *ListKBsRequest) (*ListKBsResponse, error)
	// 上传并处理文档
	UploadDocument(context.Context, *UploadDocRequest) (*UploadDocResponse, error)
	// 从知识库检索信息
	RetrieveInfo(context.Context, *RetrieveRequest) (*RetrieveResponse, error)
	// 获取支持的Embedding模型列表
	ListEmbeddingModels(context.Context, *ListEmbeddingModelsRequest) (*ListEmbeddingModelsResponse, error)
	// 获取知识库中的文档列表
	ListDocuments(context.Context, *ListDocsRequest) (*ListDocsResponse, error)
	// 删除知识库
	DeleteKnowledgeBase(context.Context, *DeleteKBRequest) (*DeleteKBResponse, error)
	// 删除文档
	DeleteDocument(context.Context, *DeleteDocRequest) (*DeleteDocResponse, error)
	// contains filtered or unexported methods
}

type ListDocsRequest struct {
	KbId string `protobuf:"bytes,1,opt,name=kb_id,json=kbId,proto3" json:"kb_id,omitempty"` // 知识库ID
	// contains filtered or unexported fields
}

type ListDocsResponse struct {
	Success bool            `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"` // 是否成功
	Docs    []*DocumentInfo `protobuf:"bytes,2,rep,name=docs,proto3" json:"docs,omitempty"`        // 文档信息列表
	Message string          `protobuf:"bytes,3,opt,name=message,proto3" json:"message,omitempty"`  // 响应消息
	// contains filtered or unexported fields
}

type ListEmbeddingModelsRequest struct {
	// contains filtered or unexported fields
}

type ListEmbeddingModelsResponse struct {
	Success bool     `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"` // 是否成功
	Models  []string `protobuf:"bytes,2,rep,name=models,proto3" json:"models,omitempty"`    // 支持的模型名称列表
	Message string   `protobuf:"bytes,3,opt,name=message,proto3" json:"message,omitempty"`  // 响应消息
	// contains filtered or unexported fields
}

type ListKBsRequest struct {
	Uid string `protobuf:"bytes,1,opt,name=uid,proto3" json:"uid,omitempty"` // 用户ID
	// contains filtered or unexported fields
}

type ListKBsResponse struct {
	Success bool                 `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"` // 是否成功
	Kbs     []*KnowledgeBaseInfo `protobuf:"bytes,2,rep,name=kbs,proto3" json:"kbs,omitempty"`          // 知识库信息列表
	Message string               `protobuf:"bytes,3,opt,name=message,proto3" json:"message,omitempty"`  // 响应消息
	// contains filtered or unexported fields
}

type RetrieveRequest struct {
	KbId  string `protobuf:"bytes,1,opt,name=kb_id,json=kbId,proto3" json:"kb_id,omitempty"`  // 知识库ID
	Query string `protobuf:"bytes,2,opt,name=query,proto3" json:"query,omitempty"`            // 查询问题
	TopK  int32  `protobuf:"varint,3,opt,name=top_k,json=topK,proto3" json:"top_k,omitempty"` // 返回结果数量
	// contains filtered or unexported fields
}

type RetrieveResponse struct {
	Success bool              `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"` // 是否成功
	Results []*RetrieveResult `protobuf:"bytes,2,rep,name=results,proto3" json:"results,omitempty"`  // 检索结果列表
	Message string            `protobuf:"bytes,3,opt,name=message,proto3" json:"message,omitempty"`  // 响应消息
	// contains filtered or unexported fields
}

type RetrieveResult struct {
	Content string  `protobuf:"bytes,1,opt,name=content,proto3" json:"content,omitempty"`                // 检索到的内容
	Score   float32 `protobuf:"fixed32,2,opt,name=score,proto3" json:"score,omitempty"`                  // 相关性分数
	DocId   string  `protobuf:"bytes,3,opt,name=doc_id,json=docId,proto3" json:"doc_id,omitempty"`       // 文档ID
	DocName string  `protobuf:"bytes,4,opt,name=doc_name,json=docName,proto3" json:"doc_name,omitempty"` // 文档名称
	// contains filtered or unexported fields
}

type UnimplementedKnowledgeBaseServiceServer struct {
}

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

type UploadDocRequest struct {
	Uid         string `protobuf:"bytes,1,opt,name=uid,proto3" json:"uid,omitempty"`                                    // 用户ID (用于权限检查)
	KbId        string `protobuf:"bytes,2,opt,name=kb_id,json=kbId,proto3" json:"kb_id,omitempty"`                      // 知识库ID
	DocName     string `protobuf:"bytes,3,opt,name=doc_name,json=docName,proto3" json:"doc_name,omitempty"`             // 文档名称
	FileContent []byte `protobuf:"bytes,4,opt,name=file_content,json=fileContent,proto3" json:"file_content,omitempty"` // 文件内容
	FileType    string `protobuf:"bytes,5,opt,name=file_type,json=fileType,proto3" json:"file_type,omitempty"`          // 文件类型 (txt, docx)
	// contains filtered or unexported fields
}

type UploadDocResponse struct {
	Success bool   `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"`         // 是否成功
	DocId   string `protobuf:"bytes,2,opt,name=doc_id,json=docId,proto3" json:"doc_id,omitempty"` // 文档ID
	Message string `protobuf:"bytes,3,opt,name=message,proto3" json:"message,omitempty"`          // 响应消息
	// contains filtered or unexported fields
}