diff --git a/specification/resources/gen-ai/definitions.yml b/specification/resources/gen-ai/definitions.yml index c7cee19a..b5fe820d 100644 --- a/specification/resources/gen-ai/definitions.yml +++ b/specification/resources/gen-ai/definitions.yml @@ -1602,6 +1602,8 @@ apiCreateKnowledgeBaseInputPublic: description: The datacenter region to deploy the knowledge base in. example: '"tor1"' type: string + reranking_config: + $ref: '#/apiRerankingConfiguration' tags: description: Tags to organize your knowledge base. example: @@ -3267,6 +3269,8 @@ apiKnowledgeBase: description: Region code example: example string type: string + reranking_config: + $ref: '#/apiRerankingConfiguration' tags: description: Tags to organize related resources example: @@ -4585,6 +4589,18 @@ apiReleaseStatus: - RELEASE_STATUS_BUILDING example: RELEASE_STATUS_UNKNOWN type: string +apiRerankingConfiguration: + description: Configuration for cross-encoder reranking during retrieval. + properties: + enabled: + description: Whether reranking is enabled for retrieval + example: true + type: boolean + model: + description: Reranker model internal name + example: '"bge-reranker-v2-m3"' + type: string + type: object apiResourceUsage: description: Resource Usage Description properties: @@ -5296,6 +5312,8 @@ apiUpdateKnowledgeBaseInputPublic: to example: '"12345678-1234-1234-1234-123456789012"' type: string + reranking_config: + $ref: '#/apiRerankingConfiguration' tags: description: Tags to organize your knowledge base. example: diff --git a/specification/resources/gen-ai/examples/curl/genai_create_knowledge_base.yml b/specification/resources/gen-ai/examples/curl/genai_create_knowledge_base.yml index 4a7775ec..7611b575 100644 --- a/specification/resources/gen-ai/examples/curl/genai_create_knowledge_base.yml +++ b/specification/resources/gen-ai/examples/curl/genai_create_knowledge_base.yml @@ -49,6 +49,10 @@ source: |- }, ], "region": "tor1", - "vpc_uuid": "f7176e0b-8c5e-4e32-948e-79327e56225a" + "vpc_uuid": "f7176e0b-8c5e-4e32-948e-79327e56225a", + "reranking_config": { + "enabled": true, + "model": "bge-reranker-v2-m3" + } }' diff --git a/specification/resources/gen-ai/examples/curl/genai_update_knowledge_base.yml b/specification/resources/gen-ai/examples/curl/genai_update_knowledge_base.yml index 97d8355c..aa824cfa 100644 --- a/specification/resources/gen-ai/examples/curl/genai_update_knowledge_base.yml +++ b/specification/resources/gen-ai/examples/curl/genai_update_knowledge_base.yml @@ -6,5 +6,9 @@ source: |- "https://api.digitalocean.com/v2/gen-ai/knowledge_bases/e51dba65-cf7a-11ef-bf8f-4e013e2ddde4" \ -d '{ "uuid": "e51dba65-cf7a-11ef-bf8f-4e013e2ddde4", - "name": "kb-api-rename" + "name": "kb-api-rename", + "reranking_config": { + "enabled": true, + "model": "bge-reranker-v2-m3" + } }'