tests.system.amazon.aws.example_bedrock_batch_inference¶

Attributes¶

`generate_prompts`(_env_id, _bucket, _key)	Bedrock Batch Inference requires one or more jsonl-formatted files in an S3 bucket.
`stop_batch_inference`(job_arn)

tests.system.amazon.aws.example_bedrock_batch_inference.ROLE_ARN_KEY = 'ROLE_ARN'[source]¶

tests.system.amazon.aws.example_bedrock_batch_inference.sys_test_context_task[source]¶

tests.system.amazon.aws.example_bedrock_batch_inference.DAG_ID = 'example_bedrock_batch_inference'[source]¶

tests.system.amazon.aws.example_bedrock_batch_inference.CLAUDE_MODEL_ID = 'anthropic.claude-3-sonnet-20240229-v1:0'[source]¶

tests.system.amazon.aws.example_bedrock_batch_inference.ANTHROPIC_VERSION = 'bedrock-2023-05-31'[source]¶

tests.system.amazon.aws.example_bedrock_batch_inference.MIN_NUM_PROMPTS = 300[source]¶

tests.system.amazon.aws.example_bedrock_batch_inference.PROMPT_TEMPLATE = 'Even numbers are red. Odd numbers are blue. What color is {n}?'[source]¶

tests.system.amazon.aws.example_bedrock_batch_inference.generate_prompts(_env_id, _bucket, _key)[source]¶

Bedrock Batch Inference requires one or more jsonl-formatted files in an S3 bucket.

The JSONL format requires one serialized json object per prompt per line.

tests.system.amazon.aws.example_bedrock_batch_inference.stop_batch_inference(job_arn)[source]¶

tests.system.amazon.aws.example_bedrock_batch_inference.test_context[source]¶