| | |
| | | from sqlalchemy.orm import Session |
| | | from starlette.responses import StreamingResponse, Response |
| | | from app.api import get_current_user, get_api_key |
| | | from app.config.const import smart_message_error, http_400, http_500, http_200, complex_dialog_chat |
| | | from app.config.const import smart_message_error, http_400, http_500, http_200, complex_dialog_chat, \ |
| | | complex_knowledge_chat_deep, complex_knowledge_chat |
| | | from app.models import UserModel |
| | | from app.models.base_model import get_db |
| | | from app.models.v2.chat import RetrievalRequest, ChatDataRequest, ComplexChatDao |
| | |
| | | |
| | | @chat_router_v2.post("/complex/chat/completions") |
| | | async def api_complex_chat_completions(chat: ChatDataRequest, current_user: UserModel = Depends(get_current_user), db: Session = Depends(get_db)): # current_user: UserModel = Depends(get_current_user) |
| | | # chat_mode = chat.chatMode |
| | | if chat.isDeep == 2 and chat.chatMode == complex_knowledge_chat: |
| | | chat.chatMode = complex_knowledge_chat_deep |
| | | complex_chat = await ComplexChatDao(db).get_complex_chat_by_mode(chat.chatMode) |
| | | if complex_chat: |
| | | if not chat.sessionId: |