SYMBOL INDEX (431 symbols across 67 files) FILE: builtin_tools/aws/aws.py class SageMakerProvider (line 6) | class SageMakerProvider(BuiltinToolProviderController): method _validate_credentials (line 7) | def _validate_credentials(self, credentials: dict) -> None: FILE: builtin_tools/aws/tools/apply_guardrail.py class GuardrailParameters (line 16) | class GuardrailParameters(BaseModel): class ApplyGuardrailTool (line 24) | class ApplyGuardrailTool(BuiltinTool): method _invoke (line 25) | def _invoke( FILE: builtin_tools/aws/tools/bedrock_retrieve.py class BedrockRetrieveTool (line 11) | class BedrockRetrieveTool(BuiltinTool): method _bedrock_retrieve (line 16) | def _bedrock_retrieve( method _invoke (line 73) | def _invoke( method validate_parameters (line 151) | def validate_parameters(self, parameters: dict[str, Any]) -> None: FILE: builtin_tools/aws/tools/bedrock_retrieve_and_generate.py class BedrockRetrieveAndGenerateTool (line 10) | class BedrockRetrieveAndGenerateTool(BuiltinTool): method _invoke (line 13) | def _invoke( method validate_parameters (line 111) | def validate_parameters(self, parameters: dict[str, Any]) -> None: FILE: builtin_tools/aws/tools/extract_frame.py class FrameExtractor (line 15) | class FrameExtractor(BuiltinTool): method _extract_specific_frames (line 16) | def _extract_specific_frames(self, gif_path, output_folder, frame_coun... method _clean_temp_dir (line 68) | def _clean_temp_dir(self, temp_dir): method _invoke (line 82) | def _invoke( FILE: builtin_tools/aws/tools/lambda_translate_utils.py class LambdaTranslateUtilsTool (line 10) | class LambdaTranslateUtilsTool(BuiltinTool): method _invoke_lambda (line 13) | def _invoke_lambda(self, text_content, src_lang, dest_lang, model_id, ... method _invoke (line 32) | def _invoke( FILE: builtin_tools/aws/tools/lambda_yaml_to_json.py class LambdaYamlToJsonTool (line 17) | class LambdaYamlToJsonTool(BuiltinTool): method _invoke_lambda (line 20) | def _invoke_lambda(self, lambda_name: str, yaml_content: str) -> str: method _invoke (line 38) | def _invoke( FILE: builtin_tools/aws/tools/nova_canvas.py class NovaCanvasTool (line 19) | class NovaCanvasTool(BuiltinTool): method _invoke (line 20) | def _invoke( method _validate_color_string (line 192) | def _validate_color_string(self, color_string) -> bool: method get_runtime_parameters (line 199) | def get_runtime_parameters(self) -> list[ToolParameter]: FILE: builtin_tools/aws/tools/nova_reel.py class NovaReelTool (line 32) | class NovaReelTool(BuiltinTool): method _invoke (line 33) | def _invoke( method _validate_and_extract_parameters (line 76) | def _validate_and_extract_parameters( method _initialize_aws_clients (line 108) | def _initialize_aws_clients(self, region: str) -> tuple[Any, Any]: method _prepare_model_input (line 114) | def _prepare_model_input(self, params: dict[str, Any], s3_client: Any)... method _process_and_validate_image (line 154) | def _process_and_validate_image(self, image_data: bytes) -> Union[Imag... method _get_image_from_s3 (line 205) | def _get_image_from_s3(self, s3_client: Any, s3_uri: str) -> Optional[... method _start_video_generation (line 214) | def _start_video_generation(self, bedrock: Any, model_input: dict[str,... method _handle_generation_mode (line 222) | def _handle_generation_mode( method _wait_for_completion (line 237) | def _wait_for_completion(self, bedrock: Any, s3_client: Any, invocatio... method _handle_completed_video (line 254) | def _handle_completed_video(self, s3_client: Any, video_path: str) -> ... method get_runtime_parameters (line 274) | def get_runtime_parameters(self) -> list[ToolParameter]: FILE: builtin_tools/aws/tools/opensearch_knn_search.py class OpenSearchRetrieveTool (line 13) | class OpenSearchRetrieveTool(BuiltinTool): method _get_embedding (line 18) | def _get_embedding(self, model_id:str, text:str=None, image_path:str=N... method _search_by_aos_knn (line 67) | def _search_by_aos_knn(self, q_embedding, index_name:str, embedding_fi... method _invoke (line 94) | def _invoke( FILE: builtin_tools/aws/tools/s3_operator.py class S3Operator (line 10) | class S3Operator(BuiltinTool): method _invoke (line 13) | def _invoke( FILE: builtin_tools/aws/tools/sagemaker_audio_to_text.py class WhisperTranscriptionTool (line 12) | class WhisperTranscriptionTool(BuiltinTool): method _get_s3_object_from_url (line 17) | def _get_s3_object_from_url(self, s3_url: str) -> tuple[str, str]: method _download_from_s3 (line 24) | def _download_from_s3(self, s3_url: str) -> str: method _invoke_sagemaker (line 41) | def _invoke_sagemaker(self, audio_data: bytes, endpoint: str): method _invoke (line 53) | def _invoke(self, FILE: builtin_tools/aws/tools/sagemaker_chinese_toxicity_detector.py class ContentModerationTool (line 13) | class ContentModerationTool(BuiltinTool): method _invoke_sagemaker (line 17) | def _invoke_sagemaker(self, payload: dict, endpoint: str): method _invoke (line 39) | def _invoke( FILE: builtin_tools/aws/tools/sagemaker_text_rerank.py class SageMakerReRankTool (line 11) | class SageMakerReRankTool(BuiltinTool): method _sagemaker_rerank (line 15) | def _sagemaker_rerank(self, query_input: str, docs: list[str], rerank_... method _invoke (line 27) | def _invoke( FILE: builtin_tools/aws/tools/sagemaker_tts.py class TTSModelType (line 11) | class TTSModelType(Enum): class SageMakerTTSTool (line 18) | class SageMakerTTSTool(BuiltinTool): method _detect_lang_code (line 24) | def _detect_lang_code(self, content: str, map_dict: Optional[dict] = N... method _build_tts_payload (line 31) | def _build_tts_payload( method _invoke_sagemaker (line 52) | def _invoke_sagemaker(self, payload: dict, endpoint: str): method _invoke (line 62) | def _invoke( FILE: builtin_tools/aws/tools/transcribe_asr.py function is_url (line 131) | def is_url(text): function upload_file_from_url_to_s3 (line 150) | def upload_file_from_url_to_s3(s3_client, url, bucket_name, s3_key=None,... class TranscribeTool (line 210) | class TranscribeTool(BuiltinTool): method _transcribe_audio (line 220) | def _transcribe_audio(self, audio_file_uri, file_type, **extra_args): method _download_and_read_transcript (line 244) | def _download_and_read_transcript(self, transcript_file_uri: str, max_... method _invoke (line 330) | def _invoke( FILE: model_provider/sagemaker/llm/llm.py function inference (line 45) | def inference(predictor, messages: list[dict[str, Any]], params: dict[st... class SageMakerLargeLanguageModel (line 79) | class SageMakerLargeLanguageModel(LargeLanguageModel): method _handle_chat_generate_response (line 88) | def _handle_chat_generate_response( method _handle_chat_stream_response (line 124) | def _handle_chat_stream_response( method _invoke (line 217) | def _invoke( method _convert_prompt_message_to_dict (line 286) | def _convert_prompt_message_to_dict(self, message: PromptMessage) -> d... method _num_tokens_from_messages (line 328) | def _num_tokens_from_messages( method get_num_tokens (line 386) | def get_num_tokens( method validate_credentials (line 408) | def validate_credentials(self, model: str, credentials: dict) -> None: method _invoke_error_mapping (line 423) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method get_customizable_model_schema (line 440) | def get_customizable_model_schema(self, model: str, credentials: dict)... FILE: model_provider/sagemaker/rerank/rerank.py class SageMakerRerankModel (line 25) | class SageMakerRerankModel(RerankModel): method _sagemaker_rerank (line 32) | def _sagemaker_rerank(self, query_input: str, docs: list[str], rerank_... method _invoke (line 44) | def _invoke( method validate_credentials (line 118) | def validate_credentials(self, model: str, credentials: dict) -> None: method _invoke_error_mapping (line 143) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method get_customizable_model_schema (line 160) | def get_customizable_model_schema(self, model: str, credentials: dict)... FILE: model_provider/sagemaker/sagemaker.py class SageMakerProvider (line 10) | class SageMakerProvider(ModelProvider): method validate_provider_credentials (line 11) | def validate_provider_credentials(self, credentials: dict) -> None: function buffer_to_s3 (line 22) | def buffer_to_s3(s3_client: Any, file: IO[bytes], bucket: str, s3_prefix... function generate_presigned_url (line 31) | def generate_presigned_url(s3_client: Any, file: IO[bytes], bucket_name:... FILE: model_provider/sagemaker/speech2text/speech2text.py class SageMakerSpeech2TextModel (line 24) | class SageMakerSpeech2TextModel(Speech2TextModel): method _invoke (line 32) | def _invoke(self, model: str, credentials: dict, file: IO[bytes], user... method validate_credentials (line 95) | def validate_credentials(self, model: str, credentials: dict) -> None: method _invoke_error_mapping (line 106) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method get_customizable_model_schema (line 123) | def get_customizable_model_schema(self, model: str, credentials: dict)... FILE: model_provider/sagemaker/text_embedding/text_embedding.py function batch_generator (line 30) | def batch_generator(generator, batch_size): class SageMakerEmbeddingModel (line 38) | class SageMakerEmbeddingModel(TextEmbeddingModel): method _sagemaker_embedding (line 45) | def _sagemaker_embedding(self, sm_client, endpoint_name, content_list:... method _invoke (line 56) | def _invoke( method get_num_tokens (line 122) | def get_num_tokens(self, model: str, credentials: dict, texts: list[st... method validate_credentials (line 133) | def validate_credentials(self, model: str, credentials: dict) -> None: method _calc_response_usage (line 146) | def _calc_response_usage(self, model: str, credentials: dict, tokens: ... method _invoke_error_mapping (line 174) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method get_customizable_model_schema (line 183) | def get_customizable_model_schema(self, model: str, credentials: dict)... FILE: model_provider/sagemaker/tts/tts.py class TTSModelType (line 26) | class TTSModelType(Enum): class SageMakerText2SpeechModel (line 33) | class SageMakerText2SpeechModel(TTSModel): method __init__ (line 38) | def __init__(self): method validate_credentials (line 70) | def validate_credentials(self, model: str, credentials: dict) -> None: method _detect_lang_code (line 80) | def _detect_lang_code(self, content: str, map_dict: Optional[dict] = N... method _build_tts_payload (line 88) | def _build_tts_payload( method _invoke (line 109) | def _invoke( method get_customizable_model_schema (line 162) | def get_customizable_model_schema(self, model: str, credentials: dict)... method _invoke_error_mapping (line 178) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method _get_model_default_voice (line 195) | def _get_model_default_voice(self, model: str, credentials: dict) -> Any: method _get_model_word_limit (line 198) | def _get_model_word_limit(self, model: str, credentials: dict) -> int: method _get_model_audio_type (line 201) | def _get_model_audio_type(self, model: str, credentials: dict) -> str: method _get_model_workers_limit (line 204) | def _get_model_workers_limit(self, model: str, credentials: dict) -> int: method get_tts_model_voices (line 207) | def get_tts_model_voices(self, model: str, credentials: dict, language... method _invoke_sagemaker (line 218) | def _invoke_sagemaker(self, payload: dict, endpoint: str): method _tts_invoke_streaming (line 228) | def _tts_invoke_streaming(self, model_type: str, payload: dict, sagema... FILE: notebook/byoc_qwen_rerank_deploy/app/inference.py function main (line 11) | def main() -> int: FILE: notebook/cosyvoice/code/api_server.py class LaunchFailed (line 26) | class LaunchFailed(Exception): function lifespan (line 30) | async def lifespan(app: FastAPI): function inference_fn (line 42) | def inference_fn(data): function ping (line 59) | async def ping(): function invocations (line 63) | async def invocations(request: Request): function roles (line 69) | async def roles(): FILE: notebook/cosyvoice/code/inference.py class SftRequest (line 43) | class SftRequest(BaseModel): class ZeroShotRequest (line 47) | class ZeroShotRequest(BaseModel): class InstructRequest (line 52) | class InstructRequest(BaseModel): function validate_sft_request (line 57) | def validate_sft_request(data: dict) -> SftRequest: function validate_zero_shot_request (line 60) | def validate_zero_shot_request(data: dict) -> ZeroShotRequest: function validate_instruct_request (line 63) | def validate_instruct_request(data: dict) -> InstructRequest: function save_to_s3 (line 66) | def save_to_s3(output) -> str: function generate_presigned_url (line 83) | def generate_presigned_url(s3_uri, expiration=3600): function get_audio (line 107) | def get_audio(url): class CosyVoiceService (line 112) | class CosyVoiceService(): method __init__ (line 113) | def __init__(self, model_dir:str): method list_avaliable_spks (line 117) | def list_avaliable_spks(self): method predict_fn (line 120) | def predict_fn(self, request): FILE: notebook/cosyvoice_china_region/code/api_server.py class LaunchFailed (line 26) | class LaunchFailed(Exception): function lifespan (line 30) | async def lifespan(app: FastAPI): function inference_fn (line 42) | def inference_fn(data): function ping (line 59) | async def ping(): function invocations (line 63) | async def invocations(request: Request): function roles (line 69) | async def roles(): FILE: notebook/cosyvoice_china_region/code/inference.py class SftRequest (line 43) | class SftRequest(BaseModel): class ZeroShotRequest (line 47) | class ZeroShotRequest(BaseModel): class InstructRequest (line 52) | class InstructRequest(BaseModel): function validate_sft_request (line 57) | def validate_sft_request(data: dict) -> SftRequest: function validate_zero_shot_request (line 60) | def validate_zero_shot_request(data: dict) -> ZeroShotRequest: function validate_instruct_request (line 63) | def validate_instruct_request(data: dict) -> InstructRequest: function save_to_s3 (line 66) | def save_to_s3(output) -> str: function generate_presigned_url (line 83) | def generate_presigned_url(s3_uri, expiration=3600): function get_audio (line 107) | def get_audio(url): class CosyVoiceService (line 112) | class CosyVoiceService(): method __init__ (line 113) | def __init__(self, model_dir:str): method list_avaliable_spks (line 117) | def list_avaliable_spks(self): method predict_fn (line 120) | def predict_fn(self, request): FILE: plugins/aws_tools/provider/aws_tools.py class AwsToolsProvider (line 7) | class AwsToolsProvider(ToolProvider): method _validate_credentials (line 8) | def _validate_credentials(self, credentials: dict[str, Any]) -> None: FILE: plugins/aws_tools/provider/utils.py class ParameterStoreManager (line 7) | class ParameterStoreManager: method __init__ (line 10) | def __init__(self, region_name: str = 'us-east-1'): method get_parameter (line 13) | def get_parameter(self, name: str, decrypt: bool = True, as_dict: bool... method put_parameter (line 43) | def put_parameter(self, name: str, value: Union[str, Dict, Any], param... method delete_parameter (line 74) | def delete_parameter(self, name: str) -> bool: FILE: plugins/aws_tools/tools/agentcore-browser-session-manager.py class AgentcoreBrowserSessionManagerTool (line 28) | class AgentcoreBrowserSessionManagerTool(Tool): method _init_browser_session (line 30) | def _init_browser_session(self, aws_region, session_timeout_seconds) -... method _close_browser_session (line 61) | def _close_browser_session(self, aws_region, session_id) -> dict: method _invoke (line 76) | def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolIn... FILE: plugins/aws_tools/tools/agentcore-browser-tool.py class AgentcoreBrowserToolTool (line 34) | class AgentcoreBrowserToolTool(Tool): method __init__ (line 38) | def __init__(self, **kwargs): method _setup_temp_dir (line 42) | def _setup_temp_dir(self): method _get_session (line 59) | def _get_session(self, session_id: str) -> dict: method _set_session (line 63) | def _set_session(self, session_id: str, browser, page, playwright): method _init_browser_session (line 71) | async def _init_browser_session(self, browser_session_id:str, aws_regi... method _cleanup_browser (line 137) | async def _cleanup_browser(self, session_id: str = None): method _browse_url (line 153) | async def _browse_url(self, browser_session_id:str, url: str, wait_tim... method _search_web (line 196) | async def _search_web(self, browser_session_id:str, query: str, wait_t... method _extract_content (line 260) | async def _extract_content(self, browser_session_id:str, url: str = No... method _fill_form (line 350) | async def _fill_form(self, browser_session_id:str, url: str = None, fo... method _execute_script (line 425) | async def _execute_script(self, browser_session_id:str, url: str = Non... method _invoke (line 459) | def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolIn... FILE: plugins/aws_tools/tools/agentcore_code_interpreter.py class AgentcoreCodeInterpreterTool (line 9) | class AgentcoreCodeInterpreterTool(Tool): method _invoke (line 10) | def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolIn... method execute (line 15) | def execute(self, language=None, code=None, command=None, session_id=N... method create_client (line 60) | def create_client(self, aws_ak=None, aws_sk=None, aws_region=None, ser... method create_code_interpreter (line 70) | def create_code_interpreter(self, client): method exec_code_internal (line 80) | def exec_code_internal(self, client, code_interpreter_id, session_id, ... method exec_command_internal (line 94) | def exec_command_internal(self, client, code_interpreter_id, session_i... method init_session (line 108) | def init_session(self, data_client, ci_id): method get_tool_result (line 117) | def get_tool_result(self, response): FILE: plugins/aws_tools/tools/agentcore_memory.py class AgentCoreMemoryTool (line 25) | class AgentCoreMemoryTool(Tool): method _clean_id_parameter (line 31) | def _clean_id_parameter(self, value: str) -> str: method _initialize_memory_client (line 40) | def _initialize_memory_client(self, tool_parameters: dict[str, Any]) -... method _create_new_memory_resource (line 72) | def _create_new_memory_resource(self, tool_parameters: dict[str, Any])... method _record_information (line 129) | def _record_information(self, tool_parameters: dict[str, Any]) -> Gene... method _retrieve_history (line 180) | def _retrieve_history(self, tool_parameters: dict[str, Any]) -> Genera... method _invoke (line 259) | def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolIn... FILE: plugins/aws_tools/tools/agentcore_memory_search.py class AgentCoreMemorySearchTool (line 25) | class AgentCoreMemorySearchTool(Tool): method _clean_id_parameter (line 28) | def _clean_id_parameter(self, value: str) -> str: method _initialize_memory_client (line 37) | def _initialize_memory_client(self, tool_parameters: dict[str, Any]) -... method _search_memories (line 69) | def _search_memories(self, tool_parameters: dict[str, Any]) -> Generat... method _invoke (line 154) | def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolIn... FILE: plugins/aws_tools/tools/apply_guardrail.py class GuardrailParameters (line 16) | class GuardrailParameters(BaseModel): class ApplyGuardrailTool (line 24) | class ApplyGuardrailTool(Tool): method _invoke (line 25) | def _invoke( FILE: plugins/aws_tools/tools/bedrock_retrieve.py class BedrockRetrieveTool (line 11) | class BedrockRetrieveTool(Tool): method convert_to_dify_kb_format (line 16) | def convert_to_dify_kb_format(self, kb_repsonse): method _bedrock_retrieve (line 63) | def _bedrock_retrieve( method _invoke (line 111) | def _invoke( method validate_parameters (line 189) | def validate_parameters(self, parameters: dict[str, Any]) -> None: FILE: plugins/aws_tools/tools/bedrock_retrieve_and_generate.py class BedrockRetrieveAndGenerateTool (line 10) | class BedrockRetrieveAndGenerateTool(Tool): method _format_text_with_citations (line 13) | def _format_text_with_citations(self, result: dict[str, Any]) -> str: method _invoke (line 33) | def _invoke( method validate_parameters (line 131) | def validate_parameters(self, parameters: dict[str, Any]) -> None: FILE: plugins/aws_tools/tools/dynamodb_manager.py class DynamoDBManager (line 11) | class DynamoDBManager(Tool): method _invoke (line 15) | def _invoke( method _create_table (line 52) | def _create_table(self, params: dict) -> str: method _put_item (line 80) | def _put_item(self, params: dict) -> str: method _get_item (line 104) | def _get_item(self, params: dict) -> str: method _delete_item (line 126) | def _delete_item(self, params: dict) -> str: method _scan (line 145) | def _scan(self, params: dict) -> dict: FILE: plugins/aws_tools/tools/extract_frame.py class FrameExtractor (line 10) | class FrameExtractor(Tool): method _extract_specific_frames (line 11) | def _extract_specific_frames(self, gif_path, output_folder, frame_coun... method _clean_temp_dir (line 63) | def _clean_temp_dir(self, temp_dir): method _invoke (line 77) | def _invoke( FILE: plugins/aws_tools/tools/lambda_translate_utils.py class LambdaTranslateUtilsTool (line 10) | class LambdaTranslateUtilsTool(Tool): method _invoke_lambda (line 13) | def _invoke_lambda(self, text_content, src_lang, dest_lang, model_id, ... method _invoke (line 32) | def _invoke( FILE: plugins/aws_tools/tools/lambda_yaml_to_json.py class LambdaYamlToJsonTool (line 18) | class LambdaYamlToJsonTool(Tool): method _invoke_lambda (line 21) | def _invoke_lambda(self, lambda_name: str, yaml_content: str) -> str: method _invoke (line 39) | def _invoke( FILE: plugins/aws_tools/tools/nova_canvas.py class NovaCanvasTool (line 24) | class NovaCanvasTool(Tool): method _invoke (line 25) | def _invoke( method _validate_color_string (line 194) | def _validate_color_string(self, color_string) -> bool: method get_runtime_parameters (line 201) | def get_runtime_parameters(self) -> list[ToolParameter]: FILE: plugins/aws_tools/tools/nova_reel.py class NovaReelTool (line 41) | class NovaReelTool(Tool): method _invoke (line 42) | def _invoke( method _validate_and_extract_parameters (line 85) | def _validate_and_extract_parameters( method _initialize_aws_clients (line 117) | def _initialize_aws_clients(self, region: str) -> tuple[Any, Any]: method _prepare_model_input (line 123) | def _prepare_model_input(self, params: dict[str, Any], s3_client: Any)... method _process_and_validate_image (line 163) | def _process_and_validate_image(self, image_data: bytes) -> Union[Imag... method _get_image_from_s3 (line 214) | def _get_image_from_s3(self, s3_client: Any, s3_uri: str) -> Optional[... method _start_video_generation (line 223) | def _start_video_generation(self, bedrock: Any, model_input: dict[str,... method _handle_generation_mode (line 231) | def _handle_generation_mode( method _wait_for_completion (line 246) | def _wait_for_completion(self, bedrock: Any, s3_client: Any, invocatio... method _handle_completed_video (line 263) | def _handle_completed_video(self, s3_client: Any, video_path: str) -> ... method get_runtime_parameters (line 283) | def get_runtime_parameters(self) -> list[ToolParameter]: FILE: plugins/aws_tools/tools/opensearch_knn_search.py class OpenSearchRetrieveTool (line 13) | class OpenSearchRetrieveTool(Tool): method _get_embedding (line 18) | def _get_embedding(self, model_id:str, text:str=None, image_path:str=N... method _search_by_aos_knn (line 67) | def _search_by_aos_knn(self, q_embedding, index_name:str, embedding_fi... method _invoke (line 94) | def _invoke( FILE: plugins/aws_tools/tools/s3_operator.py class S3Operator (line 15) | class S3Operator(Tool): method _invoke (line 18) | def _invoke( FILE: plugins/aws_tools/tools/sagemaker_chinese_toxicity_detector.py class ContentModerationTool (line 15) | class ContentModerationTool(Tool): method _invoke_sagemaker (line 19) | def _invoke_sagemaker(self, payload: dict, endpoint: str): method _invoke (line 41) | def _invoke( FILE: plugins/aws_tools/tools/sagemaker_text_rerank.py class SageMakerReRankTool (line 11) | class SageMakerReRankTool(Tool): method _sagemaker_rerank (line 15) | def _sagemaker_rerank(self, query_input: str, docs: list[str], rerank_... method _invoke (line 27) | def _invoke( FILE: plugins/aws_tools/tools/sagemaker_tts.py class TTSModelType (line 12) | class TTSModelType(Enum): class SageMakerTTSTool (line 19) | class SageMakerTTSTool(Tool): method _detect_lang_code (line 25) | def _detect_lang_code(self, content: str, map_dict: Optional[dict] = N... method _build_tts_payload (line 32) | def _build_tts_payload( method _invoke_sagemaker (line 53) | def _invoke_sagemaker(self, payload: dict, endpoint: str): method _invoke (line 63) | def _invoke( FILE: plugins/aws_tools/tools/transcribe_asr.py function is_url (line 132) | def is_url(text): function upload_file_from_url_to_s3 (line 151) | def upload_file_from_url_to_s3(s3_client, url, bucket_name, s3_key=None,... class TranscribeTool (line 211) | class TranscribeTool(Tool): method _transcribe_audio (line 221) | def _transcribe_audio(self, audio_file_uri, file_type, **extra_args): method _download_and_read_transcript (line 245) | def _download_and_read_transcript(self, transcript_file_uri: str, max_... method _invoke (line 331) | def _invoke( FILE: plugins/aws_tools/tools/translation_evaluator.py function tokenize_zh (line 18) | def tokenize_zh(text): function calculate_bleu (line 21) | def calculate_bleu(references, hypotheses , weights=(0.25, 0.25, 0.25, 0... function chinese_meteor_score (line 39) | def chinese_meteor_score(references, hypothesis, alpha=0.9, beta=3.0, ga... function calculate_nist (line 55) | def calculate_nist(references, hypotheses): function evaluate_with_metric (line 65) | def evaluate_with_metric(references, hypotheses): function evaluate_with_model (line 80) | def evaluate_with_model(sagemaker_endpoint, source, translation): class TranslationEvalTool (line 83) | class TranslationEvalTool(Tool): method _invoke (line 87) | def _invoke( FILE: plugins/bedrock/models/llm/cache_config.py function is_cache_supported (line 77) | def is_cache_supported(model_id: str) -> bool: function get_cache_config (line 90) | def get_cache_config(model_id: str) -> dict: FILE: plugins/bedrock/models/llm/llm.py class BedrockLargeLanguageModel (line 75) | class BedrockLargeLanguageModel(LargeLanguageModel): method _find_model_info (line 106) | def _find_model_info(model_id): method _code_block_mode_wrapper (line 113) | def _code_block_mode_wrapper( method _invoke (line 146) | def _invoke( method _get_model_info (line 238) | def _get_model_info(self, model: str, credentials: dict, model_paramet... method _generate_with_converse (line 317) | def _generate_with_converse( method _handle_converse_response (line 479) | def _handle_converse_response( method _extract_tool_use (line 541) | def _extract_tool_use(self, content: dict) -> tuple[str, dict]: method _handle_converse_stream_response (line 553) | def _handle_converse_stream_response( method _convert_converse_api_model_parameters (line 713) | def _convert_converse_api_model_parameters( method _convert_converse_prompt_messages (line 760) | def _convert_converse_prompt_messages(self, prompt_messages: list[Prom... method _convert_converse_tool_config (line 827) | def _convert_converse_tool_config(self, tools: Optional[list[PromptMes... method _convert_prompt_message_to_dict (line 844) | def _convert_prompt_message_to_dict(self, message: PromptMessage) -> d... method get_num_tokens (line 932) | def get_num_tokens( method get_customizable_model_schema (line 968) | def get_customizable_model_schema(self, model: str, credentials: dict)... method validate_credentials (line 1082) | def validate_credentials(self, model: str, credentials: dict) -> None: method _list_foundation_models (line 1112) | def _list_foundation_models(self, credentials: dict) -> list[str]: method _convert_one_message_to_text (line 1141) | def _convert_one_message_to_text( method _convert_messages_to_prompt (line 1172) | def _convert_messages_to_prompt( method _create_payload (line 1194) | def _create_payload( method _generate (line 1234) | def _generate( method _handle_generate_response (line 1289) | def _handle_generate_response( method _handle_generate_stream_response (line 1340) | def _handle_generate_stream_response( method _invoke_error_mapping (line 1424) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method _map_client_to_invoke_error (line 1441) | def _map_client_to_invoke_error(self, error_code: str, error_msg: str)... method _get_inference_profile_parameter_rules (line 1468) | def _get_inference_profile_parameter_rules(self, model_schema, underly... method _model_id_matches_schema (line 1501) | def _model_id_matches_schema(self, model_id: str, model_schema) -> bool: method _map_model_id_to_name (line 1530) | def _map_model_id_to_name(self, model_id: str) -> Optional[str]: method _get_model_specific_pricing (line 1553) | def _get_model_specific_pricing(self, model: str, model_name: str, mod... method _calc_response_usage (line 1638) | def _calc_response_usage(self, model: str, credentials: dict, prompt_t... FILE: plugins/bedrock/models/llm/model_ids.py function is_support_cross_region (line 73) | def is_support_cross_region(model_id): function get_model_id (line 85) | def get_model_id(model_type, model_name): function get_region_area (line 98) | def get_region_area(region_name, prefer_global=False): FILE: plugins/bedrock/models/rerank/model_ids.py function get_model_id (line 17) | def get_model_id(model_type, model_name): function get_all_model_choices (line 30) | def get_all_model_choices(): FILE: plugins/bedrock/models/rerank/rerank.py class BedrockRerankModel (line 33) | class BedrockRerankModel(RerankModel): method _invoke (line 38) | def _invoke( method get_customizable_model_schema (line 152) | def get_customizable_model_schema(self, model: str, credentials: dict)... method validate_credentials (line 227) | def validate_credentials(self, model: str, credentials: dict) -> None: method validate_credentials (line 252) | def validate_credentials(self, model: str, credentials: dict) -> None: method _invoke_error_mapping (line 277) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... FILE: plugins/bedrock/models/text_embedding/model_ids.py function get_model_id (line 20) | def get_model_id(model_type, model_name): function get_all_model_choices (line 33) | def get_all_model_choices(): FILE: plugins/bedrock/models/text_embedding/text_embedding.py class BedrockTextEmbeddingModel (line 38) | class BedrockTextEmbeddingModel(TextEmbeddingModel): method _invoke (line 39) | def _invoke( method get_num_tokens (line 155) | def get_num_tokens(self, model: str, credentials: dict, texts: list[st... method validate_credentials (line 180) | def validate_credentials(self, model: str, credentials: dict) -> None: method _invoke_error_mapping (line 190) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method _create_payload (line 207) | def _create_payload( method _calc_response_usage (line 223) | def _calc_response_usage(self, model: str, credentials: dict, tokens: ... method _map_client_to_invoke_error (line 250) | def _map_client_to_invoke_error(self, error_code: str, error_msg: str)... method _invoke_bedrock_embedding (line 277) | def _invoke_bedrock_embedding( method get_customizable_model_schema (line 305) | def get_customizable_model_schema(self, model: str, credentials: dict)... method validate_credentials (line 380) | def validate_credentials(self, model: str, credentials: dict) -> None: method _invoke_multimodal (line 409) | def _invoke_multimodal( method _get_image_format (line 496) | def _get_image_format(self, base64_string: str) -> str: FILE: plugins/bedrock/provider/bedrock.py class AmazonBedrockModelProvider (line 14) | class AmazonBedrockModelProvider(ModelProvider): method validate_provider_credentials (line 15) | def validate_provider_credentials(self, credentials: Mapping) -> None: method validate_model_credentials (line 33) | def validate_model_credentials(self, model: str, model_type: ModelType... method _validate_inference_profile (line 62) | def _validate_inference_profile(self, inference_profile_id: str, crede... FILE: plugins/bedrock/provider/get_bedrock_client.py function get_bedrock_client (line 10) | def get_bedrock_client(service_name: str, credentials: Mapping[str, str]): FILE: plugins/bedrock/utils/inference_profile.py function _get_fetch_lock (line 25) | def _get_fetch_lock(cache_key: str) -> threading.Lock: function get_inference_profile_info (line 42) | def get_inference_profile_info(inference_profile_id: str, credentials: d... function validate_inference_profile (line 108) | def validate_inference_profile(inference_profile_id: str, credentials: d... function extract_model_id_from_arn (line 137) | def extract_model_id_from_arn(model_arn: str) -> str: function extract_model_info_from_profile (line 151) | def extract_model_info_from_profile(profile_info: dict) -> Dict[str, Any]: FILE: plugins/sagemaker/models/llm/llm.py function inference (line 54) | def inference(predictor, messages: list[dict[str, Any]], params: dict[st... class SageMakerLargeLanguageModel (line 89) | class SageMakerLargeLanguageModel(LargeLanguageModel): method _handle_chat_generate_response (line 102) | def _handle_chat_generate_response( method _handle_chat_stream_response (line 138) | def _handle_chat_stream_response( method _refresh_token (line 231) | def _refresh_token(self): method _invoke (line 253) | def _invoke( method _convert_prompt_message_to_dict (line 343) | def _convert_prompt_message_to_dict(self, message: PromptMessage) -> d... method _num_tokens_from_messages (line 385) | def _num_tokens_from_messages( method get_num_tokens (line 443) | def get_num_tokens( method validate_credentials (line 465) | def validate_credentials(self, model: str, credentials: dict) -> None: method _invoke_error_mapping (line 480) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method get_customizable_model_schema (line 497) | def get_customizable_model_schema(self, model: str, credentials: dict)... FILE: plugins/sagemaker/models/rerank/rerank.py class SageMakerRerankModel (line 24) | class SageMakerRerankModel(RerankModel): method _sagemaker_rerank (line 31) | def _sagemaker_rerank(self, query_input: str, docs: list[str], rerank_... method _invoke (line 43) | def _invoke( method validate_credentials (line 118) | def validate_credentials(self, model: str, credentials: dict) -> None: method _invoke_error_mapping (line 143) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method get_customizable_model_schema (line 160) | def get_customizable_model_schema(self, model: str, credentials: dict)... FILE: plugins/sagemaker/models/speech2text/speech2text.py class SageMakerSpeech2TextModel (line 24) | class SageMakerSpeech2TextModel(Speech2TextModel): method _invoke (line 32) | def _invoke(self, model: str, credentials: dict, file: IO[bytes], user... method validate_credentials (line 95) | def validate_credentials(self, model: str, credentials: dict) -> None: method _invoke_error_mapping (line 106) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method get_customizable_model_schema (line 123) | def get_customizable_model_schema(self, model: str, credentials: dict)... FILE: plugins/sagemaker/models/text_embedding/text_embedding.py function batch_generator (line 35) | def batch_generator(generator, batch_size): class SageMakerEmbeddingModel (line 43) | class SageMakerEmbeddingModel(TextEmbeddingModel): method _sagemaker_embedding (line 50) | def _sagemaker_embedding(self, sm_client, endpoint_name, content_list:... method _invoke (line 61) | def _invoke( method get_num_tokens (line 128) | def get_num_tokens(self, model: str, credentials: dict, texts: list[st... method validate_credentials (line 139) | def validate_credentials(self, model: str, credentials: dict) -> None: method _calc_response_usage (line 152) | def _calc_response_usage(self, model: str, credentials: dict, tokens: ... method _invoke_error_mapping (line 180) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method get_customizable_model_schema (line 189) | def get_customizable_model_schema(self, model: str, credentials: dict)... FILE: plugins/sagemaker/models/tts/tts.py class TTSModelType (line 24) | class TTSModelType(Enum): class SageMakerText2SpeechModel (line 31) | class SageMakerText2SpeechModel(TTSModel): method __init__ (line 36) | def __init__(self, model_schemas: list[AIModelEntity]) -> None: method validate_credentials (line 57) | def validate_credentials(self, model: str, credentials: dict) -> None: method _detect_lang_code (line 67) | def _detect_lang_code(self, content: str, map_dict: Optional[dict] = N... method _build_tts_payload (line 75) | def _build_tts_payload( method _invoke (line 96) | def _invoke( method get_customizable_model_schema (line 149) | def get_customizable_model_schema(self, model: str, credentials: dict)... method _invoke_error_mapping (line 165) | def _invoke_error_mapping(self) -> dict[type[InvokeError], list[type[E... method _get_model_default_voice (line 182) | def _get_model_default_voice(self, model: str, credentials: dict) -> Any: method _get_model_word_limit (line 185) | def _get_model_word_limit(self, model: str, credentials: dict) -> int: method _get_model_audio_type (line 188) | def _get_model_audio_type(self, model: str, credentials: dict) -> str: method _get_model_workers_limit (line 191) | def _get_model_workers_limit(self, model: str, credentials: dict) -> int: method get_tts_model_voices (line 194) | def get_tts_model_voices(self, model: str, credentials: dict, language... method _invoke_sagemaker (line 205) | def _invoke_sagemaker(self, payload: dict, endpoint: str): method _tts_invoke_streaming (line 215) | def _tts_invoke_streaming(self, model_type: str, payload: dict, sagema... FILE: plugins/sagemaker/provider/sagemaker.py class SageMakerProvider (line 12) | class SageMakerProvider(ModelProvider): method validate_provider_credentials (line 13) | def validate_provider_credentials(self, credentials: dict) -> None: function buffer_to_s3 (line 23) | def buffer_to_s3(s3_client: Any, file: IO[bytes], bucket: str, s3_prefix... function generate_presigned_url (line 32) | def generate_presigned_url(s3_client: Any, file: IO[bytes], bucket_name:... FILE: workflow/eks_upgrade_planning/eks_cluster_info.py class TokenGenerator (line 15) | class TokenGenerator(object): method __init__ (line 16) | def __init__(self, sts_client): method get_token (line 19) | def get_token(self, k8s_aws_id): method _get_presigned_url (line 27) | def _get_presigned_url(self, k8s_aws_id): class STSClientFactory (line 35) | class STSClientFactory(object): method __init__ (line 36) | def __init__(self, session): method get_sts_client (line 39) | def get_sts_client(self, region_name=None, role_arn=None): method _get_role_credentials (line 50) | def _get_role_credentials(self, region_name, role_arn): method _register_k8s_aws_id_handlers (line 56) | def _register_k8s_aws_id_handlers(self, sts_client): method _retrieve_k8s_aws_id (line 66) | def _retrieve_k8s_aws_id(self, params, context, **kwargs): method _inject_k8s_aws_id_header (line 70) | def _inject_k8s_aws_id_header(self, request, **kwargs): function get_cluster_info (line 74) | def get_cluster_info(cluster_name, region, profile=None): function get_current_version (line 84) | def get_current_version(cluster_info): function get_health_issues (line 87) | def get_health_issues(cluster_info): function get_compatibility_issues (line 90) | def get_compatibility_issues(cluster_name, region, profile=None): function get_addon_compatibility_issues (line 120) | def get_addon_compatibility_issues(compatibility_issues): function get_addon_compatible_versions (line 133) | def get_addon_compatible_versions(compatibility_issues): function get_deprecated_api_versions (line 139) | def get_deprecated_api_versions(compatibility_issues): function get_nodegroups (line 153) | def get_nodegroups(cluster_name, region, profile=None): function get_fargate_profiles (line 174) | def get_fargate_profiles(cluster_name, region, profile=None): function get_installed_addons (line 184) | def get_installed_addons(cluster_name, region, profile=None): function get_addon_upgrade_info (line 201) | def get_addon_upgrade_info(cluster_name, region, current_version, target... function check_version_skew (line 351) | def check_version_skew(current_version, min_nodegroup_version, kube_prox... function create_temp_cert_file (line 395) | def create_temp_cert_file(cert_data): function validate_target_version (line 405) | def validate_target_version(current_version, target_version): function connect_to_cluster (line 440) | def connect_to_cluster(cluster_name, region=None, profile=None): function get_node_versions (line 485) | def get_node_versions(api_client): function get_opensource_addons (line 562) | def get_opensource_addons(api_client, eks_addons=None): function extract_helm_info_from_labels (line 730) | def extract_helm_info_from_labels(labels, addon_info): function get_image_version (line 763) | def get_image_version(image_string): function get_core_components_version (line 785) | def get_core_components_version(api_client, eks_addons): function main (line 859) | def main(cluster_name, region, target_version, profile=None):