update prepare_messages to work with multi turn
Browse files
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 |
-
|
70 |
-
|
71 |
-
{"role": "user", "content":
|
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 |
|