本文是基于LangChain的Python库结合DeepSeek进行多轮次对话; 对于测开人员来说,可以集成到自己的自动化测试框架中,自动生成测试用例。
首先看下LangChain的简单说明,大佬就不用看了。网上也有很多资料,这里简单说明下,针对没有基础的Tester。
pip install --upgrade langchain langchain-community deepseek python-dotenv
DEEPSEEK_API_KEY=你的API密钥
import os
from dotenv import load_dotenv
from langchain.chains import LLMChain
from langchain.prompts import ChatPromptTemplate
from langchain_community.llms import Deepseek
load_dotenv()
DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
"""创建带有记忆功能的DeepSeek对话链,支持多轮对话"""
# 确认DeepSeek的API基础地址和模型名称
llm = ChatOpenAI(
model_name="deepseek-chat",
openai_api_base="https://api.deepseek.com/v1",
openai_api_key=DEEPSEEK_API_KEY,
temperature=0.7,
max_tokens=1024
)
# 创建对话提示模板
prompt = PromptTemplate(
input_variables=["history", "question"],
template="历史对话: {history}\n用户现在的问题: {question}\n请根据历史对话内容,用简洁明了的语言回答:"
)
# 创建对话记忆
memory = ConversationBufferMemory(memory_key="history")
# 创建带有记忆功能的LLM链
return LLMChain(
llm=llm,
prompt=prompt,
memory=memory,
verbose=False
)
import os
from dotenv import load_dotenv
from langchain.chains import LLMChain
from langchain.prompts import PromptTemplate
from langchain_community.chat_models import ChatOpenAI
from langchain.memory import ConversationBufferMemory
from langchain.schema import HumanMessage, AIMessage
# 加载环境变量
load_dotenv()
# 从环境变量获取DeepSeek API密钥
DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
def create_deepseek_chain_with_memory():
"""创建带有记忆功能的DeepSeek对话链,支持多轮对话"""
# 确认DeepSeek的API基础地址和模型名称
llm = ChatOpenAI(
model_name="deepseek-chat",
openai_api_base="https://api.deepseek.com/v1",
openai_api_key=DEEPSEEK_API_KEY,
temperature=0.7,
max_tokens=1024
)
# 创建对话提示模板
prompt = PromptTemplate(
input_variables=["history", "question"],
template="历史对话: {history}\n用户现在的问题: {question}\n请根据历史对话内容,用简洁明了的语言回答:"
)
# 创建对话记忆
memory = ConversationBufferMemory(memory_key="history")
# 创建带有记忆功能的LLM链
return LLMChain(
llm=llm,
prompt=prompt,
memory=memory,
verbose=False
)
def chat_with_deepseek():
"""与DeepSeek模型进行多轮对话"""
print("DeepSeek对话助手(输入'退出'结束对话)")
# 检查API密钥是否存在
if not DEEPSEEK_API_KEY:
print("错误: 未找到DEEPSEEK_API_KEY,请检查.env文件")
return
try:
chain = create_deepseek_chain_with_memory()
while True:
user_input = input("\n你: ")
if user_input.lower() == "退出":
print("再见!")
break
try:
response = chain.run(question=user_input)
print(f"DeepSeek: {response}")
except Exception as e:
print(f"对话错误: {str(e)}")
except Exception as e:
print(f"初始化错误: {str(e)}")
if __name__ == "__main__":
chat_with_deepseek()
DeepSeek 的 API 和 OpenAI 的 API 不兼容,如果你直接使用 openai 库但未修改 base_url,就会默认调用 OpenAI 的接口,导致 402 错误(OpenAI 的 API 需要付费)。 正确调用 DeepSeek API 的方式:
from openai import OpenAI
# 使用 DeepSeek 的 API 地址和你的 API Key
client = OpenAI(
api_key="your_deepseek_api_key_here", # 替换成你的 DeepSeek API Key
base_url="https://api.deepseek.com/v1", # DeepSeek 的 API 地址
)
response = client.chat.completions.create(
model="deepseek-chat", # 检查 DeepSeek 支持的最新模型名称
messages=[{"role": "user", "content": "你好!"}],
temperature=0.3,
)
print(response.choices[0].message.content)
关键点:base_url 必须指向 DeepSeek 的 API 地址(如 https://api.deepseek.com/v1)。 api_key 必须是 DeepSeek 提供的 API Key(不是 OpenAI 的 Key)。 model 要使用 DeepSeek 支持的模型(如 deepseek-chat)。
即使 DeepSeek 提供免费额度,也可能:免费额度已用完(比如每月限额)。 未正确申请 API 权限(部分 API 需要申请白名单)。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。