community: fix error in sambastudio embeddings (#26260)

fix error in samba studio embeddings  result unpacking
pull/20092/merge
Jorge Piedrahita Ortiz 3 days ago committed by GitHub
parent 37b72023fe
commit 55b641b761
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -213,7 +213,7 @@ class SambaStudioEmbeddings(BaseModel, Embeddings):
) )
try: try:
if params.get("select_expert"): if params.get("select_expert"):
embedding = response.json()["predictions"][0] embedding = response.json()["predictions"]
else: else:
embedding = response.json()["predictions"] embedding = response.json()["predictions"]
embeddings.extend(embedding) embeddings.extend(embedding)
@ -299,7 +299,7 @@ class SambaStudioEmbeddings(BaseModel, Embeddings):
) )
try: try:
if params.get("select_expert"): if params.get("select_expert"):
embedding = response.json()["predictions"][0][0] embedding = response.json()["predictions"][0]
else: else:
embedding = response.json()["predictions"][0] embedding = response.json()["predictions"][0]
except KeyError: except KeyError:

Loading…
Cancel
Save