Skip to content

Commit fe2f14c

Browse files
Update Inference specification for Watsonx's completion and chat completion tasks (#4505)
1 parent 2eeacf8 commit fe2f14c

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
@@ -1255,6 +1255,7 @@ export class WatsonxServiceSettings {
12551255
/**
12561256
* The name of the model to use for the inference task.
12571257
* Refer to the IBM Embedding Models section in the Watsonx documentation for the list of available text embedding models.
1258+
* Refer to the IBM library - Foundation models in Watsonx.ai.
12581259
* @ext_doc_id watsonx-api-models
12591260
*/
12601261
model_id: string
@@ -1274,7 +1275,9 @@ export class WatsonxServiceSettings {
12741275
}
12751276

12761277
export enum WatsonxTaskType {
1277-
text_embedding
1278+
text_embedding,
1279+
chat_completion,
1280+
completion
12781281
}
12791282

12801283
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)