From 13dbe95ed02caeee97e1f5591813e5c7cb64cc1b Mon Sep 17 00:00:00 2001 From: junjun Date: Mon, 28 Jul 2025 17:38:22 +0800 Subject: [PATCH] feat: mcp param desc --- backend/apps/chat/models/chat_model.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/backend/apps/chat/models/chat_model.py b/backend/apps/chat/models/chat_model.py index eb618ed3..2f771051 100644 --- a/backend/apps/chat/models/chat_model.py +++ b/backend/apps/chat/models/chat_model.py @@ -1,17 +1,18 @@ from datetime import datetime from typing import List, Optional +from fastapi import Body from pydantic import BaseModel from sqlalchemy import Column, Integer, Text, BigInteger, DateTime, Identity, Boolean from sqlmodel import SQLModel, Field +from apps.template.filter.generator import get_permissions_template from apps.template.generate_analysis.generator import get_analysis_template from apps.template.generate_chart.generator import get_chart_template from apps.template.generate_guess_question.generator import get_guess_question_template from apps.template.generate_predict.generator import get_predict_template from apps.template.generate_sql.generator import get_sql_template from apps.template.select_datasource.generator import get_datasource_template -from apps.template.filter.generator import get_permissions_template class Chat(SQLModel, table=True): @@ -153,14 +154,14 @@ class AiModelQuestion(BaseModel): class ChatQuestion(AiModelQuestion): - question: str - chat_id: int + question: str = Body(description='用户提问') + chat_id: int = Body(description='会话ID') class ChatMcp(ChatQuestion): - token: str + token: str = Body(description='token') class ChatStart(BaseModel): - username: str - password: str + username: str = Body(description='用户名') + password: str = Body(description='密码')