admin/basicmachines-co-basic-memory
1
0
Fork 0
mirror of https://github.com/basicmachines-co/basic-memory synced 2026-06-21 13:47:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(core): handle dict LiteLLM embedding items

Browse Source 
Signed-off-by: phernandez <paul@basicmachines.co>
This commit is contained in:
phernandez
2026-06-06 12:50:13 -05:00
parent 750c7c9920
commit bfb63724d4
2 changed files with 25 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/basic_memory/repository/litellm_provider.py
+7 -2
View File
@@ -155,12 +155,17 @@ class LiteLLMEmbeddingProvider(EmbeddingProvider):
vectors_by_index: dict[int, list[float]] = {}
for item in response.data:
response_index = int(item.index)
if isinstance(item, dict):
response_index = int(item["index"])
embedding = item["embedding"]
else:
response_index = int(item.index)
embedding = item.embedding
if response_index in vectors_by_index:
raise RuntimeError(
"LiteLLM embedding response returned duplicate vector indexes."
)
vectors_by_index[response_index] = [float(v) for v in item.embedding]
vectors_by_index[response_index] = [float(v) for v in embedding]
ordered_vectors: list[list[float]] = []
for index in range(len(batch)):