community: fix error in sambastudio embeddings (langchain-ai#26260)

fix error in samba studio embeddings result unpacking
Sheepsta300 · Oct 1, 2024 · 6833ebd · 6833ebd
1 parent 3741065
commit 6833ebd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/community/langchain_community/embeddings/sambanova.py b/libs/community/langchain_community/embeddings/sambanova.py
@@ -213,7 +213,7 @@ def embed_documents(
                     )
                 try:
                     if params.get("select_expert"):
-                        embedding = response.json()["predictions"][0]
+                        embedding = response.json()["predictions"]
                     else:
                         embedding = response.json()["predictions"]
                     embeddings.extend(embedding)
@@ -299,7 +299,7 @@ def embed_query(self, text: str) -> List[float]:
                 )
             try:
                 if params.get("select_expert"):
-                    embedding = response.json()["predictions"][0][0]
+                    embedding = response.json()["predictions"][0]
                 else:
                     embedding = response.json()["predictions"][0]
             except KeyError: