muryshev commited on
Commit
e11aef6
·
1 Parent(s): fd78d64
Files changed (1) hide show
  1. routes/llm.py +3 -3
routes/llm.py CHANGED
@@ -127,7 +127,7 @@ def collapse_history_to_first_message(chat_request: ChatRequest) -> ChatRequest:
127
 
128
  # Собираем историю в одну строку
129
  collapsed_content = []
130
- collapsed_content.append("<history>\n")
131
  for msg in chat_request.history[:-1]:
132
  if msg.content.strip():
133
  tabulated_content = msg.content.strip().replace("\n", "\n\t\t")
@@ -146,12 +146,12 @@ def collapse_history_to_first_message(chat_request: ChatRequest) -> ChatRequest:
146
  tabulated_reasoning = last_user_message.reasoning.strip().replace("\n", "\n\t\t")
147
  tabulated_search_results = last_user_message.searchResults.strip().replace("\n", "\n\t\t")
148
  # collapsed_content.append(f"\t<reasoning>\n\t\t{tabulated_reasoning}\n\t</reasoning>\n")
149
- collapsed_content.append(f"\t<user>\n\t\t{tabulated_content}\n</user>\n")
150
  collapsed_content.append(f"\t<search-results>\n\t\t{tabulated_search_results}\n\t</search-results>\n")
 
151
 
152
  collapsed_content.append("</last-request>\n")
153
 
154
- collapsed_content.append("assistant:\n")
155
  new_content = "".join(collapsed_content)
156
 
157
  new_message = Message(
 
127
 
128
  # Собираем историю в одну строку
129
  collapsed_content = []
130
+ collapsed_content.append("<INPUT><history>\n")
131
  for msg in chat_request.history[:-1]:
132
  if msg.content.strip():
133
  tabulated_content = msg.content.strip().replace("\n", "\n\t\t")
 
146
  tabulated_reasoning = last_user_message.reasoning.strip().replace("\n", "\n\t\t")
147
  tabulated_search_results = last_user_message.searchResults.strip().replace("\n", "\n\t\t")
148
  # collapsed_content.append(f"\t<reasoning>\n\t\t{tabulated_reasoning}\n\t</reasoning>\n")
 
149
  collapsed_content.append(f"\t<search-results>\n\t\t{tabulated_search_results}\n\t</search-results>\n")
150
+ collapsed_content.append(f"\t<user>\n\t\t{tabulated_content}\n</user>\n")
151
 
152
  collapsed_content.append("</last-request>\n")
153
 
154
+ collapsed_content.append("</INPUT><OUTPUT>\n")
155
  new_content = "".join(collapsed_content)
156
 
157
  new_message = Message(