wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-06-27 16:10:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix duplicate ack_task api-calls

Browse Source
This commit is contained in:
Andreas Köpf
2022-12-22 18:58:19 +01:00
parent 8a48722e72
commit 3a96cb062b
2 changed files with 1 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/oasst_backend/api/v1/tasks.py
+1 -1
View File
@@ -169,8 +169,8 @@ def acknowledge_task(
pr = PromptRepository(db, api_client, user=None)
# here we store the post id in the database for the task
pr.bind_frontend_post_id(task_id=task_id, post_id=ack_request.post_id)
logger.info(f"Frontend acknowledges task {task_id=}, {ack_request=}.")
pr.bind_frontend_post_id(task_id=task_id, post_id=ack_request.post_id)
except Exception:
logger.exception("Failed to acknowledge task.")
bot/task_handlers.py
-8
View File
@@ -86,13 +86,11 @@ class SummarizeStoryHandler(ChannelTaskBase):
thread_name: str = "Summaries"
async def send_first_message(self) -> discord.message:
expiry_time = discord_timestamp(self.expiry_date, DiscordTimestampStyle.long_time)
expiry_relatve = discord_timestamp(self.expiry_date, DiscordTimestampStyle.relative_time)
msg = await self.bot.post_template(
"task_summarize_story_teaser.msg", task=self.task, expiry_time=expiry_time, expiry_relatve=expiry_relatve
)
self.backend.ack_task(self.task.id, str(msg.id))
return msg
async def on_thread_created(self, thread: discord.Thread) -> None:
@@ -110,7 +108,6 @@ class InitialPromptHandler(ChannelTaskBase):
async def send_first_message(self) -> discord.message:
msg = await self.bot.post_template("task_initial_prompt.msg", task=self.task)
self.backend.ack_task(self.task.id, str(msg.id))
return msg
async def handler_loop(self):
@@ -125,7 +122,6 @@ class UserReplyHandler(ChannelTaskBase):
async def send_first_message(self) -> discord.message:
msg = await self.bot.post_template("task_user_reply.msg", task=self.task)
self.backend.ack_task(self.task.id, str(msg.id))
return msg
async def handler_loop(self):
@@ -140,7 +136,6 @@ class AssistantReplyHandler(ChannelTaskBase):
async def send_first_message(self) -> discord.message:
msg = await self.bot.post_template("task_assistant_reply.msg", task=self.task)
self.backend.ack_task(self.task.id, str(msg.id))
return msg
async def handler_loop(self):
@@ -160,7 +155,6 @@ class RankInitialPromptsHandler(ChannelTaskBase):
async def send_first_message(self) -> discord.message:
msg = await self.bot.post_template("task_rank_initial_prompts.msg", task=self.task)
self.backend.ack_task(self.task.id, str(msg.id))
return msg
async def handler_loop(self):
@@ -176,7 +170,6 @@ class RankConversationsHandler(ChannelTaskBase):
async def send_first_message(self) -> discord.message:
msg = await self.bot.post_template("task_rank_conversation_replies.msg", task=self.task)
self.backend.ack_task(self.task.id, str(msg.id))
return msg
async def handler_loop(self):
@@ -215,7 +208,6 @@ class RateSummaryHandler(ChannelTaskBase):
async def send_first_message(self) -> discord.message:
msg = await self.bot.post("first message")
self.backend.ack_task(self.task.id, str(msg.id))
return msg
async def on_thread_created(self, thread: discord.Thread) -> None: