INNER CODE UNIT · Python

parse_args

aws-samples/rag-using-langchain-amazon-bedrock-and-opensearch · load-data-to-opensearch.py:17

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--recreate", type=bool, default=0)
    parser.add_argument("--early-stop", type=bool, default=0)
    parser.add_argument("--index", type=str, default="rag")
    parser.add_argument("--region", type=str, default="us-east-1")
    parser.add_argument("--multi-tenant", type=bool, default=0)
    
    return parser.parse_known_args()


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


def create_vector_embedding_with_bedrock(text, name, bedrock_client):
    payload = {"inputText": f"{text}"}

View source record →

📰 Research Paper
Loading…
⏳ Fetching content…