Spaces:
Sleeping
Sleeping
File size: 563 Bytes
95f7828 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | # Load and interact with LLM
from langchain_openai import ChatOpenAI
from pydantic import SecretStr
from .config import settings
gpt4o = ChatOpenAI(
model="gpt-4o",
api_key=SecretStr(settings.OPENAI_API_KEY),
temperature=0,
max_tokens=8192,
timeout=60,
)
gpt4o_mini = ChatOpenAI(
model="gpt-4o-mini",
api_key=SecretStr(settings.OPENAI_API_KEY),
temperature=0,
max_tokens=8192,
timeout=60,
)
o1_mini = ChatOpenAI(
model="o1-mini",
api_key=SecretStr(settings.OPENAI_API_KEY),
temperature=1,
timeout=60,
)
|