diff --git a/specification/inference/_types/Services.ts b/specification/inference/_types/Services.ts
index 7c3ea7fc9d..110ad4a5e6 100644
--- a/specification/inference/_types/Services.ts
+++ b/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
  */
diff --git a/specification/inference/_types/TaskType.ts b/specification/inference/_types/TaskType.ts
index 7299480cbe..9eba8b1fc1 100644
--- a/specification/inference/_types/TaskType.ts
+++ b/specification/inference/_types/TaskType.ts
@@ -27,3 +27,8 @@ export enum TaskType {
   completion,
   chat_completion
 }
+
+export enum TaskTypeJinaAi {
+  text_embedding,
+  rerank
+}
diff --git a/specification/inference/put_jinaai/PutJinaAiResponse.ts b/specification/inference/put_jinaai/PutJinaAiResponse.ts
index 581909e650..a2123296af 100644
--- a/specification/inference/put_jinaai/PutJinaAiResponse.ts
+++ b/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
 }