From edd9107971c0f390833d3b3228bd92f7fd54efe5 Mon Sep 17 00:00:00 2001 From: michaelfeil <63565275+michaelfeil@users.noreply.github.com> Date: Mon, 9 Dec 2024 23:15:22 -0800 Subject: [PATCH] format --- libs/infinity_emb/infinity_emb/infinity_server.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/libs/infinity_emb/infinity_emb/infinity_server.py b/libs/infinity_emb/infinity_emb/infinity_server.py index 582d7fa3..36f7c6a4 100644 --- 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)