Merge pull request wagtail#79 from wagtail/fix/litellm-aembed-returns…

…-iterator

LiteLLM's aembed method returns an Iterator to conform to BaseEmbeddingBackend

src/wagtail_vector_index/ai_utils/backends/litellm.py

@@ -177,9 +177,9 @@ def embed(self, inputs: Iterable[str], **kwargs) -> Iterator[list[float]]:
         assert isinstance(response, litellm.EmbeddingResponse)
         yield from [data["embedding"] for data in response["data"]]
 
-    async def aembed(self, inputs: Iterable[str], **kwargs) -> list[float]:
+    async def aembed(self, inputs: Iterable[str], **kwargs) -> Iterator[list[float]]:
         response = await litellm.aembedding(
             model=self.config.model_id, input=inputs, **kwargs
         )
 
-        return [data["embedding"] for data in response["data"]]
+        return iter([data["embedding"] for data in response["data"]])