Skip to content
Snippets Groups Projects
Unverified Commit b76a1360 authored by Nicolò Boschi's avatar Nicolò Boschi Committed by GitHub
Browse files

fix: BedrockEmbeddings.from_credentials always uses the default model (#10640)

parent 327373c1
No related branches found
No related tags found
No related merge requests found
...@@ -311,7 +311,7 @@ class BedrockEmbedding(BaseEmbedding): ...@@ -311,7 +311,7 @@ class BedrockEmbedding(BaseEmbedding):
client = session.client("bedrock") client = session.client("bedrock")
return cls( return cls(
client=client, client=client,
model_name=model_name, model=model_name,
embed_batch_size=embed_batch_size, embed_batch_size=embed_batch_size,
callback_manager=callback_manager, callback_manager=callback_manager,
verbose=verbose, verbose=verbose,
......
...@@ -312,7 +312,7 @@ class BedrockEmbedding(BaseEmbedding): ...@@ -312,7 +312,7 @@ class BedrockEmbedding(BaseEmbedding):
client = session.client("bedrock") client = session.client("bedrock")
return cls( return cls(
client=client, client=client,
model_name=model_name, model=model_name,
embed_batch_size=embed_batch_size, embed_batch_size=embed_batch_size,
callback_manager=callback_manager, callback_manager=callback_manager,
verbose=verbose, verbose=verbose,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment