diff --git a/specification/inference/_types/Services.ts b/specification/inference/_types/Services.ts index 05beaa6ac0..aa69b2bb22 100644 --- a/specification/inference/_types/Services.ts +++ b/specification/inference/_types/Services.ts @@ -22,7 +22,21 @@ import { UserDefinedValue } from '@spec_utils/UserDefinedValue' import { TaskType, TaskTypeAlibabaCloudAI, - TaskTypeJinaAi + TaskTypeAmazonBedrock, + TaskTypeAnthropic, + TaskTypeAzureAIStudio, + TaskTypeAzureOpenAI, + TaskTypeCohere, + TaskTypeElasticsearch, + TaskTypeELSER, + TaskTypeGoogleAIStudio, + TaskTypeGoogleVertexAI, + TaskTypeHuggingFace, + TaskTypeJinaAi, + TaskTypeMistral, + TaskTypeOpenAI, + TaskTypeVoyageAI, + TaskTypeWatsonx } from '../_types/TaskType' /** @@ -83,6 +97,160 @@ export class InferenceEndpointInfoAlibabaCloudAI extends InferenceEndpoint { task_type: TaskTypeAlibabaCloudAI } +export class InferenceEndpointInfoAmazonBedrock extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeAmazonBedrock +} + +export class InferenceEndpointInfoAnthropic extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeAnthropic +} + +export class InferenceEndpointInfoAzureAIStudio extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeAzureAIStudio +} + +export class InferenceEndpointInfoAzureOpenAI extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeAzureOpenAI +} + +export class InferenceEndpointInfoCohere extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeCohere +} + +export class InferenceEndpointInfoElasticsearch extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeElasticsearch +} + +export class InferenceEndpointInfoELSER extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeELSER +} + +export class InferenceEndpointInfoGoogleAIStudio extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeGoogleAIStudio +} + +export class InferenceEndpointInfoGoogleVertexAI extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeGoogleVertexAI +} + +export class InferenceEndpointInfoHuggingFace extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeHuggingFace +} + +export class InferenceEndpointInfoMistral extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeMistral +} + +export class InferenceEndpointInfoOpenAI extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeOpenAI +} + +export class InferenceEndpointInfoVoyageAI extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeVoyageAI +} + +export class InferenceEndpointInfoWatsonx extends InferenceEndpoint { + /** + * The inference Id + */ + inference_id: string + /** + * The task type + */ + task_type: TaskTypeWatsonx +} + /** * Chunking configuration object */ diff --git a/specification/inference/_types/TaskType.ts b/specification/inference/_types/TaskType.ts index 5025ef40ba..7c36323488 100644 --- a/specification/inference/_types/TaskType.ts +++ b/specification/inference/_types/TaskType.ts @@ -39,3 +39,71 @@ export enum TaskTypeAlibabaCloudAI { completion, sparse_embedding } + +export enum TaskTypeAmazonBedrock { + text_embedding, + completion +} + +export enum TaskTypeAnthropic { + completion +} + +export enum TaskTypeAzureAIStudio { + text_embedding, + completion +} + +export enum TaskTypeAzureOpenAI { + text_embedding, + completion +} + +export enum TaskTypeCohere { + text_embedding, + rerank, + completion +} + +export enum TaskTypeElasticsearch { + sparse_embedding, + text_embedding, + rerank +} + +export enum TaskTypeELSER { + sparse_embedding +} + +export enum TaskTypeGoogleAIStudio { + text_embedding, + completion +} + +export enum TaskTypeGoogleVertexAI { + text_embedding, + rerank +} + +export enum TaskTypeHuggingFace { + text_embedding +} + +export enum TaskTypeMistral { + text_embedding +} + +export enum TaskTypeOpenAI { + text_embedding, + chat_completion, + completion +} + +export enum TaskTypeVoyageAI { + text_embedding, + rerank +} + +export enum TaskTypeWatsonx { + text_embedding +} diff --git a/specification/inference/put_amazonbedrock/PutAmazonBedrockResponse.ts b/specification/inference/put_amazonbedrock/PutAmazonBedrockResponse.ts index d40639b031..499423095e 100644 --- a/specification/inference/put_amazonbedrock/PutAmazonBedrockResponse.ts +++ b/specification/inference/put_amazonbedrock/PutAmazonBedrockResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoAmazonBedrock } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoAmazonBedrock } diff --git a/specification/inference/put_anthropic/PutAnthropicResponse.ts b/specification/inference/put_anthropic/PutAnthropicResponse.ts index d40639b031..27ed424609 100644 --- a/specification/inference/put_anthropic/PutAnthropicResponse.ts +++ b/specification/inference/put_anthropic/PutAnthropicResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoAnthropic } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoAnthropic } diff --git a/specification/inference/put_azureaistudio/PutAzureAiStudioResponse.ts b/specification/inference/put_azureaistudio/PutAzureAiStudioResponse.ts index d40639b031..bfd651ed98 100644 --- a/specification/inference/put_azureaistudio/PutAzureAiStudioResponse.ts +++ b/specification/inference/put_azureaistudio/PutAzureAiStudioResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoAzureAIStudio } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoAzureAIStudio } diff --git a/specification/inference/put_azureopenai/PutAzureOpenAiResponse.ts b/specification/inference/put_azureopenai/PutAzureOpenAiResponse.ts index d40639b031..ebc2a4edff 100644 --- a/specification/inference/put_azureopenai/PutAzureOpenAiResponse.ts +++ b/specification/inference/put_azureopenai/PutAzureOpenAiResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoAzureOpenAI } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoAzureOpenAI } diff --git a/specification/inference/put_cohere/PutCohereResponse.ts b/specification/inference/put_cohere/PutCohereResponse.ts index d40639b031..2502a12c50 100644 --- a/specification/inference/put_cohere/PutCohereResponse.ts +++ b/specification/inference/put_cohere/PutCohereResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoCohere } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoCohere } diff --git a/specification/inference/put_elasticsearch/PutElasticsearchResponse.ts b/specification/inference/put_elasticsearch/PutElasticsearchResponse.ts index d40639b031..b2228afeb7 100644 --- a/specification/inference/put_elasticsearch/PutElasticsearchResponse.ts +++ b/specification/inference/put_elasticsearch/PutElasticsearchResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoElasticsearch } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoElasticsearch } diff --git a/specification/inference/put_elser/PutElserResponse.ts b/specification/inference/put_elser/PutElserResponse.ts index d40639b031..cce8046af5 100644 --- a/specification/inference/put_elser/PutElserResponse.ts +++ b/specification/inference/put_elser/PutElserResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoELSER } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoELSER } diff --git a/specification/inference/put_googleaistudio/PutGoogleAiStudioResponse.ts b/specification/inference/put_googleaistudio/PutGoogleAiStudioResponse.ts index d40639b031..c874a9c049 100644 --- a/specification/inference/put_googleaistudio/PutGoogleAiStudioResponse.ts +++ b/specification/inference/put_googleaistudio/PutGoogleAiStudioResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoGoogleAIStudio } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoGoogleAIStudio } diff --git a/specification/inference/put_googlevertexai/PutGoogleVertexAiResponse.ts b/specification/inference/put_googlevertexai/PutGoogleVertexAiResponse.ts index d40639b031..8fd4aec5a2 100644 --- a/specification/inference/put_googlevertexai/PutGoogleVertexAiResponse.ts +++ b/specification/inference/put_googlevertexai/PutGoogleVertexAiResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoGoogleVertexAI } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoGoogleVertexAI } diff --git a/specification/inference/put_hugging_face/PutHuggingFaceResponse.ts b/specification/inference/put_hugging_face/PutHuggingFaceResponse.ts index d40639b031..a8371d3445 100644 --- a/specification/inference/put_hugging_face/PutHuggingFaceResponse.ts +++ b/specification/inference/put_hugging_face/PutHuggingFaceResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoHuggingFace } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoHuggingFace } diff --git a/specification/inference/put_mistral/PutMistralResponse.ts b/specification/inference/put_mistral/PutMistralResponse.ts index d40639b031..96bdf0346c 100644 --- a/specification/inference/put_mistral/PutMistralResponse.ts +++ b/specification/inference/put_mistral/PutMistralResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoMistral } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoMistral } diff --git a/specification/inference/put_openai/PutOpenAiResponse.ts b/specification/inference/put_openai/PutOpenAiResponse.ts index d40639b031..12f31f200c 100644 --- a/specification/inference/put_openai/PutOpenAiResponse.ts +++ b/specification/inference/put_openai/PutOpenAiResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoOpenAI } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoOpenAI } diff --git a/specification/inference/put_voyageai/PutVoyageAIResponse.ts b/specification/inference/put_voyageai/PutVoyageAIResponse.ts index 581909e650..6f9d929762 100644 --- a/specification/inference/put_voyageai/PutVoyageAIResponse.ts +++ b/specification/inference/put_voyageai/PutVoyageAIResponse.ts @@ -17,9 +17,9 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoVoyageAI } from '@inference/_types/Services' export class Response { /** @codegen_name endpoint_info */ - body: InferenceEndpointInfo + body: InferenceEndpointInfoVoyageAI } diff --git a/specification/inference/put_watsonx/PutWatsonxResponse.ts b/specification/inference/put_watsonx/PutWatsonxResponse.ts index d40639b031..6dca948096 100644 --- a/specification/inference/put_watsonx/PutWatsonxResponse.ts +++ b/specification/inference/put_watsonx/PutWatsonxResponse.ts @@ -17,8 +17,8 @@ * under the License. */ -import { InferenceEndpointInfo } from '@inference/_types/Services' +import { InferenceEndpointInfoWatsonx } from '@inference/_types/Services' export class Response { - body: InferenceEndpointInfo + body: InferenceEndpointInfoWatsonx }