Skip to content

Commit 8a17693

Browse files
authored
Fixes ChatBedrock for llama3 (#115)
1 parent 19822bc commit 8a17693

File tree

1 file changed

+17
-18
lines changed

1 file changed

+17
-18
lines changed

libs/aws/langchain_aws/llms/bedrock.py

Lines changed: 17 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -847,24 +847,23 @@ async def _aprepare_input_and_invoke_stream(
847847
_model_kwargs["stream"] = True
848848

849849
params = {**_model_kwargs, **kwargs}
850-
if "claude-3" in self._get_model():
851-
if _tools_in_params(params):
852-
input_body = LLMInputOutputAdapter.prepare_input(
853-
provider=provider,
854-
model_kwargs=params,
855-
prompt=prompt,
856-
system=system,
857-
messages=messages,
858-
tools=params["tools"],
859-
)
860-
else:
861-
input_body = LLMInputOutputAdapter.prepare_input(
862-
provider=provider,
863-
prompt=prompt,
864-
system=system,
865-
messages=messages,
866-
model_kwargs=params,
867-
)
850+
if "claude-3" in self._get_model() and _tools_in_params(params):
851+
input_body = LLMInputOutputAdapter.prepare_input(
852+
provider=provider,
853+
model_kwargs=params,
854+
prompt=prompt,
855+
system=system,
856+
messages=messages,
857+
tools=params["tools"],
858+
)
859+
else:
860+
input_body = LLMInputOutputAdapter.prepare_input(
861+
provider=provider,
862+
prompt=prompt,
863+
system=system,
864+
messages=messages,
865+
model_kwargs=params,
866+
)
868867
body = json.dumps(input_body)
869868

870869
response = await asyncio.get_running_loop().run_in_executor(

0 commit comments

Comments
 (0)