ruff

Author: mkhludnev

libs/community/langchain_community/embeddings/localai.py

@@ -299,12 +299,16 @@ async def _aembedding_func(self, text: str, *, engine: str) -> List[float]:
             # replace newlines, which can negatively affect performance.
             text = text.replace("\n", " ")
         return (
-            await async_embed_with_retry(
-                self,
-                input=[text],
-                **self._invocation_params,
+            (
+                await async_embed_with_retry(
+                    self,
+                    input=[text],
+                    **self._invocation_params,
+                )
             )
-        ).data[0].embedding
+            .data[0]
+            .embedding
+        )
 
     def embed_documents(
         self, texts: List[str], chunk_size: Optional[int] = 0

libs/community/tests/integration_tests/embeddings/test_localai.py

@@ -1,4 +1,3 @@
-
 import pytest
 
 from langchain_community.embeddings import LocalAIEmbeddings
@@ -7,20 +6,20 @@
 @pytest.mark.requires("openai")
 @pytest.mark.vcr
 def test_localai_embed() -> None:
-   llm=LocalAIEmbeddings(
-                         openai_api_key="foo", model="bge-m3",
-                        openai_api_base="https://foo.bar/v1")
-   eqq=llm.embed_query("foo bar")
-   assert len(eqq) > 100
-   assert eqq[0] != eqq[1]
+    llm = LocalAIEmbeddings(
+        openai_api_key="foo", model="bge-m3", openai_api_base="https://foo.bar/v1"
+    )
+    eqq = llm.embed_query("foo bar")
+    assert len(eqq) > 100
+    assert eqq[0] != eqq[1]
 
 
 @pytest.mark.requires("openai")
 @pytest.mark.vcr
 async def test_localai_aembed() -> None:
-   llm=LocalAIEmbeddings(
-                         openai_api_key="foo", model="bge-m3",
-                        openai_api_base="https://foo.bar/v1")
-   eqq=await llm.aembed_query("foo bar")
-   assert len(eqq) > 100
-   assert eqq[0] != eqq[1]
+    llm = LocalAIEmbeddings(
+        openai_api_key="foo", model="bge-m3", openai_api_base="https://foo.bar/v1"
+    )
+    eqq = await llm.aembed_query("foo bar")
+    assert len(eqq) > 100
+    assert eqq[0] != eqq[1]