plaguss commited on
Commit
69d4f9d
·
verified ·
1 Parent(s): 08d388a

update prepare_messages to work with multi turn

Browse files
Files changed (1) hide show
  1. README.md +10 -4
README.md CHANGED
@@ -66,10 +66,16 @@ def prepare_messages(query: str, tools: Optional[dict[str, any]] = None) -> list
66
  if tools is None:
67
  tools = []
68
 
69
- return [
70
- {"role": "system", "content": SYSTEM_PROMPT},
71
- {"role": "user", "content": prompt.render(tools=json.dumps(tools), query=query)}
72
- ]
 
 
 
 
 
 
73
 
74
  ````
75
 
 
66
  if tools is None:
67
  tools = []
68
 
69
+ if conversation_history:
70
+ messages = conversation_history.copy()
71
+ messages.append({"role": "user", "content": query})
72
+ else:
73
+ messages = [
74
+ {"role": "system", "content": system_prompt},
75
+ {"role": "user", "content": prompt.render(tools=json.dumps(tools), query=query)}
76
+ ]
77
+
78
+ return messages
79
 
80
  ````
81