update embedding strategy

microsoft · Oct 25, 2024 · a020472 · a020472
1 parent 4ef83a3
commit a020472
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphrag/index/operations/embed_text/strategies/openai.py b/graphrag/index/operations/embed_text/strategies/openai.py
@@ -95,7 +95,7 @@ async def _execute(
     async def embed(chunk: list[str]):
         async with semaphore:
             chunk_embeddings = await llm(chunk)
-            result = np.array(chunk_embeddings.output)
+            result = np.array(chunk_embeddings.output.embeddings)
             tick(1)
         return result