INNER CODE UNIT · Python

parse_args

aws-samples/rag-using-langchain-amazon-bedrock-and-opensearch · ask-bedrock-with-rag.py:20

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--ask", type=str, default="What is the meaning of <3?")
    parser.add_argument("--index", type=str, default="rag")
    parser.add_argument("--region", type=str, default="us-east-1")
    parser.add_argument("--tenant-id", type=str, default=None)
    parser.add_argument("--bedrock-model-id", type=str, default="anthropic.claude-3-sonnet-20240229-v1:0")
    parser.add_argument("--bedrock-embedding-model-id", type=str, default="amazon.titan-embed-text-v1")
    
    return parser.parse_known_args()


def get_bedrock_client(region):
    bedrock_client = boto3.client("bedrock-runtime", region_name=region)
    return bedrock_client
    

def create_langchain_vector_embedding_using_bedrock(bedrock_client, bedrock_embedding_model_id):

View source record →