ML-Learner / python_backend /src /Agents /llm /llm_loader.py
VashuTheGreat2's picture
Upload folder using huggingface_hub
c01955c verified
from langchain_aws import ChatBedrockConverse
from src.Agents.constants import LLM_MODEL_ID, LLM_REGION
import logging
llm = ChatBedrockConverse(
model_id=LLM_MODEL_ID,
region_name=LLM_REGION
)
logging.info(f"LLM initialized with model_id={LLM_MODEL_ID}, region_name={LLM_REGION}")