feat(multimodel): support multimodal memory display and improve code style
This commit is contained in:
@@ -7,6 +7,7 @@ from sqlalchemy.dialects.postgresql import UUID
|
||||
from sqlalchemy.dialects.postgresql import JSONB
|
||||
|
||||
from app.db import Base
|
||||
from app.schemas import FileType
|
||||
|
||||
|
||||
class PerceptualType(IntEnum):
|
||||
@@ -15,6 +16,16 @@ class PerceptualType(IntEnum):
|
||||
TEXT = 3
|
||||
CONVERSATION = 4
|
||||
|
||||
@staticmethod
|
||||
def trans_from_file_type(file_type: FileType | str):
|
||||
type_map = {
|
||||
FileType.IMAGE: PerceptualType.VISION,
|
||||
FileType.AUDIO: PerceptualType.AUDIO,
|
||||
FileType.VIDEO: PerceptualType.VISION,
|
||||
FileType.DOCUMENT: PerceptualType.TEXT
|
||||
}
|
||||
return type_map.get(file_type, PerceptualType.TEXT)
|
||||
|
||||
|
||||
class FileStorageService(IntEnum):
|
||||
LOCAL = 1
|
||||
|
||||
Reference in New Issue
Block a user