Merge pull request #13675 from feddersen-group/task/remove_files_from_tasks
perf: remove files from tasks
This commit is contained in:
@@ -186,20 +186,9 @@ async def generate_title(
|
|||||||
else:
|
else:
|
||||||
template = DEFAULT_TITLE_GENERATION_PROMPT_TEMPLATE
|
template = DEFAULT_TITLE_GENERATION_PROMPT_TEMPLATE
|
||||||
|
|
||||||
messages = form_data["messages"]
|
|
||||||
|
|
||||||
# Remove reasoning details from the messages
|
|
||||||
for message in messages:
|
|
||||||
message["content"] = re.sub(
|
|
||||||
r"<details\s+type=\"reasoning\"[^>]*>.*?<\/details>",
|
|
||||||
"",
|
|
||||||
message["content"],
|
|
||||||
flags=re.S,
|
|
||||||
).strip()
|
|
||||||
|
|
||||||
content = title_generation_template(
|
content = title_generation_template(
|
||||||
template,
|
template,
|
||||||
messages,
|
form_data["messages"],
|
||||||
{
|
{
|
||||||
"name": user.name,
|
"name": user.name,
|
||||||
"location": user.info.get("location") if user.info else None,
|
"location": user.info.get("location") if user.info else None,
|
||||||
|
|||||||
@@ -977,6 +977,20 @@ async def process_chat_response(
|
|||||||
if message:
|
if message:
|
||||||
messages = get_message_list(message_map, message.get("id"))
|
messages = get_message_list(message_map, message.get("id"))
|
||||||
|
|
||||||
|
# Remove reasoning details and files from the messages.
|
||||||
|
# as get_message_list creates a new list, it does not affect
|
||||||
|
# the original messages outside of this handler
|
||||||
|
for message in messages:
|
||||||
|
message["content"] = re.sub(
|
||||||
|
r"<details\s+type=\"reasoning\"[^>]*>.*?<\/details>",
|
||||||
|
"",
|
||||||
|
message["content"],
|
||||||
|
flags=re.S,
|
||||||
|
).strip()
|
||||||
|
|
||||||
|
if message.get("files"):
|
||||||
|
message["files"] = []
|
||||||
|
|
||||||
if tasks and messages:
|
if tasks and messages:
|
||||||
if TASKS.TITLE_GENERATION in tasks:
|
if TASKS.TITLE_GENERATION in tasks:
|
||||||
if tasks[TASKS.TITLE_GENERATION]:
|
if tasks[TASKS.TITLE_GENERATION]:
|
||||||
|
|||||||
Reference in New Issue
Block a user