
How to Build an AI-Powered Semantic Search with Bitnimbus VectorDB on AWS
Last updated
/import boto3, json
client = boto3.client("bedrock-runtime", region_name="us-east-1")
req = {"inputText": "Explain RAG in one sentence."}
res = client.invoke_model(modelId="amazon.titan-embed-text-v2:0", body=json.dumps(req))
vector = json.loads(res["body"].read())["embedding"]import chromadb
client = chromadb.HttpClient(
host="YOUR_BITNIMBUS_ENDPOINT", port=443, headers={"X-API-Key": "YOUR_KEY"}
)
col = client.create_collection(name="papers")
col.add(documents=[...], embeddings=[...], ids=[...])results = col.query(query_texts=["recommend pizza toppings"], n_results=3)
print(results["documents"], results["distances"])prompt = "\n\n".join(results["documents"]) + "\n\nUser: What's a good pizza combo?"
res = client.invoke_model(
modelId="amazon.titan-text-premier-v1:0",
body=json.dumps({"inputText": prompt, "textGenerationConfig": {"maxTokenCount":256}})
)
print(json.loads(res["body"].read())["generatedText"])