ContentHandlerBase#
- class langchain_aws.llms.sagemaker_endpoint.ContentHandlerBase[source]#
A handler class to transform input from LLM to a format that SageMaker endpoint expects.
Similarly, the class handles transforming output from the SageMaker endpoint to a format that LLM class expects.
Attributes
accepts
The MIME type of the response data returned from endpoint
content_type
The MIME type of the input data passed to endpoint
Methods
__init__
()transform_input
(prompt,Β model_kwargs)Transforms the input to a format that model can accept as the request Body.
transform_output
(output)Transforms the output from the model to string that the LLM class expects.
- __init__()#
- abstract transform_input(prompt: INPUT_TYPE, model_kwargs: Dict) bytes [source]#
Transforms the input to a format that model can accept as the request Body. Should return bytes or seekable file like object in the format specified in the content_type request header.
- Parameters:
prompt (INPUT_TYPE) β
model_kwargs (Dict) β
- Return type:
bytes
Examples using ContentHandlerBase