|
1 | 1 | import database |
| 2 | +from database_constants import PromptRating |
2 | 3 | from sqlalchemy.orm import Session |
3 | 4 | from fastapi import APIRouter, Depends, Header |
4 | 5 | from fastapi import HTTPException |
@@ -117,16 +118,16 @@ def get_all_chatbot_ratings(token: Annotated[str, Header()], db: Session = Depen |
117 | 118 | prompts = get_prompts_from_chatbot_id(chatbot_id) |
118 | 119 | for prompt in prompts: |
119 | 120 | if prompt.chatbot_user_rating is not None: |
120 | | - chatbot_user_ratings.append(prompt.chatbot_user_rating) |
121 | | - sum_chatbot_user_ratings += prompt.chatbot_user_rating |
| 121 | + chatbot_user_ratings.append(PromptRating(prompt.chatbot_user_rating).value) |
| 122 | + sum_chatbot_user_ratings += PromptRating(prompt.chatbot_user_rating).value |
122 | 123 | total_chatbot_user_ratings += 1 |
123 | 124 | if prompt.user_rating is not None: |
124 | | - developer_ratings.append(prompt.user_rating) |
125 | | - sum_developer_ratings += prompt.user_rating |
| 125 | + developer_ratings.append(PromptRating(prompt.user_rating).value) |
| 126 | + sum_developer_ratings += PromptRating(prompt.user_rating).value |
126 | 127 | total_developer_ratings += 1 |
127 | 128 | if prompt.gpt_rating is not None: |
128 | | - openai_ratings.append(prompt.gpt_rating) |
129 | | - sum_openai_ratings += prompt.gpt_rating |
| 129 | + openai_ratings.append(int(prompt.gpt_rating)) |
| 130 | + sum_openai_ratings += int(prompt.gpt_rating) |
130 | 131 | total_openai_ratings += 1 |
131 | 132 | if prompt.num_tokens is not None: |
132 | 133 | total_tokens_processed += prompt.num_tokens |
|
0 commit comments