more docker fixes

This commit is contained in:
Sebarocks 2025-07-31 22:50:03 -04:00
parent e2bd322814
commit c4eebafff6
4 changed files with 9 additions and 10 deletions

View file

@ -2,15 +2,16 @@ from langchain_openai import ChatOpenAI
from langchain_core.messages import HumanMessage, AIMessage
from os import getenv
from dotenv import load_dotenv
from pydantic import SecretStr
load_dotenv()
def get_llm(provider: str):
"""Return a LangChain chat model for the requested provider."""
return ChatOpenAI(
openai_api_key=getenv("OPENROUTER_API_KEY"),
openai_api_base=getenv("OPENROUTER_BASE_URL"),
model_name=provider,
api_key=SecretStr(getenv("OPENROUTER_API_KEY","")),
base_url=getenv("OPENROUTER_BASE_URL"),
model=provider,
)
def get_messages(chats, chat_id):