优化代码，提高解耦

04_langchain_basics/01_models.py

@@ -1,15 +1,10 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-
+from config import OPENAI_API_KEY
 from langchain_openai import ChatOpenAI
 
 # 初始化模型
 llm = ChatOpenAI(
     model="deepseek-chat",
-    api_key=api_key,
+    api_key=OPENAI_API_KEY,
     base_url="https://api.deepseek.com"
 )
 

04_langchain_basics/02_prompt.py

@@ -1,10 +1,3 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-
-
 from langchain_core.prompts import ChatPromptTemplate
 
 # 定义提示词模板(推荐写法)

04_langchain_basics/03_chain.py

@@ -1,10 +1,4 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-
-
+from config import OPENAI_API_KEY
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_openai import ChatOpenAI
 from langchain_core.output_parsers import StrOutputParser
@@ -18,7 +12,7 @@ prompt = ChatPromptTemplate.from_messages([
 # 初始化模型
 llm = ChatOpenAI(
     model="deepseek-chat",
-    api_key=api_key,
+    api_key=OPENAI_API_KEY,
     base_url="https://api.deepseek.com"
 )
 

04_langchain_basics/04_memory.py

@@ -1,18 +1,9 @@
-import os
-from dotenv import load_dotenv
-
-
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-
-
-
 from langchain_core.prompts import ChatPromptTemplate,MessagesPlaceholder
 from langchain_openai import ChatOpenAI
 from langchain_core.output_parsers import StrOutputParser
 from langchain_community.chat_message_histories import ChatMessageHistory
 from langchain_core.runnables import RunnableWithMessageHistory
-
+from config import OPENAI_API_KEY
 
 prompt = ChatPromptTemplate.from_messages([
     ("system", "你非常可爱，说话末尾会带个喵"),
@@ -21,7 +12,7 @@ prompt = ChatPromptTemplate.from_messages([
 ])
 llm = ChatOpenAI(
     model="deepseek-chat",
-    api_key=api_key,
+    api_key=OPENAI_API_KEY,
     base_url="https://api.deepseek.com"
 )
 parser = StrOutputParser()
@@ -58,32 +49,3 @@ while 1:
         config={"configurable":{"session_id":session_id}}
     )
     print(f'AI:{response}')
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-

04_langchain_basics/05_practice.py

@@ -1,13 +1,4 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-
-def create_client():
-    api_key = os.getenv("OPENAI_API_KEY")
-    return api_key
-
-
+from config import OPENAI_API_KEY
 from langchain_core.prompts import ChatPromptTemplate,MessagesPlaceholder
 from langchain_openai import ChatOpenAI
 from langchain_core.output_parsers import StrOutputParser
@@ -44,7 +35,7 @@ def main():
     # 此处集中配置LLM
     llm = ChatOpenAI(
         model="deepseek-chat",
-        api_key=create_client(),
+        api_key=OPENAI_API_KEY,
         base_url="https://api.deepseek.com"
     )
     prompt = """你是‘小智’，一位专业、耐心且记忆力出色的 AI 助手。