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 ] )