github
/
minimind
mirror of https://github.com/jingyaogong/minimind.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[fix] messages num

pull/825/head
jingyaogong 2026-01-04 11:03:16 +08:00
parent dbfeff23e3
commit 4138d7584c
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/chat_openai_api.py
View File

@ -7,14 +7,17 @@ client = OpenAI(
stream = True stream = True
conversation_history_origin = [] conversation_history_origin = []
conversation_history = conversation_history_origin.copy() conversation_history = conversation_history_origin.copy()
history_messages_num = 2 # 设置为偶数（Q+A），为0则每次不携带历史对话进行独立QA history_messages_num = 0 # 必须设置为偶数（Q+A），为0则不携带历史对话
while True: while True:
query = input('[Q]: ') query = input('[Q]: ')
conversation_history.append({"role": "user", "content": query}) conversation_history.append({"role": "user", "content": query})
response = client.chat.completions.create( response = client.chat.completions.create(
model="minimind", model="minimind",
messages=conversation_history[-history_messages_num:], messages=conversation_history[-(history_messages_num or 1):],
stream=stream stream=stream,
temperature=0.7,
max_tokens=2048,
top_p=0.9
) )
if not stream: if not stream:
assistant_res = response.choices[0].message.content assistant_res = response.choices[0].message.content