25 lines
808 B
Python
25 lines
808 B
Python
"""LLM factory — Bedrock Converse API (production) or direct Anthropic SDK (local dev)."""
|
|
|
|
import os
|
|
|
|
|
|
def get_agent_llm():
|
|
"""Returns a LangChain chat model for agent orchestration."""
|
|
if os.getenv("USE_BEDROCK", "").lower() == "true":
|
|
from langchain_aws import ChatBedrockConverse
|
|
|
|
return ChatBedrockConverse(
|
|
model=os.getenv("BEDROCK_MODEL_ID", "anthropic.claude-sonnet-4-20250514-v1:0"),
|
|
region_name=os.getenv("AWS_DEFAULT_REGION", "us-east-1"),
|
|
temperature=0.3,
|
|
max_tokens=4096,
|
|
)
|
|
else:
|
|
from langchain_anthropic import ChatAnthropic
|
|
|
|
return ChatAnthropic(
|
|
model=os.getenv("ANTHROPIC_MODEL", "claude-sonnet-4-20250514"),
|
|
temperature=0.3,
|
|
max_tokens=4096,
|
|
)
|