format

michaelfeil · Dec 10, 2024 · edd9107 · edd9107
1 parent c335df8
commit edd9107
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/libs/infinity_emb/infinity_emb/infinity_server.py b/libs/infinity_emb/infinity_emb/infinity_server.py
@@ -354,21 +354,27 @@ def url_to_base64(url, modality = "image"):
                     "[📝] Received request with %s input texts ",
                     len(input_),  # type: ignore
                 )
-                embedding, usage = await engine.embed(sentences=input_,matryoshka_dim=data_root.dimensions)
+                embedding, usage = await engine.embed(
+                    sentences=input_, matryoshka_dim=data_root.dimensions
+                )
             elif modality == Modality.audio:
                 urls_or_bytes = _resolve_mixed_input(data_root.input)  # type: ignore
                 logger.debug(
                     "[📝] Received request with %s input audios ",
                     len(urls_or_bytes),  # type: ignore
                 )
-                embedding, usage = await engine.audio_embed(audios=urls_or_bytes,matryoshka_dim=data_root.dimensions)
+                embedding, usage = await engine.audio_embed(
+                    audios=urls_or_bytes, matryoshka_dim=data_root.dimensions
+                )
             elif modality == Modality.image:
                 urls_or_bytes = _resolve_mixed_input(data_root.input)  # type: ignore
                 logger.debug(
                     "[📝] Received request with %s input images ",
                     len(urls_or_bytes),  # type: ignore
                 )
-                embedding, usage = await engine.image_embed(images=urls_or_bytes,matryoshka_dim=data_root.dimensions)
+                embedding, usage = await engine.image_embed(
+                    images=urls_or_bytes, matryoshka_dim=data_root.dimensions
+                )
 
             duration = (time.perf_counter() - start) * 1000
             logger.debug("[✅] Done in %s ms", duration)