Fix new embed function return data format (to dictionary) #48

Open-Multi-Modal-Personal-Assistant · Oct 20, 2024 · de4618f · de4618f
1 parent da371c0
commit de4618f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/functions/fn_impl/embed.py b/functions/fn_impl/embed.py
@@ -108,9 +108,9 @@ def embed(req: https_fn.Request) -> https_fn.Response:
             )
 
             if multi_modal_embeddings:
-                embeddings.image = multi_modal_embeddings.image_embedding
+                embeddings['image'] = multi_modal_embeddings.image_embedding
                 if video and multi_modal_embeddings.video_embeddings:
-                    embeddings.video.extend(
+                    embeddings['video'].extend(
                         [embedding.embedding for embedding in multi_modal_embeddings.video_embeddings]
                     )
         except Exception as e: