|
|
@ -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:
|
|
|
|