mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-03-26 01:31:51 +01:00
f25e1e80f6
* Add option to not re-index * Add quantizaton / dimensionality override support * Fix build / ut
97 lines
2.2 KiB
Python
97 lines
2.2 KiB
Python
from enum import Enum as PyEnum
|
|
|
|
|
|
class IndexingStatus(str, PyEnum):
|
|
NOT_STARTED = "not_started"
|
|
IN_PROGRESS = "in_progress"
|
|
SUCCESS = "success"
|
|
CANCELED = "canceled"
|
|
FAILED = "failed"
|
|
COMPLETED_WITH_ERRORS = "completed_with_errors"
|
|
|
|
def is_terminal(self) -> bool:
|
|
terminal_states = {
|
|
IndexingStatus.SUCCESS,
|
|
IndexingStatus.COMPLETED_WITH_ERRORS,
|
|
IndexingStatus.CANCELED,
|
|
IndexingStatus.FAILED,
|
|
}
|
|
return self in terminal_states
|
|
|
|
|
|
class IndexingMode(str, PyEnum):
|
|
UPDATE = "update"
|
|
REINDEX = "reindex"
|
|
|
|
|
|
class SyncType(str, PyEnum):
|
|
DOCUMENT_SET = "document_set"
|
|
USER_GROUP = "user_group"
|
|
CONNECTOR_DELETION = "connector_deletion"
|
|
PRUNING = "pruning" # not really a sync, but close enough
|
|
EXTERNAL_PERMISSIONS = "external_permissions"
|
|
EXTERNAL_GROUP = "external_group"
|
|
|
|
def __str__(self) -> str:
|
|
return self.value
|
|
|
|
|
|
class SyncStatus(str, PyEnum):
|
|
IN_PROGRESS = "in_progress"
|
|
SUCCESS = "success"
|
|
FAILED = "failed"
|
|
CANCELED = "canceled"
|
|
|
|
def is_terminal(self) -> bool:
|
|
terminal_states = {
|
|
SyncStatus.SUCCESS,
|
|
SyncStatus.FAILED,
|
|
}
|
|
return self in terminal_states
|
|
|
|
|
|
# Consistent with Celery task statuses
|
|
class TaskStatus(str, PyEnum):
|
|
PENDING = "PENDING"
|
|
STARTED = "STARTED"
|
|
SUCCESS = "SUCCESS"
|
|
FAILURE = "FAILURE"
|
|
|
|
|
|
class IndexModelStatus(str, PyEnum):
|
|
PAST = "PAST"
|
|
PRESENT = "PRESENT"
|
|
FUTURE = "FUTURE"
|
|
|
|
def is_current(self) -> bool:
|
|
return self == IndexModelStatus.PRESENT
|
|
|
|
|
|
class ChatSessionSharedStatus(str, PyEnum):
|
|
PUBLIC = "public"
|
|
PRIVATE = "private"
|
|
|
|
|
|
class ConnectorCredentialPairStatus(str, PyEnum):
|
|
ACTIVE = "ACTIVE"
|
|
PAUSED = "PAUSED"
|
|
DELETING = "DELETING"
|
|
INVALID = "INVALID"
|
|
|
|
def is_active(self) -> bool:
|
|
return self == ConnectorCredentialPairStatus.ACTIVE
|
|
|
|
|
|
class AccessType(str, PyEnum):
|
|
PUBLIC = "public"
|
|
PRIVATE = "private"
|
|
SYNC = "sync"
|
|
|
|
|
|
class EmbeddingPrecision(str, PyEnum):
|
|
# matches vespa tensor type
|
|
# only support float / bfloat16 for now, since there's not a
|
|
# good reason to specify anything else
|
|
BFLOAT16 = "bfloat16"
|
|
FLOAT = "float"
|