Update OpenAI embedding handling for both list and base64 embeddings
- Fix OpenAI embedding array parsing - Improve embedding data type safety
This commit is contained in:
parent
cf064579ce
commit
ecd7777e61
2 changed files with 3 additions and 1 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -70,4 +70,5 @@ gui/
|
||||||
test_*
|
test_*
|
||||||
|
|
||||||
# Cline files
|
# Cline files
|
||||||
|
memory-bank
|
||||||
memory-bank/
|
memory-bank/
|
||||||
|
|
|
||||||
|
|
@ -477,7 +477,8 @@ async def openai_embed(
|
||||||
)
|
)
|
||||||
return np.array(
|
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
|
for dp in response.data
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue