Skip to content

Commit

Permalink
done
Browse files Browse the repository at this point in the history
  • Loading branch information
Yougigun committed Jun 20, 2024
1 parent 60b2bd3 commit 1232230
Show file tree
Hide file tree
Showing 2 changed files with 45 additions and 54 deletions.
67 changes: 29 additions & 38 deletions artifact/artifact/v1alpha/artifact.proto
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@ message CreateRepositoryTagResponse {
// KnowledgeBase represents a knowledge base.
message KnowledgeBase {
// The knowledge base identifier.
string id = 1;
string kb_id = 1;
// The knowledge base name.
string name = 2;
// The knowledge base description.
Expand All @@ -152,86 +152,77 @@ message KnowledgeBase {
string owner_name = 7;
// The knowledge base tags.
repeated string tags = 8;
// The knowledge base converting pipelines.
repeated string converting_pipelines = 9;
// The knowledge base splitting pipelines.
repeated string splitting_pipelines = 10;
// The knowledge base embedding pipelines.
repeated string embedding_pipelines = 11;
// The downstream apps
repeated string downstream_apps = 12;
}

// KnowledgeBasesList represents a list of knowledge bases.
message KnowledgeBasesList {
// The list of knowledge bases.
repeated KnowledgeBase knowledge_bases = 1;
}

// CreateKnowledgeBaseRequest represents a request to create a knowledge base.
message CreateKnowledgeBaseRequest {
// The knowledge base owner(nammespace).
string owner_id = 1;
// The knowledge base name.
string name = 1;
string name = 2;
// The knowledge base description.
string description = 2;
string description = 3;
// The knowledge base tags.
repeated string tags = 3;
repeated string tags = 4;
}

// CreateKnowledgeBaseResponse represents a response for creating a knowledge base.
message CreateKnowledgeBaseResponse {
// The created knowledge base.
KnowledgeBase body = 1;
// The error message.
string error_msg = 2;
// The status code.
int32 status_code = 3;
KnowledgeBase knowledge_base = 1;
}

// Request message for ListKnowledgeBases
message ListKnowledgeBasesRequest {
// User ID for which to list the knowledge bases
string uid = 1;
string owner_id = 1;
}

// GetKnowledgeBasesResponse represents a response for getting all knowledge bases from users.
message ListKnowledgeBasesResponse {
// The knowledge bases container.
KnowledgeBasesList body = 1;
// The error message.
string error_msg = 2;
// The status code.
int32 status_code = 3;
repeated KnowledgeBase knowledge_bases = 1;
}

// UpdateKnowledgeBaseRequest represents a request to update a knowledge base.
message UpdateKnowledgeBaseRequest {
// The knowledge base identifier.
string id = 1;
// The knowledge base name.
string name = 2;
// The knowledge base id.
string kb_id = 1;
// The knowledge base description.
string description = 3;
string description = 2;
// The knowledge base tags.
repeated string tags = 4;
// The knowledge base owner.
string owner_name = 5;
repeated string tags = 3;
// The knowledge base owner(namespace).
string owner_id = 4;
}

// UpdateKnowledgeBaseResponse represents a response for updating a knowledge base.
message UpdateKnowledgeBaseResponse {
// The updated knowledge base.
KnowledgeBase body = 1;
// The error message.
string error_msg = 2;
// The status code.
int32 status_code = 3;
KnowledgeBase knowledge_base = 1;
}

// DeleteKnowledgeBaseRequest represents a request to delete a knowledge base.
message DeleteKnowledgeBaseRequest {
// The owner's id. i.e. namespace.
string owner_id = 1;
// The knowledge base identifier.
string id = 1;
string kb_id = 2;
}

// DeleteKnowledgeBaseResponse represents a response for deleting a knowledge base.
message DeleteKnowledgeBaseResponse {
// The error message.
string error_msg = 1;
// The status code.
int32 status_code = 2;
// The knowledge base identifier.
KnowledgeBase knowledge_base = 1;
}

// file embedding process status
Expand Down
32 changes: 16 additions & 16 deletions artifact/artifact/v1alpha/artifact_public_service.proto
Original file line number Diff line number Diff line change
Expand Up @@ -40,33 +40,33 @@ service ArtifactPublicService {
option (google.api.method_visibility).restriction = "INTERNAL";
}

// Create a knowledge base
// Create a knowledge base
rpc CreateKnowledgeBase(CreateKnowledgeBaseRequest) returns (CreateKnowledgeBaseResponse) {
option (google.api.http) = {
post: "/v1alpha/knowledge-base"
body: "*"
};
option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {tags: "KnowledgeBase"};
}

// Get all knowledge bases info
option (google.api.http) = {
post: "/v1alpha/owners/{owner_id}/knowledge-bases"
body: "*"
};
option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {tags: "KnowledgeBase"};
}
// Get all knowledge bases info
rpc ListKnowledgeBases(ListKnowledgeBasesRequest) returns (ListKnowledgeBasesResponse) {
option (google.api.http) = {get: "/v1alpha/users/{uid}/knowledge-base"};
option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {tags: "KnowledgeBase"};
}

option (google.api.http) = {get: "/v1alpha/owners/{owner_id}/knowledge-bases"};
option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {tags: "KnowledgeBase"};
}
// Update a knowledge base info
rpc UpdateKnowledgeBase(UpdateKnowledgeBaseRequest) returns (UpdateKnowledgeBaseResponse) {
option (google.api.http) = {
put: "/v1alpha/knowledge-base/{id}"
put: "/v1alpha/owners/{owner_id}/knowledge-bases/{kb_id}"
body: "*"
};
option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {tags: "KnowledgeBase"};
}

// Delete a knowledge base
rpc DeleteKnowledgeBase(DeleteKnowledgeBaseRequest) returns (DeleteKnowledgeBaseResponse) {
option (google.api.http) = {delete: "/v1alpha/knowledge-base/{id}"};
option (google.api.http) = {delete: "/v1alpha/owners/{owner_id}/knowledge-bases/{kb_id}"};
option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {tags: "KnowledgeBase"};
}

Expand Down

0 comments on commit 1232230

Please sign in to comment.