niuniu/ShiErFeng
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

初始提交

Browse Source
This commit is contained in:
焦龙言
2026-07-24 09:43:10 +08:00
commit b5b4721456
793 changed files with 116305 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
shierfeng-fastapi-backend/module_ai/entity/do/ai_chat_do.py
+26
View File
@@ -0,0 +1,26 @@
from datetime import datetime
from sqlalchemy import CHAR, BigInteger, Column, DateTime, Float, Integer, Text
from config.database import Base
class AiChatConfig(Base):
"""
AI对话配置表
"""
__tablename__ = 'ai_chat_config'
__table_args__ = {'comment': 'AI对话配置表'}
chat_config_id = Column(BigInteger, primary_key=True, nullable=False, autoincrement=True, comment='配置主键')
user_id = Column(BigInteger, nullable=False, unique=True, comment='用户ID')
temperature = Column(Float, nullable=True, comment='默认温度')
add_history_to_context = Column(CHAR(1), server_default='0', comment='是否添加历史记录(0是, 1否)')
num_history_runs = Column(Integer, nullable=True, comment='历史记录条数')
system_prompt = Column(Text, nullable=True, comment='系统提示词')
metrics_default_visible = Column(CHAR(1), server_default='0', comment='默认显示指标(0是, 1否)')
vision_enabled = Column(CHAR(1), server_default='1', comment='是否开启视觉(0是, 1否)')
image_max_size_mb = Column(Integer, nullable=True, comment='图片最大大小(MB)')
create_time = Column(DateTime, nullable=True, default=datetime.now, comment='创建时间')
update_time = Column(DateTime, nullable=True, default=datetime.now, comment='更新时间')
shierfeng-fastapi-backend/module_ai/entity/do/ai_model_do.py
+62
View File
@@ -0,0 +1,62 @@
from datetime import datetime
from sqlalchemy import CHAR, BigInteger, Column, DateTime, Float, Integer, String
from config.database import Base
from config.env import DataBaseConfig
from utils.common_util import SqlalchemyUtil
class AiModels(Base):
"""
AI模型表
"""
__tablename__ = 'ai_models'
__table_args__ = {'comment': 'AI模型表'}
model_id = Column(BigInteger, primary_key=True, nullable=False, autoincrement=True, comment='模型主键')
model_code = Column(String(100), nullable=False, comment='模型编码')
model_name = Column(
String(100),
nullable=True,
server_default=SqlalchemyUtil.get_server_default_null(DataBaseConfig.db_type),
comment='模型名称',
)
provider = Column(String(50), nullable=False, comment='提供商')
model_sort = Column(Integer, nullable=False, comment='显示顺序')
api_key = Column(
String(255),
nullable=True,
server_default=SqlalchemyUtil.get_server_default_null(DataBaseConfig.db_type),
comment='API Key',
)
base_url = Column(
String(255),
nullable=True,
server_default=SqlalchemyUtil.get_server_default_null(DataBaseConfig.db_type),
comment='Base URL',
)
model_type = Column(
String(50),
nullable=True,
server_default=SqlalchemyUtil.get_server_default_null(DataBaseConfig.db_type),
comment='模型类型',
)
max_tokens = Column(Integer, nullable=True, comment='最大输出token')
temperature = Column(Float, nullable=True, comment='默认温度')
support_reasoning = Column(CHAR(1), server_default='N', comment='是否支持推理')
support_images = Column(CHAR(1), server_default='N', comment='是否支持图片')
status = Column(CHAR(1), server_default='0', comment='模型状态')
user_id = Column(BigInteger, nullable=True, comment='用户ID')
dept_id = Column(BigInteger, nullable=True, comment='部门ID')
create_by = Column(String(64), nullable=True, server_default="''", comment='创建者')
create_time = Column(DateTime, nullable=True, default=datetime.now(), comment='创建时间')
update_by = Column(String(64), nullable=True, server_default="''", comment='更新者')
update_time = Column(DateTime, nullable=True, default=datetime.now(), comment='更新时间')
remark = Column(
String(500),
nullable=True,
server_default=SqlalchemyUtil.get_server_default_null(DataBaseConfig.db_type),
comment='备注',
)
shierfeng-fastapi-backend/module_ai/entity/vo/ai_chat_vo.py
+159
View File
@@ -0,0 +1,159 @@
from datetime import datetime
from typing import Literal
from pydantic import BaseModel, ConfigDict, Field
from pydantic.alias_generators import to_camel
class AiChatRequestModel(BaseModel):
"""
AI对话请求模型
"""
model_config = ConfigDict(alias_generator=to_camel)
session_id: str | None = Field(default=None, description='会话ID')
model_id: int = Field(description='模型ID')
message: str = Field(description='用户消息')
is_reasoning: bool | None = Field(default=None, description='本次是否开启深度思考')
images: list[str] | None = Field(default=None, description='图片URL列表')
class AiChatConfigModel(BaseModel):
"""
AI对话配置模型
"""
model_config = ConfigDict(alias_generator=to_camel)
chat_config_id: int | None = Field(default=None, description='配置主键')
user_id: int | None = Field(default=None, description='用户ID')
temperature: float | None = Field(default=None, description='默认温度')
add_history_to_context: Literal['0', '1'] | None = Field(default=None, description='是否添加历史记录')
num_history_runs: int | None = Field(default=3, description='历史记录条数')
system_prompt: str | None = Field(default=None, description='系统提示词')
metrics_default_visible: Literal['0', '1'] | None = Field(default=None, description='默认显示指标')
vision_enabled: Literal['0', '1'] | None = Field(default=None, description='是否开启视觉')
image_max_size_mb: int | None = Field(default=None, description='图片最大大小(MB)')
create_time: datetime | None = Field(default=None, description='创建时间')
update_time: datetime | None = Field(default=None, description='更新时间')
class AiChatSessionBaseModel(BaseModel):
"""
AI对话会话基础模型
"""
model_config = ConfigDict(alias_generator=to_camel)
session_id: str = Field(description='会话ID')
session_title: str | None = Field(default=None, description='会话标题')
session_type: str | None = Field(default=None, description='会话类型')
user_id: str | None = Field(default=None, description='用户ID')
created_at: datetime | None = Field(default=None, description='创建时间')
updated_at: datetime | None = Field(default=None, description='更新时间')
class ModelInfoModel(BaseModel):
"""
对话会话数据模型-模型信息模型
"""
model_config = ConfigDict(alias_generator=to_camel)
id: str | None = Field(default=None, description='模型ID')
name: str | None = Field(default=None, description='模型名称')
provider: str | None = Field(default=None, description='模型提供者')
temperature: float | None = Field(default=None, description='模型温度')
class AgentDataModel(BaseModel):
"""
对话会话数据模型-智能体数据模型
"""
model_config = ConfigDict(alias_generator=to_camel)
model: ModelInfoModel | None = Field(default=None, description='模型信息')
agent_id: str | None = Field(default=None, description='智能体ID')
class SessionMetricsModel(BaseModel):
"""
对话会话数据模型-会话指标模型
"""
model_config = ConfigDict(alias_generator=to_camel)
cost: float | None = Field(default=None, description='成本（元）')
timer: float | None = Field(default=None, description='运行时长（秒）')
duration: float | None = Field(default=None, description='运行时长（秒）')
input_tokens: int | None = Field(default=None, description='输入Token数量')
total_tokens: int | None = Field(default=None, description='总Token数量')
output_tokens: int | None = Field(default=None, description='输出Token数量')
provider_metrics: dict | None = Field(default=None, description='提供者指标')
reasoning_tokens: int | None = Field(default=None, description='推理Token数量')
cache_read_tokens: int | None = Field(default=None, description='缓存读取Token数量')
additional_metrics: dict | None = Field(default=None, description='其他指标')
audio_input_tokens: int | None = Field(default=None, description='音频输入Token数量')
audio_total_tokens: int | None = Field(default=None, description='音频总Token数量')
cache_write_tokens: int | None = Field(default=None, description='缓存写入Token数量')
audio_output_tokens: int | None = Field(default=None, description='音频输出Token数量')
time_to_first_token: float | None = Field(default=None, description='到第一个Token的时间（秒）')
class SessionDataModel(BaseModel):
"""
对话会话数据模型
"""
model_config = ConfigDict(alias_generator=to_camel)
session_state: dict | None = Field(default=None, description='会话状态')
session_metrics: SessionMetricsModel | None = Field(default=None, description='会话指标')
class MessageMetrics(BaseModel):
"""
对话消息模型-消息指标模型
"""
model_config = ConfigDict(alias_generator=to_camel)
input_tokens: int | None = Field(default=None, description='输入Token数量')
total_tokens: int | None = Field(default=None, description='总Token数量')
output_tokens: int | None = Field(default=None, description='输出Token数量')
reasoning_tokens: int | None = Field(default=None, description='推理Token数量')
duration: float | None = Field(default=None, description='运行时长（秒）')
time_to_first_token: float | None = Field(default=None, description='到第一个Token的时间（秒）')
class ChatMessageModel(BaseModel):
"""
对话消息模型
"""
model_config = ConfigDict(alias_generator=to_camel)
id: str | None = Field(default=None, description='消息ID')
role: str | None = Field(default=None, description='角色')
content: str | None = Field(default=None, description='内容')
images: list[str] | None = Field(default=None, description='图片列表')
metrics: MessageMetrics | None = Field(default=None, description='Token使用统计')
created_at: datetime | None = Field(default=None, description='创建时间')
from_history: bool | None = Field(default=None, description='是否来自历史记录')
reasoning_content: str | None = Field(default=None, description='推理/思考内容')
stop_after_tool_call: bool | None = Field(default=None, description='是否在工具调用后停止')
class AiChatSessionModel(AiChatSessionBaseModel):
"""
AI对话会话模型
"""
model_config = ConfigDict(alias_generator=to_camel)
agent_id: str | None = Field(default=None, description='智能体ID')
session_data: SessionDataModel | None = Field(default=None, description='会话数据')
agent_data: AgentDataModel | None = Field(default=None, description='智能体数据')
messages: list[ChatMessageModel] | None = Field(default=None, description='消息列表')
shierfeng-fastapi-backend/module_ai/entity/vo/ai_model_vo.py
+64
View File
@@ -0,0 +1,64 @@
from datetime import datetime
from typing import Literal
from pydantic import BaseModel, ConfigDict, Field
from pydantic.alias_generators import to_camel
from pydantic_validation_decorator import NotBlank, Size
class AiModelModel(BaseModel):
"""
AI模型表对应pydantic模型
"""
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
model_id: int | None = Field(default=None, description='模型主键')
model_code: str | None = Field(default=None, description='模型编码')
model_name: str | None = Field(default=None, description='模型名称')
provider: str | None = Field(default=None, description='提供商')
model_sort: int | None = Field(default=None, description='显示顺序')
api_key: str | None = Field(default=None, description='API Key')
base_url: str | None = Field(default=None, description='Base URL')
max_tokens: int | None = Field(default=None, description='最大输出token')
temperature: float | None = Field(default=None, description='默认温度')
support_reasoning: Literal['Y', 'N'] | None = Field(default=None, description='是否支持推理(深度思考)')
support_images: Literal['Y', 'N'] | None = Field(default=None, description='是否支持图片')
model_type: str | None = Field(default=None, description='模型类型')
status: Literal['0', '1'] | None = Field(default=None, description='模型状态')
user_id: int | None = Field(default=None, description='用户ID')
dept_id: int | None = Field(default=None, description='部门ID')
create_by: str | None = Field(default=None, description='创建者')
create_time: datetime | None = Field(default=None, description='创建时间')
update_by: str | None = Field(default=None, description='更新者')
update_time: datetime | None = Field(default=None, description='更新时间')
remark: str | None = Field(default=None, description='备注')
@NotBlank(field_name='model_code', message='模型编码不能为空')
@Size(field_name='model_code', min_length=0, max_length=100, message='模型编码长度不能超过100个字符')
def get_model_code(self) -> str | None:
return self.model_code
@NotBlank(field_name='base_url', message='Base URL不能为空')
@Size(field_name='base_url', min_length=0, max_length=255, message='Base URL长度不能超过255个字符')
def get_base_url(self) -> str | None:
return self.base_url
class AiModelPageQueryModel(AiModelModel):
"""
AI模型管理分页查询模型
"""
page_num: int = Field(default=1, description='当前页码')
page_size: int = Field(default=10, description='每页记录数')
class DeleteAiModelModel(BaseModel):
"""
删除AI模型模型
"""
model_config = ConfigDict(alias_generator=to_camel)
model_ids: str = Field(description='需要删除的模型主键')