File tree Expand file tree Collapse file tree 1 file changed +17
-18
lines changed
libs/aws/langchain_aws/llms Expand file tree Collapse file tree 1 file changed +17
-18
lines changed Original file line number Diff line number Diff line change @@ -847,24 +847,23 @@ async def _aprepare_input_and_invoke_stream(
847
847
_model_kwargs ["stream" ] = True
848
848
849
849
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
+ )
868
867
body = json .dumps (input_body )
869
868
870
869
response = await asyncio .get_running_loop ().run_in_executor (
You can’t perform that action at this time.
0 commit comments