SarahXia0405 commited on
Commit
7bd3326
·
verified ·
1 Parent(s): e8634b4

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +17 -13
config.py CHANGED
@@ -4,28 +4,32 @@ from typing import List, Dict
4
  from openai import OpenAI
5
  from langchain_openai import ChatOpenAI, OpenAIEmbeddings
6
 
7
- llm_default = ChatOpenAI(
8
- model=DEFAULT_MODEL,
9
- temperature=0.5,
10
- timeout=20,
11
- )
12
-
13
- embedding_client = OpenAIEmbeddings(model=EMBEDDING_MODEL)
14
-
15
-
16
- # ---------- 环境变量 & OpenAI Client ----------
17
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
18
  if not OPENAI_API_KEY:
19
  raise RuntimeError(
20
  "OPENAI_API_KEY is not set. Please go to Settings → Secrets and add it."
21
  )
22
 
23
- client = OpenAI(api_key=OPENAI_API_KEY)
24
-
25
- # ---------- 模型配置 ----------
26
  DEFAULT_MODEL = "gpt-4.1-mini"
27
  EMBEDDING_MODEL = "text-embedding-3-small"
28
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  # ---------- 默认 GenAI 课程大纲 ----------
30
  DEFAULT_COURSE_TOPICS: List[str] = [
31
  "Week 0 – Welcome & What is Generative AI; course outcomes LO1–LO5.",
 
4
  from openai import OpenAI
5
  from langchain_openai import ChatOpenAI, OpenAIEmbeddings
6
 
7
+ # ---------- 环境变量 & OpenAI API Key ----------
 
 
 
 
 
 
 
 
 
8
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
9
  if not OPENAI_API_KEY:
10
  raise RuntimeError(
11
  "OPENAI_API_KEY is not set. Please go to Settings → Secrets and add it."
12
  )
13
 
14
+ # ---------- 模型配置(一定要在用之前定义) ----------
 
 
15
  DEFAULT_MODEL = "gpt-4.1-mini"
16
  EMBEDDING_MODEL = "text-embedding-3-small"
17
 
18
+ # ---------- OpenAI 原生 Client(老代码还在用) ----------
19
+ client = OpenAI(api_key=OPENAI_API_KEY)
20
+
21
+ # ---------- LangChain 封装的 LLM & Embedding ----------
22
+ # 以后如果你想用 LangChain 的 ChatOpenAI / OpenAIEmbeddings,可以直接用这两个
23
+ llm_default = ChatOpenAI(
24
+ model=DEFAULT_MODEL,
25
+ temperature=0.5,
26
+ timeout=20,
27
+ )
28
+
29
+ embedding_client = OpenAIEmbeddings(
30
+ model=EMBEDDING_MODEL,
31
+ )
32
+
33
  # ---------- 默认 GenAI 课程大纲 ----------
34
  DEFAULT_COURSE_TOPICS: List[str] = [
35
  "Week 0 – Welcome & What is Generative AI; course outcomes LO1–LO5.",