fix: use current attachments for follow-up questions instead of first… (#1167)
Some checks are pending
CodeQL Advanced / Analyze (actions) (push) Waiting to run
CodeQL Advanced / Analyze (javascript-typescript) (push) Waiting to run
CodeQL Advanced / Analyze (python) (push) Waiting to run
Pre-commit / pre-commit (push) Waiting to run
Test / Run Python Tests (push) Waiting to run

Co-authored-by: Wendong-Fan <w3ndong.fan@gmail.com>
Co-authored-by: Wendong-Fan <133094783+Wendong-Fan@users.noreply.github.com>
This commit is contained in:
Tong Chen 2026-02-06 20:48:36 +08:00 committed by GitHub
parent 941da81d74
commit 09200a8cf6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 108 additions and 40 deletions

View file

@ -45,6 +45,7 @@ from app.service.task import (
ActionSkipTaskData,
ActionStopData,
ActionSupplementData,
ImprovePayload,
delete_task_lock,
get_or_create_task_lock,
get_task_lock,
@ -224,7 +225,13 @@ async def post(data: Chat, request: Request):
# Put initial action in queue to start processing
await task_lock.put_queue(
ActionImproveData(data=data.question, new_task_id=data.task_id)
ActionImproveData(
data=ImprovePayload(
question=data.question,
attaches=data.attaches or [],
),
new_task_id=data.task_id,
)
)
chat_logger.info(
@ -331,7 +338,13 @@ def improve(id: str, data: SupplementChat):
asyncio.run(
task_lock.put_queue(
ActionImproveData(data=data.question, new_task_id=data.task_id)
ActionImproveData(
data=ImprovePayload(
question=data.question,
attaches=data.attaches or [],
),
new_task_id=data.task_id,
)
)
)
chat_logger.info(