Adds custom InferenceEndpointInfo class for JinaAI (#4391) (#4392)

* Adds custom InferenceEndpointInfo class for JinaAI.

* Format fix.

(cherry picked from commit 6e92645)

Co-authored-by: István Zoltán Szabó <szabosteve@gmail.com>

Author: github-actions[bot]

specification/inference/_types/Services.ts

@@ -19,7 +19,7 @@
 
 import { UserDefinedValue } from '@spec_utils/UserDefinedValue'
 import { integer } from '@_types/Numeric'
-import { TaskType } from '../_types/TaskType'
+import { TaskType, TaskTypeJinaAi } from '../_types/TaskType'
 
 /**
  * Configuration options when storing the inference endpoint
@@ -57,6 +57,17 @@ export class InferenceEndpointInfo extends InferenceEndpoint {
   task_type: TaskType
 }
 
+export class InferenceEndpointInfoJinaAi extends InferenceEndpoint {
+  /**
+   * The inference Id
+   */
+  inference_id: string
+  /**
+   * The task type
+   */
+  task_type: TaskTypeJinaAi
+}
+
 /**
  * Chunking configuration object
  */

specification/inference/_types/TaskType.ts

@@ -27,3 +27,8 @@ export enum TaskType {
   completion,
   chat_completion
 }
+
+export enum TaskTypeJinaAi {
+  text_embedding,
+  rerank
+}

specification/inference/put_jinaai/PutJinaAiResponse.ts

@@ -17,9 +17,9 @@
  * under the License.
  */
 
-import { InferenceEndpointInfo } from '@inference/_types/Services'
+import { InferenceEndpointInfoJinaAi } from '@inference/_types/Services'
 
 export class Response {
   /** @codegen_name endpoint_info */
-  body: InferenceEndpointInfo
+  body: InferenceEndpointInfoJinaAi
 }