Skip to content

Commit f4c9bd7

Browse files
committed
Adds custom InferenceEndpointInfo class for AlibabaCloudAi
1 parent f15ccea commit f4c9bd7

File tree

3 files changed

+23
-3
lines changed

3 files changed

+23
-3
lines changed

specification/inference/_types/Services.ts

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
import { UserDefinedValue } from '@spec_utils/UserDefinedValue'
2121
import { integer } from '@_types/Numeric'
22-
import { TaskType, TaskTypeJinaAi } from '../_types/TaskType'
22+
import { TaskType, TaskTypeJinaAi, TaskTypeAlibabaCloudAI } from '../_types/TaskType'
2323

2424
/**
2525
* Configuration options when storing the inference endpoint
@@ -68,6 +68,17 @@ export class InferenceEndpointInfoJinaAi extends InferenceEndpoint {
6868
task_type: TaskTypeJinaAi
6969
}
7070

71+
export class InferenceEndpointInfoAlibabaCloudAI extends InferenceEndpoint {
72+
/**
73+
* The inference Id
74+
*/
75+
inference_id: string
76+
/**
77+
* The task type
78+
*/
79+
task_type: TaskTypeAlibabaCloudAI
80+
}
81+
7182
/**
7283
* Chunking configuration object
7384
*/

specification/inference/_types/TaskType.ts

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,3 +32,12 @@ export enum TaskTypeJinaAi {
3232
text_embedding,
3333
rerank
3434
}
35+
36+
export enum TaskTypeAlibabaCloudAI {
37+
text_embedding,
38+
rerank,
39+
completion,
40+
sparse_embedding
41+
}
42+
43+

specification/inference/put_alibabacloud/PutAlibabaCloudResponse.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@
1717
* under the License.
1818
*/
1919

20-
import { InferenceEndpointInfo } from '@inference/_types/Services'
20+
import { InferenceEndpointInfoAlibabaCloudAI } from '@inference/_types/Services'
2121

2222
export class Response {
2323
/** @codegen_name endpoint_info */
24-
body: InferenceEndpointInfo
24+
body: InferenceEndpointInfoAlibabaCloudAI
2525
}

0 commit comments

Comments
 (0)