From 80c9fae0ffc849d04be22c57786984eeac0cfc85 Mon Sep 17 00:00:00 2001
From: Zac Li <zac.li@jina.ai>
Date: Thu, 26 Sep 2024 12:40:53 +0800
Subject: [PATCH] fix: task-type bug

---
 jina_sagemaker/client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/jina_sagemaker/client.py b/jina_sagemaker/client.py
index 65bbfca..d45643d 100644
--- a/jina_sagemaker/client.py
+++ b/jina_sagemaker/client.py
@@ -288,7 +288,7 @@ def embed(
 
                 if 'jina-embeddings-v3' in self._arn:
                     data["parameters"] = {
-                        "task": task_type.value,
+                        "task": task_type.value if task_type else None,
                         "dimensions": dimensions,
                         "late_chunking": late_chunking,
                     }