fix:rag serve error

This commit is contained in:
aries_ckt
2024-03-18 19:37:06 +08:00
parent 0ed30aa44a
commit 4413ff682f
10 changed files with 168 additions and 21 deletions

View File

@@ -1,4 +1,5 @@
from datetime import datetime
from enum import Enum
from typing import Dict, List, Optional, Union
from fastapi import File, UploadFile
@@ -60,6 +61,15 @@ class ChatCompletionRequestBody(BaseModel):
)
class ChatMode(Enum):
"""Chat mode"""
CHAT_NORMAL = "chat_normal"
CHAT_APP = "chat_app"
CHAT_AWEL_FLOW = "chat_flow"
CHAT_KNOWLEDGE = "chat_knowledge"
class SpaceModel(BaseModel):
"""name: knowledge space name"""