system-prompts-and-models-o.../task/llm_bedrock.py
2026-06-18 21:23:01 +05:30

14 lines
507 B
Python

"""Legacy module name kept for import compatibility.
Bedrock has been removed from this service — `model` is now the Gemini-backed
chat model from the unified Gemini-only interface in task/llm.py. Existing
`from llm_bedrock import model` call sites keep working unchanged.
"""
try:
from llm import get_chat_model
except ImportError:
from task.llm import get_chat_model # type: ignore
model_parameter = {"temperature": 0, "max_tokens": 4096}
model = get_chat_model(model_kwargs=model_parameter)