优化代码，提高解耦

05_langchain_advanced/01_define_toolbox.py

@@ -1,10 +1,3 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-
-
 from langchain_core.tools import tool
 
 @tool

05_langchain_advanced/02_general_agent.py

@@ -1,20 +1,15 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-
-
-import os
+from config import OPENAI_API_KEY
 from langchain_openai import ChatOpenAI
-from langchain.agents import AgentExecutor, create_tool_calling_agent
+# 在LangChain 1.0+版本中，以下俩组件移到了langchain-classic包中
+from langchain_classic.agents import AgentExecutor
+from langchain_classic.agents import create_tool_calling_agent
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.tools import tool # 导入 @tool
 
 # 配置LLM
 llm = ChatOpenAI(
     model="deepseek-chat",
-    api_key=api_key,
+    api_key=OPENAI_API_KEY,
     base_url="https://api.deepseek.com"
 )
 

05_langchain_advanced/03_sql_agent.py

@@ -1,18 +1,14 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-
+from config import OPENAI_API_KEY
 import sqlite3
 from langchain_openai import ChatOpenAI
 from langchain_community.agent_toolkits import create_sql_agent
 from langchain_community.utilities import SQLDatabase  # 导入 SQLDatabase
+import os
 
 # 配置llm
 llm = ChatOpenAI(
     model="deepseek-chat",
-    api_key=api_key,
+    api_key=OPENAI_API_KEY,
     base_url="https://api.deepseek.com"
 )
 

05_langchain_advanced/04_memory_general_agent.py

@@ -1,13 +1,7 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-
-
-import os
+from config import OPENAI_API_KEY
 from langchain_openai import ChatOpenAI
-from langchain.agents import AgentExecutor, create_tool_calling_agent
+from langchain_classic.agents import AgentExecutor
+from langchain_classic.agents import create_tool_calling_agent
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.tools import tool # 导入 @tool
 from langchain_community.chat_message_histories import ChatMessageHistory
@@ -17,7 +11,7 @@ from langchain_core.runnables import RunnableWithMessageHistory
 # 配置llm
 llm = ChatOpenAI(
     model="deepseek-chat",
-    api_key=api_key,
+    api_key=OPENAI_API_KEY,
     base_url="https://api.deepseek.com"
 )
 # 配置prompt(新增俩占位符 一个为对话历史记录，一个为agent的思考过程)

05_langchain_advanced/05_caching.py

@@ -1,19 +1,13 @@
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-api_key = os.getenv("OPENAI_API_KEY")
-
-
+from config import OPENAI_API_KEY
 import time
 from langchain_openai import ChatOpenAI
-from langchain.globals import set_llm_cache
+from langchain_core.globals import set_llm_cache
 from langchain_community.cache import InMemoryCache
 
 # 配置llm
 llm = ChatOpenAI(
     model="deepseek-chat",
-    api_key=api_key,
+    api_key=OPENAI_API_KEY,
     base_url="https://api.deepseek.com"
 )
 # 设置全局缓存

05_langchain_advanced/06_streaming.py

@@ -1,23 +1,18 @@
-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
-from langchain.agents import AgentExecutor, create_tool_calling_agent
+from langchain_classic.agents import AgentExecutor
+from langchain_classic.agents import create_tool_calling_agent
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.tools import tool # 导入 @tool
 from langchain_community.chat_message_histories import ChatMessageHistory
 from langchain_core.runnables import RunnableWithMessageHistory
-from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
+from langchain_core.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
 
 # 配置llm
 llm = ChatOpenAI(
     model="deepseek-chat",
-    api_key=api_key,
+    api_key=OPENAI_API_KEY,
     base_url="https://api.deepseek.com",
     streaming=True,
     callbacks=[StreamingStdOutCallbackHandler()]