bedrock_client.py

import os
from anthropic import AnthropicBedrock
from langchain_aws.chat_models import ChatBedrockConverse
from langchain_aws.llms.bedrock import BedrockLLM




# Initialize the streaming Bedrock chat model
bedrock_llm = ChatBedrockConverse(
    aws_access_key_id       = os.environ.get("AWS_ACCESS_KEY_ID"),
    aws_secret_access_key   = os.environ.get("AWS_SECRET_ACCESS_KEY"),
    region_name             = os.environ.get("AWS_DEFAULT_REGION", "eu-west-1"),
    provider                = os.environ.get("PROVIDER", "mistral"),
    model_id                = os.environ.get("MODEL_ID", "mistral.mistral-large-2402-v1:0"),  # or your preferred Bedrock model
    temperature             = os.environ.get("TEMPERATURE", 0.7)
    )