Skip to content

Commit 13dbe95

Browse files
committed
feat: mcp param desc
1 parent fe8d2cb commit 13dbe95

1 file changed

Lines changed: 7 additions & 6 deletions

File tree

backend/apps/chat/models/chat_model.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,18 @@
11
from datetime import datetime
22
from typing import List, Optional
33

4+
from fastapi import Body
45
from pydantic import BaseModel
56
from sqlalchemy import Column, Integer, Text, BigInteger, DateTime, Identity, Boolean
67
from sqlmodel import SQLModel, Field
78

9+
from apps.template.filter.generator import get_permissions_template
810
from apps.template.generate_analysis.generator import get_analysis_template
911
from apps.template.generate_chart.generator import get_chart_template
1012
from apps.template.generate_guess_question.generator import get_guess_question_template
1113
from apps.template.generate_predict.generator import get_predict_template
1214
from apps.template.generate_sql.generator import get_sql_template
1315
from apps.template.select_datasource.generator import get_datasource_template
14-
from apps.template.filter.generator import get_permissions_template
1516

1617

1718
class Chat(SQLModel, table=True):
@@ -153,14 +154,14 @@ def filter_user_question(self):
153154

154155

155156
class ChatQuestion(AiModelQuestion):
156-
question: str
157-
chat_id: int
157+
question: str = Body(description='用户提问')
158+
chat_id: int = Body(description='会话ID')
158159

159160

160161
class ChatMcp(ChatQuestion):
161-
token: str
162+
token: str = Body(description='token')
162163

163164

164165
class ChatStart(BaseModel):
165-
username: str
166-
password: str
166+
username: str = Body(description='用户名')
167+
password: str = Body(description='密码')

0 commit comments

Comments
 (0)