community: fix error in sambastudio embeddings (#26260)

fix error in samba studio embeddings result unpacking
langchain-ai · Sep 19, 2024 · 55b641b · 55b641b
1 parent 37b7202
commit 55b641b
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: