Skip to content

Commit

Permalink
chore(ai-help): rename message status
Browse files Browse the repository at this point in the history
  • Loading branch information
caugner committed Feb 29, 2024
1 parent 7013e45 commit 4572cd4
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 9 deletions.
6 changes: 3 additions & 3 deletions migrations/2024-02-20-093804_ai_help_metadata/up.sql
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
CREATE TYPE ai_help_message_status AS ENUM (
'success',
'search_error',
'open_ai_api_error',
'ai_api_error',
'completion_error',
'moderation_error',
'no_user_prompt_error',
'token_limit_error',
'stopped',
'timeout',
'user_stopped',
'user_timeout',
'unknown'
);

Expand Down
10 changes: 4 additions & 6 deletions src/db/types.rs
Original file line number Diff line number Diff line change
Expand Up @@ -202,23 +202,21 @@ pub enum EngineType {
pub enum AiHelpMessageStatus {
Success,
SearchError,
OpenAiApiError,
AiApiError,
CompletionError,
ModerationError,
NoUserPromptError,
TokenLimitError,
Stopped,
Timeout,
UserStopped,
UserTimeout,
#[default]
Unknown,
}

impl From<&AIError> for AiHelpMessageStatus {
fn from(e: &AIError) -> Self {
match e {
crate::ai::error::AIError::OpenAIError(_) => {
db::types::AiHelpMessageStatus::OpenAiApiError
}
crate::ai::error::AIError::OpenAIError(_) => db::types::AiHelpMessageStatus::AiApiError,
crate::ai::error::AIError::SqlXError(_) => db::types::AiHelpMessageStatus::SearchError,
crate::ai::error::AIError::FlaggedError => {
db::types::AiHelpMessageStatus::ModerationError
Expand Down

0 comments on commit 4572cd4

Please sign in to comment.