mirror of
https://github.com/jingyaogong/minimind.git
synced 2026-04-25 08:48:16 +08:00
[fix] messages num
This commit is contained in:
parent
42a4e8c86a
commit
a8455ca8a3
@ -7,14 +7,17 @@ client = OpenAI(
|
||||
stream = True
|
||||
conversation_history_origin = []
|
||||
conversation_history = conversation_history_origin.copy()
|
||||
history_messages_num = 2 # 设置为偶数(Q+A),为0则每次不携带历史对话进行独立QA
|
||||
history_messages_num = 0 # 必须设置为偶数(Q+A),为0则不携带历史对话
|
||||
while True:
|
||||
query = input('[Q]: ')
|
||||
conversation_history.append({"role": "user", "content": query})
|
||||
response = client.chat.completions.create(
|
||||
model="minimind",
|
||||
messages=conversation_history[-history_messages_num:],
|
||||
stream=stream
|
||||
messages=conversation_history[-(history_messages_num or 1):],
|
||||
stream=stream,
|
||||
temperature=0.7,
|
||||
max_tokens=2048,
|
||||
top_p=0.9
|
||||
)
|
||||
if not stream:
|
||||
assistant_res = response.choices[0].message.content
|
||||
|
||||
Loading…
Reference in New Issue
Block a user