Skip to content

Commit effc337

Browse files
committed
fix: chatbot metrics
1 parent 2cc5097 commit effc337

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

server/api/metrics.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import database
2+
from database_constants import PromptRating
23
from sqlalchemy.orm import Session
34
from fastapi import APIRouter, Depends, Header
45
from fastapi import HTTPException
@@ -117,16 +118,16 @@ def get_all_chatbot_ratings(token: Annotated[str, Header()], db: Session = Depen
117118
prompts = get_prompts_from_chatbot_id(chatbot_id)
118119
for prompt in prompts:
119120
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
122123
total_chatbot_user_ratings += 1
123124
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
126127
total_developer_ratings += 1
127128
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)
130131
total_openai_ratings += 1
131132
if prompt.num_tokens is not None:
132133
total_tokens_processed += prompt.num_tokens

0 commit comments

Comments
 (0)