h ddlmZmZmZmZmZddlZddlmZddl m Z ddl m Z ddl mZGddZGd d e Zy) )AnyDictListMappingOptionalN)CallbackManagerForLLMRun)LLM) ConfigDict)enforce_stop_tokensc \eZdZdZededeeefdeeeffdZededefdZ y) ContentHandlerAmazonAPIGatewayzAdapter 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 ||dS)N)inputs parameters)clsrrs i/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/langchain_community/llms/amazon_api_gateway.pytransform_inputz.ContentHandlerAmazonAPIGateway.transform_inputs! ==responsec.|jddS)Nrgenerated_text)json)rrs rtransform_outputz/ContentHandlerAmazonAPIGateway.transform_outputs}}q!"233rN) __name__ __module__ __qualname____doc__ classmethodstrrrrrrrrr r se3 >>(,S#X> c3h>> 4444rr c eZdZUdZeed< dZeeed< dZ eeed< e Z e ed< e dZ ed eeeffd Zed efd Z dd ed eeedeeded ef dZy)AmazonAPIGatewayz6Amazon API Gateway to access LLM models hosted on AWS.api_urlNheadersrcontent_handlerforbid)extrarc`|jxsi}i|j|jdd|iS)zGet the identifying parameters.)r&r'r)rr&r')self _model_kwargss r_identifying_paramsz$AmazonAPIGateway._identifying_params3s@))/R  ,,4<<@ }-  rcy)zReturn type of llm.amazon_api_gatewayr)r,s r _llm_typezAmazonAPIGateway._llm_type<s$rrstop run_managerkwargsc H|jxsi}|jj||} tj|j |j |}|jj|}| t||}|S#t$r} td| d} ~ wwxYw)abCall 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 ) r,rr2r3r4r-payloadrtexterrors r_callzAmazonAPIGateway._callAs*))/R &&66v}M F}}  H ''88BD  &tT2D  F &&tCy!&67 &  & &rr%)typingrrrrrr6langchain_core.callbacksr#langchain_core.language_models.llmsr pydanticr langchain_community.llms.utilsr r r%rrrrFs255=3>44$JsJr