tests.system.providers.amazon.aws.example_bedrock

Module Contents

Functions

delete_custom_model(model_name)

Attributes

ROLE_ARN_KEY

sys_test_context_task

DAG_ID

SKIP_LONG_TASKS

LLAMA_MODEL_ID

PROMPT

TITAN_MODEL_ID

TRAIN_DATA

HYPERPARAMETERS

test_context

test_run

tests.system.providers.amazon.aws.example_bedrock.ROLE_ARN_KEY = 'ROLE_ARN'[source]
tests.system.providers.amazon.aws.example_bedrock.sys_test_context_task[source]
tests.system.providers.amazon.aws.example_bedrock.DAG_ID = 'example_bedrock'[source]
tests.system.providers.amazon.aws.example_bedrock.SKIP_LONG_TASKS = True[source]
tests.system.providers.amazon.aws.example_bedrock.LLAMA_MODEL_ID = 'meta.llama2-13b-chat-v1'[source]
tests.system.providers.amazon.aws.example_bedrock.PROMPT = 'What color is an orange?'[source]
tests.system.providers.amazon.aws.example_bedrock.TITAN_MODEL_ID = 'amazon.titan-text-express-v1'[source]
tests.system.providers.amazon.aws.example_bedrock.TRAIN_DATA[source]
tests.system.providers.amazon.aws.example_bedrock.HYPERPARAMETERS[source]
tests.system.providers.amazon.aws.example_bedrock.delete_custom_model(model_name)[source]
tests.system.providers.amazon.aws.example_bedrock.test_context[source]
tests.system.providers.amazon.aws.example_bedrock.test_run[source]

Was this entry helpful?