Skip to content

Commit ebb6559

Browse files
Update Inference specification for Watsonx's completion and chat completion tasks
1 parent 93cbe0b commit ebb6559

File tree

7 files changed

+45
-21
lines changed

7 files changed

+45
-21
lines changed

output/openapi/elasticsearch-openapi.json

Lines changed: 8 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

output/openapi/elasticsearch-serverless-openapi.json

Lines changed: 8 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

output/schema/schema.json

Lines changed: 19 additions & 7 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

output/typescript/types.ts

Lines changed: 2 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

specification/inference/_types/CommonTypes.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1184,6 +1184,7 @@ export class WatsonxServiceSettings {
11841184
/**
11851185
* The name of the model to use for the inference task.
11861186
* Refer to the IBM Embedding Models section in the Watsonx documentation for the list of available text embedding models.
1187+
* Refer to the IBM library - Foundation models in Watsonx.ai.
11871188
* @ext_doc_id watsonx-api-models
11881189
*/
11891190
model_id: string
@@ -1203,7 +1204,9 @@ export class WatsonxServiceSettings {
12031204
}
12041205

12051206
export enum WatsonxTaskType {
1206-
text_embedding
1207+
text_embedding,
1208+
chat_completion,
1209+
completion
12071210
}
12081211

12091212
export enum WatsonxServiceType {

specification/inference/_types/TaskType.ts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,5 +109,7 @@ export enum TaskTypeVoyageAI {
109109
}
110110

111111
export enum TaskTypeWatsonx {
112-
text_embedding
112+
text_embedding,
113+
chat_completion,
114+
completion
113115
}

specification/inference/put_watsonx/PutWatsonxRequest.ts

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,7 @@ export interface Request extends RequestBase {
4646
]
4747
path_parts: {
4848
/**
49-
* The task type.
50-
* The only valid task type for the model to perform is `text_embedding`.
49+
* The type of the inference task that the model will perform.
5150
*/
5251
task_type: WatsonxTaskType
5352
/**

0 commit comments

Comments
 (0)