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

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +10 -0
config.py CHANGED
@@ -2,6 +2,16 @@
2
  import os
3
  from typing import List, Dict
4
  from openai import OpenAI
 
 
 
 
 
 
 
 
 
 
5
 
6
  # ---------- 环境变量 & OpenAI Client ----------
7
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
2
  import os
3
  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")