From 225a136ad1f65948c14da5cf513d7ed2f2420156 Mon Sep 17 00:00:00 2001 From: Nil-Andreu Date: Sun, 8 Jan 2023 20:58:19 +0100 Subject: [PATCH] [NEW] Refactor name of message_embedding object --- backend/oasst_backend/prompt_repository.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/backend/oasst_backend/prompt_repository.py b/backend/oasst_backend/prompt_repository.py index 919da19b..3feb77bc 100644 --- a/backend/oasst_backend/prompt_repository.py +++ b/backend/oasst_backend/prompt_repository.py @@ -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: