more docker fixes
This commit is contained in:
parent
e2bd322814
commit
c4eebafff6
4 changed files with 9 additions and 10 deletions
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue