在langchain使用openai接口,但模型使用国内deepseek,但是要先cq,API-key才能用
代码如下:
import os
import langchain
from langchain.chat_models import ChatOpenAI
from langchain.agents import AgentType, initialize_agent
from langchain.tools import ShellTool
def main():
# 如果没有从环境变量中读取,则可以手动写入
#os.environ["OPENAI_API_KEY"] = "your_openai_api_key"
# 1. 构建 LLM(这里使用 OpenAI 的 ChatGPT)
llm = ChatOpenAI(
model_name="deepseek-chat", # 如 "gpt-3.5-turbo" 或其他
openai_api_base="https://api.deepseek.com",
openai_api_key="xxxxx",
temperature=0.7
)
# 2. 准备 Shell 工具
shell_tool = ShellTool()
# 3. 初始化 Agent
# - 这里我们使用 ZERO_SHOT_