search

vanna-ai / vanna

🤖 Chat with your SQL database 📊. Accurate Text-to-SQL Generation via LLMs using RAG 🔄.
https://vanna.ai/docs/
MIT License
9.99k stars 739 forks source link

feat: Bedrock_Converse integration #540

Closed lopezfelipe closed 2 weeks ago

lopezfelipe commented 2 weeks ago

Adding methods to call models from Amazon Bedrock using the new Converse API. This class requires that the user must pass a valid session with the Bedrock runtime, e.g.,

import boto3

session = boto3.Session()
region = "us-east-1"
bedrock_client = boto3.client(service_name = 'bedrock-runtime', region_name = region)

Issue: 456