使用pip 安装依赖
pip install openai
pip install wxauto
编写运行代码
from openai import OpenAI
from wxauto import WeChat
import time
class WeChatBot:
def __init__(self):
self.client = OpenAI(
api_key="you-deepseek-key", # DeepSeek的API密钥
base_url="https://api.deepseek.com" # DeepSeek API地址
)
self.wx = WeChat()
self.listen_list = ["admin"] # 要监听的好友/群聊列表
for chat in self.listen_list:
self.wx.AddListenChat(chat)
def __ask_deepseek(self, msg):
response = self.client.chat.completions.create(
model="deepseek-chat", # DeepSeek模型名称
messages=[
{"role": "system", "content": "假设你是一名情场高手,性格幽默风趣,你的任务就是对话过程中让他对你产生好感,每段对话不超过20个字"},
{"role": "user", "content": msg}
],
temperature=0.7,
stream=False
)
return response.choices[0].message.content
def run(self):
check_interval = 2 # 消息检查间隔(秒)
while True:
msgs = self.wx.GetListenMessage()
for chat in msgs:
msg_list = msgs.get(chat)
for msg_item in msg_list:
if msg_item.type == 'friend':
reply = self.__ask_deepseek(msg_item.content)
print(f"[收到] {msg_item.sender}: {msg_item.content}")
print(f"[回复] {msg_item.sender}: {reply}")
chat.SendMsg(reply)
time.sleep(check_interval)
if __name__ == "__main__":
bot = WeChatBot()
bot.run()