dh SSKJrJrJrJrJr SSKrSSKJr SSK J r SSK J r SSK Jr "SS5r"S S \ 5rg) )AnyDictListMappingOptionalN)CallbackManagerForLLMRun)LLM) ConfigDict)enforce_stop_tokensc h\rSrSrSr\S\S\\\4S\\\44Sj5r \S\S\4Sj5r S r g ) ContentHandlerAmazonAPIGateway zAdapter to prepare the inputs from Langchain to a format that LLM model expects. It also provides helper function to extract the generated text from the model response.prompt model_kwargsreturnc XS.$)N)inputs parameters)clsrrs c/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/llms/amazon_api_gateway.pytransform_input.ContentHandlerAmazonAPIGateway.transform_inputs !==responsec.UR5SS$)Nrgenerated_text)json)rrs rtransform_output/ContentHandlerAmazonAPIGateway.transform_outputs}}q!"233rrN) __name__ __module__ __qualname____firstlineno____doc__ classmethodstrrrrr__static_attributes__rrrr r se3 >>(,S#X> c3h>> 4444rr c \rSrSr%Sr\\S'Sr\\ \S'Sr \\ \S'\ "5r \ \S'\ "SS 9r\S \\\44S j5r\S \4S j5rSS \S\\\S\\S\S \4 SjjrSrg)AmazonAPIGatewayz6Amazon API Gateway to access LLM models hosted on AWS.api_urlNheadersrcontent_handlerforbid)extrarcjUR=(d 0n0URURS.ESU0E$)zGet the identifying parameters.)r,r-r)rr,r-)self _model_kwargss r_identifying_params$AmazonAPIGateway._identifying_params3s@))/R  ,,4<<@ }-  rcg)zReturn type of llm.amazon_api_gatewayr)r2s r _llm_typeAmazonAPIGateway._llm_type<s$rrstop run_managerkwargsc NUR=(d 0nURRX5n[R"UR UR US9nURRU5nUb [X5nU$![an [SU 35eSn A ff=f)aCall out to Amazon API Gateway model. Args: prompt: The prompt to pass into the model. stop: Optional list of stop words to use when generating. Returns: The string generated by the model. Example: .. code-block:: python response = se("Tell me a joke.") )r-rzError raised by the service: N) rr.rrequestspostr,r-r Exception ValueErrorr ) r2rr:r;r<r3payloadrtexterrors r_callAmazonAPIGateway._callAs*))/R &&66vM F}}  H ''88BD  &t2D  F &&tCy!&67 &  & &&rr*)typingrrrrrr>langchain_core.callbacksr#langchain_core.language_models.llmsr pydanticr langchain_community.llms.utilsr r r*rrrrOs255=3>44$JsJr