wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-08-17 11:13:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fetch conversation for seed data tasks, minor model fixes (#485)

Browse Source 
* Fetch conversation for seed data, fix models, remove redundant payload type checks
This commit is contained in:
Andreas Köpf
2023-01-07 15:59:54 +01:00
committed by GitHub
parent eaefa68dea
commit 96d6717be4
9 changed files with 61 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/main.py
+10 -5
View File
@@ -132,12 +132,17 @@ if settings.DEBUG_USE_SEED_DATA:
parent_message = pr.fetch_message_by_frontend_message_id(
msg.parent_message_id, fail_if_missing=True
)
task = pr.store_task(
protocol_schema.AssistantReplyTask(
conversation=protocol_schema.Conversation(
messages=[protocol_schema.ConversationMessage(text="dummy", is_assistant=False)]
conversation_messages = pr.fetch_message_conversation(parent_message)
conversation = protocol_schema.Conversation(
messages=[
protocol_schema.ConversationMessage(
text=msg.text, is_assistant=msg.role == "assistant"
)
),
for msg in conversation_messages
]
)
task = pr.store_task(
protocol_schema.AssistantReplyTask(conversation=conversation),
message_tree_id=parent_message.message_tree_id,
parent_message_id=parent_message.id,
)