fix: Added a default value for rating 🌟

NimbleBoxAI · Apr 19, 2023 · 744f953 · 744f953
1 parent 533bd95
commit 744f953
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/server/database.py b/server/database.py
@@ -93,8 +93,8 @@ class Prompt(Base):
     input_prompt = Column(Text, nullable=False)
     response = Column(Text, nullable=True)
     gpt_rating = Column(String(5), nullable=True)
-    user_rating = Column(Enum(PromptRating), nullable=True)
-    chatbot_user_rating = Column(Enum(PromptRating), nullable=True)
+    user_rating = Column(Enum(PromptRating), nullable=True, default=PromptRating.UNRATED)
+    chatbot_user_rating = Column(Enum(PromptRating), nullable=True, default=PromptRating.UNRATED)
     time_taken = Column(Float, nullable=True)
     num_tokens = Column(Integer, nullable=True)
     created_at = Column(DateTime, nullable=False)

diff --git a/server/database_constants.py b/server/database_constants.py
@@ -7,6 +7,7 @@ class PromptRating(Enum):
     SAD = 1
     NEUTRAL = 2
     HAPPY = 3
+    UNRATED = -1
 
 
 SORT_BY_TIME_TAKEN = "time_taken"