Fixes ChatBedrock for llama3 (#115)

langchain-ai · Jul 18, 2024 · 8a17693 · 8a17693
1 parent 19822bc
commit 8a17693
Showing 1 changed file with 17 additions and 18 deletions.
diff --git a/libs/aws/langchain_aws/llms/bedrock.py b/libs/aws/langchain_aws/llms/bedrock.py
@@ -847,24 +847,23 @@ async def _aprepare_input_and_invoke_stream(
             _model_kwargs["stream"] = True
 
         params = {**_model_kwargs, **kwargs}
-        if "claude-3" in self._get_model():
-            if _tools_in_params(params):
-                input_body = LLMInputOutputAdapter.prepare_input(
-                    provider=provider,
-                    model_kwargs=params,
-                    prompt=prompt,
-                    system=system,
-                    messages=messages,
-                    tools=params["tools"],
-                )
-            else:
-                input_body = LLMInputOutputAdapter.prepare_input(
-                    provider=provider,
-                    prompt=prompt,
-                    system=system,
-                    messages=messages,
-                    model_kwargs=params,
-                )
+        if "claude-3" in self._get_model() and _tools_in_params(params):
+            input_body = LLMInputOutputAdapter.prepare_input(
+                provider=provider,
+                model_kwargs=params,
+                prompt=prompt,
+                system=system,
+                messages=messages,
+                tools=params["tools"],
+            )
+        else:
+            input_body = LLMInputOutputAdapter.prepare_input(
+                provider=provider,
+                prompt=prompt,
+                system=system,
+                messages=messages,
+                model_kwargs=params,
+            )
         body = json.dumps(input_body)
 
         response = await asyncio.get_running_loop().run_in_executor(