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

[NEW] Refactor name of message_embedding object

Browse Source
This commit is contained in:
Nil-Andreu
2023-01-08 20:58:19 +01:00
parent 19eee6be58
commit 225a136ad1
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/oasst_backend/prompt_repository.py
+4 -4
View File
@@ -233,11 +233,11 @@ class PromptRepository:
if None in (message_id, model, embedding):
raise OasstError("Paramters missing to add embedding", OasstErrorCode.GENERIC_ERROR)
model = MessageEmbedding(message_id=message_id, model=model, embedding=embedding)
self.db.add(model)
message_embedding = MessageEmbedding(message_id=message_id, model=model, embedding=embedding)
self.db.add(message_embedding)
self.db.commit()
self.db.refresh(model)
return model
self.db.refresh(message_embedding)
return message_embedding
def insert_reaction(self, task_id: UUID, payload: db_payload.ReactionPayload) -> MessageReaction:
if self.user_id is None: