File tree

2 files changed

+23
-1
lines changed

2 files changed

+23
-1
lines changed
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,6 @@ fastapi==0.115.12
22
uvicorn[standard]
33
requests
44
sqlmodel
5-
psycopg[binary]
5+
psycopg[binary]
6+
langchain
7+
langchain-openai
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
import os
2+
3+
from langchain_openai import ChatOpenAI
4+
5+
6+
OPENAI_BASE_URL = os.environ.get('OPENAI_BASE_URL') or None
7+
OPENAI_MODEL_NAME = os.environ.get('OPENAI_MODEL_NAME') or 'gpt-4o-mini'
8+
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
9+
if not OPENAI_API_KEY:
10+
raise NotImplementedError("`OPENAI_API_KEY` is required")
11+
12+
13+
openai_params = {
14+
"model": OPENAI_MODEL_NAME,
15+
"api_key": OPENAI_API_KEY
16+
}
17+
if OPENAI_BASE_URL:
18+
openai_params['base_url'] = OPENAI_BASE_URL
19+
20+
llm_base = ChatOpenAI(**openai_params)

0 commit comments

Comments
 (0)