Method convert_inputs_to_api_kwargs

api/bedrock_client.py:442–473

Convert inputs to API kwargs for AWS Bedrock.

(
        self, input: Any = None, model_kwargs: Dict = None, model_type: ModelType = None
    )

        return self.call(api_kwargs, model_type)

    def convert_inputs_to_api_kwargs(
        self, input: Any = None, model_kwargs: Dict = None, model_type: ModelType = None
    ) -> Dict:
        """Convert inputs to API kwargs for AWS Bedrock."""
        model_kwargs = model_kwargs or {}
        api_kwargs = {}
        
        if model_type == ModelType.LLM:
            api_kwargs["model"] = model_kwargs.get("model", "anthropic.claude-3-sonnet-20240229-v1:0")
            api_kwargs["input"] = input
            
            # Add model parameters
            if "temperature" in model_kwargs:
                api_kwargs["temperature"] = model_kwargs["temperature"]
            if "top_p" in model_kwargs:
                api_kwargs["top_p"] = model_kwargs["top_p"]
            
            return api_kwargs
        elif model_type == ModelType.EMBEDDER:
            if isinstance(input, str):
                inputs = [input]
            elif isinstance(input, Sequence):
                inputs = list(input)
            else:
                raise TypeError("input must be a string or sequence of strings")

            api_kwargs["model"] = model_kwargs.get("model", "amazon.titan-embed-text-v2:0")
            api_kwargs["input"] = inputs
            api_kwargs["model_kwargs"] = model_kwargs
            return api_kwargs
        else:
            raise ValueError(f"Model type {model_type} is not supported by AWS Bedrock client")