|
1 | 1 | from datetime import datetime |
2 | 2 | from typing import List, Optional |
3 | 3 |
|
| 4 | +from fastapi import Body |
4 | 5 | from pydantic import BaseModel |
5 | 6 | from sqlalchemy import Column, Integer, Text, BigInteger, DateTime, Identity, Boolean |
6 | 7 | from sqlmodel import SQLModel, Field |
7 | 8 |
|
| 9 | +from apps.template.filter.generator import get_permissions_template |
8 | 10 | from apps.template.generate_analysis.generator import get_analysis_template |
9 | 11 | from apps.template.generate_chart.generator import get_chart_template |
10 | 12 | from apps.template.generate_guess_question.generator import get_guess_question_template |
11 | 13 | from apps.template.generate_predict.generator import get_predict_template |
12 | 14 | from apps.template.generate_sql.generator import get_sql_template |
13 | 15 | from apps.template.select_datasource.generator import get_datasource_template |
14 | | -from apps.template.filter.generator import get_permissions_template |
15 | 16 |
|
16 | 17 |
|
17 | 18 | class Chat(SQLModel, table=True): |
@@ -153,14 +154,14 @@ def filter_user_question(self): |
153 | 154 |
|
154 | 155 |
|
155 | 156 | class ChatQuestion(AiModelQuestion): |
156 | | - question: str |
157 | | - chat_id: int |
| 157 | + question: str = Body(description='用户提问') |
| 158 | + chat_id: int = Body(description='会话ID') |
158 | 159 |
|
159 | 160 |
|
160 | 161 | class ChatMcp(ChatQuestion): |
161 | | - token: str |
| 162 | + token: str = Body(description='token') |
162 | 163 |
|
163 | 164 |
|
164 | 165 | class ChatStart(BaseModel): |
165 | | - username: str |
166 | | - password: str |
| 166 | + username: str = Body(description='用户名') |
| 167 | + password: str = Body(description='密码') |
0 commit comments