Katon Minhas
70 lines
2.9 KiB
Python
70 lines
2.9 KiB
Python
|
|
|
|
from datetime import datetime
|
|
import boto3
|
|
from llama_index.llms.bedrock import Bedrock
|
|
|
|
# General Settings
|
|
TEST = False # True to run test prompt - just for testing model connection
|
|
VERBOSE = True
|
|
CLIENT_NAME = 'Payer'
|
|
TODAY = datetime.now().strftime("%Y%m%d")
|
|
|
|
# I/O Options
|
|
WRITE_OUTPUT = True # True writes csvs, False prints result in console but no output written
|
|
READ_MODE = '_LOCAL_' # OR '_S3_'
|
|
OUTPUT_MODE = '_INDIVIDUAL_' # or '_CONSOLIDATED_'
|
|
|
|
# Multithread Settings
|
|
MAX_WORKERS = 20
|
|
|
|
# Prompt Debugging
|
|
RUN_PRIMARY = True # Always True
|
|
RUN_LOB = True
|
|
RUN_LESSER = True
|
|
RUN_METHODOLOGY = True
|
|
RUN_FS = True
|
|
RUN_EXCEPTION = True
|
|
RUN_CODES = True
|
|
|
|
# AWS Keys
|
|
AWS_ACCESS_KEY_ID="ASIAZTMXAXNXNEUYUPUY"
|
|
AWS_SECRET_ACCESS_KEY="3HljGiz+dqWBtj+dkqs/syyxoguUoTahdLyceEeo"
|
|
AWS_SESSION_TOKEN="IQoJb3JpZ2luX2VjEBUaCXVzLWVhc3QtMiJHMEUCIQCX9LcYh9A28cLNA/lDEHJ6ZqrulN0FampT1ptNyX+9JAIgIFrmDiZS5KfHbZWptuvveZt2s7x5r0xDwCU6t6EgBAEqjAMIjv//////////ARAAGgw2NjAxMzEwNjg3ODIiDOLRiVmy5kw46/5mzCrgAscMrf+1rEm7kpwpmd+BHjLNV4ggEy0oqQQ6z5Ny5BV6Ouo0tRmXqOqSaI5/YKXeKsRo8s4lrDNNq8qIK0GxpugvmbqguKYfEzKwyihGi5IVVb2JGL1nBkL495v0w5j1fuXqNwB7VCPkJyJCqxDHeehgtQ1AymuVAiRowkWmr7vVvFQyKKcGsBJlIzBZyguTtJhT2HPCp8IW9LhQMS2XBAa4xCnHqpPZSYIoKyp2zPPwi3pBKm+7eE3NGH7V7/Vrl6IL29zNkk5K/IeqAcyBzaFgaCPilr5N6lSMeKlkz8w0ls3guZeaEjDRf/Mz3NvMD9HeUVqGfi2M+ODxudgEfmfH3Op0EC2g5fsIBu/uvcjJf7PgUo9xT6CH5P5HY/e6TLaZH55ioz468uaTfoNw10TZ/6iymqYEXSpLfqHejC6qCx2pfcYf1UizgD2rn22GfcGs8u6oKFSThJ4fETq1mogwzafCsgY6pgHl1EzoiRwiXqCDl3GsA162dYrapmEPqcIVXTTuNuYXbl7B3GQfxTmjHjqSqKFhieWEvydD4SSQAhSXz9tPsN69Myyelwg3t6r2EWGL+JiZXEFpO1peWJgu3xfd3F1tPnp7Nfv7bVGmuIvft1JYTmx20W/vH9Q4J7jM4tPqEeQs5fRORTdXcm10Yysuq5FYo+9bahUJnPYblcJveO30vVirwzVWvmPm"
|
|
|
|
# File Paths
|
|
LOCAL_PATH = 'docs/priority_health/' # Replace with local
|
|
|
|
# S3 Settings
|
|
S3_CLIENT = boto3.client('s3',
|
|
region_name="us-east-2",
|
|
aws_access_key_id=AWS_ACCESS_KEY_ID,
|
|
aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
|
|
aws_session_token=AWS_SESSION_TOKEN
|
|
)
|
|
BUCKET = "doczy-dev-infra-textract"
|
|
PREFIX = "batches/batch_1/contract-text-file/" # replace with s3 path
|
|
|
|
# Bedrock Settings
|
|
BEDROCK_RUNTIME = boto3.client(service_name="bedrock-runtime",
|
|
region_name="us-east-1",
|
|
aws_access_key_id=AWS_ACCESS_KEY_ID,
|
|
aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
|
|
aws_session_token=AWS_SESSION_TOKEN
|
|
)
|
|
MODEL_ID_CLAUDE3_HAIKU = 'anthropic.claude-3-haiku-20240307-v1:0'
|
|
MODEL_ID_CLAUDE3_SONNET = 'anthropic.claude-3-sonnet-20240229-v1:0'
|
|
MODEL_ID_CLAUDE2 = 'anthropic.claude-instant-v1'
|
|
|
|
# LLama Model
|
|
LLAMA_MODEL = Bedrock(
|
|
model=MODEL_ID_CLAUDE3_HAIKU,
|
|
aws_access_key_id=AWS_ACCESS_KEY_ID,
|
|
aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
|
|
aws_session_token=AWS_SESSION_TOKEN,
|
|
region_name="us-east-1"
|
|
)
|
|
|
|
|
|
|