From ecd7777e613ee2a4aa67a8e1e3eb150eca3e00fa Mon Sep 17 00:00:00 2001 From: yangdx Date: Sat, 9 Aug 2025 08:40:33 +0800 Subject: [PATCH] Update OpenAI embedding handling for both list and base64 embeddings - Fix OpenAI embedding array parsing - Improve embedding data type safety --- .gitignore | 1 + lightrag/llm/openai.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.gitignore b/.gitignore index dc48605d..2773b704 100644 --- a/.gitignore +++ b/.gitignore @@ -70,4 +70,5 @@ gui/ test_* # Cline files +memory-bank memory-bank/ diff --git a/lightrag/llm/openai.py b/lightrag/llm/openai.py index 5a52e767..f7895978 100644 --- a/lightrag/llm/openai.py +++ b/lightrag/llm/openai.py @@ -477,7 +477,8 @@ async def openai_embed( ) return np.array( [ - np.frombuffer(base64.b64decode(dp.embedding), dtype=np.float32) + np.array(dp.embedding, dtype=np.float32) if isinstance(dp.embedding, list) + else np.frombuffer(base64.b64decode(dp.embedding), dtype=np.float32) for dp in response.data ] )