d0dc94e36d
新功能: - 创建AI聊天助手组件（AIChatWidget） - 支持文字+图片多模态对话 - 集成Gemini AI API - 支持图片编辑和内容生成 - 实时对话历史管理 - 优雅的UI设计，支持黑夜模式 技术实现: - 后端API代理Gemini API（/api/ai-chat） - 支持最多5张图片上传 - 流式响应处理 - 对话历史管理 - Base64图片处理 配置: - 在.env中添加Gemini API配置 - GEMINI_API_KEY - GEMINI_API_BASE_URL - GEMINI_MODEL 替换原弹幕功能，提供更强大的AI助手体验！
234 lines
8.4 KiB
Python
Executable File
234 lines
8.4 KiB
Python
Executable File
# coding:utf-8
|
|
class VikingDBServerException(Exception):
|
|
def __init__(self, code, request_id, message=None):
|
|
self.code = code
|
|
self.request_id = request_id
|
|
self.message = "message:{}, code:{}, request_id:{}".format(message, self.code, self.request_id)
|
|
|
|
def __str__(self):
|
|
return self.message
|
|
|
|
|
|
class UnauthorizedException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class NoPermissionException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class InvalidRequestException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class CollectionExistException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class CollectionNotExistException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class OperationNotAllowedException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class IndexExistException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class IndexNotExistException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class QueryOpFailedException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class DataNotFoundException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class DelOpFailedException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class UpsertOpFailedException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class TokenMismatchException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class InvalidQueryVecException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class InvalidPrimaryKeyException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class InvalidPartitionException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class InvalidScalarCondException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class InvalidProxyServiceException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class IndexRecallException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class IndexFetchDataException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class IndexNotReadyException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class APINotImplementedException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class CalcEmbeddingFailedException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class ListEmbeddingModelsException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
|
|
class QuotaLimiterException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
class RerankException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
class CollectionAliasNotExistException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
class UserNoOrderException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
class UserOverdueException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
class HTTPErrException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
class TaskNotFoundException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
super().__init__(code, request_id, message)
|
|
|
|
class VikingDBException(VikingDBServerException):
|
|
def __init__(self, code, request_id, message=None):
|
|
self.code = code
|
|
self.request_id = request_id
|
|
if message is not None:
|
|
self.message = message
|
|
else:
|
|
self.message = "unknown error, please contact customer service, request_id:{}".format(self.request_id)
|
|
|
|
def __str__(self):
|
|
return self.message
|
|
|
|
|
|
ERRCODE_EXCEPTION = {
|
|
1000001: UnauthorizedException,
|
|
1000002: NoPermissionException,
|
|
1000003: InvalidRequestException,
|
|
1000004: CollectionExistException,
|
|
1000005: CollectionNotExistException,
|
|
1000006: OperationNotAllowedException,
|
|
1000007: IndexExistException,
|
|
1000008: IndexNotExistException,
|
|
1000010: QueryOpFailedException,
|
|
1000011: DataNotFoundException,
|
|
1000013: DelOpFailedException,
|
|
1000014: UpsertOpFailedException,
|
|
1000015: TokenMismatchException,
|
|
1000016: InvalidQueryVecException,
|
|
1000017: InvalidPrimaryKeyException,
|
|
1000018: InvalidPartitionException,
|
|
1000019: InvalidScalarCondException,
|
|
1000020: InvalidProxyServiceException,
|
|
1000021: IndexRecallException,
|
|
1000022: IndexFetchDataException,
|
|
1000023: IndexNotReadyException,
|
|
1000024: APINotImplementedException,
|
|
1000025: CalcEmbeddingFailedException,
|
|
1000026: ListEmbeddingModelsException,
|
|
1000028: VikingDBException,
|
|
1000029: QuotaLimiterException,
|
|
1000030: RerankException,
|
|
1000031: CollectionAliasNotExistException,
|
|
1000032: UserNoOrderException,
|
|
1000033: UserOverdueException,
|
|
1000034: HTTPErrException,
|
|
1000035: TaskNotFoundException,
|
|
}
|
|
|
|
# volcanoOK errCode = 0
|
|
# volcanoErrUnauthorized errCode = 1000001
|
|
# volcanoErrNoPermission errCode = 1000002
|
|
# volcanoErrInvalidRequest errCode = 1000003
|
|
# volcanoErrCollectionExist errCode = 1000004
|
|
# volcanoErrCollectionNotExist errCode = 1000005
|
|
# volcanoErrOperationNotAllowed errCode = 1000006
|
|
# volcanoErrIndexExist errCode = 1000007
|
|
# volcanoErrIndexNotExist errCode = 1000008
|
|
# volcanoErrQueryOpFailed errCode = 1000010
|
|
# volcanoErrDataNotFound errCode = 1000011
|
|
# volcanoErrDelOpFailed errCode = 1000013
|
|
# volcanoErrUpsertOpFailed errCode = 1000014
|
|
# volcanoErrTokenMismatch errCode = 1000015
|
|
# volcanoErrInvalidQueryVec errCode = 1000016
|
|
# volcanoErrInvalidPrimaryKey errCode = 1000017
|
|
# volcanoErrInvalidPartition errCode = 1000018
|
|
# volcanoErrInvalidScalarCond errCode = 1000019
|
|
# volcanoErrInvalidProxyService errCode = 1000020
|
|
# volcanoErrIndexRecall errCode = 1000021
|
|
# volcanoErrIndexFetchData errCode = 1000022
|
|
# volcanoErrIndexNotReady errCode = 1000023
|
|
# volcanoErrAPINotImplemented errCode = 1000024
|
|
# volcanoErrCalcEmbeddingFailed errCode = 1000025
|
|
# volcanoErrListEmbeddingModels errCode = 1000026
|
|
# volcanoErrInternal errCode = 1000028
|