58 lines
2.5 KiB
Python
58 lines
2.5 KiB
Python
#
|
||
# Copyright 2024 The InfiniFlow Authors. All Rights Reserved.
|
||
#
|
||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
# you may not use this file except in compliance with the License.
|
||
# You may obtain a copy of the License at
|
||
#
|
||
# http://www.apache.org/licenses/LICENSE-2.0
|
||
#
|
||
# Unless required by applicable law or agreed to in writing, software
|
||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
# See the License for the specific language governing permissions and
|
||
# limitations under the License.
|
||
#
|
||
|
||
from typing import Optional, List, Dict, Any
|
||
from pydantic import BaseModel, Field
|
||
|
||
|
||
class SetDialogRequest(BaseModel):
|
||
"""设置/创建对话框请求"""
|
||
dialog_id: Optional[str] = Field(default="", description="对话框ID，为空时创建新对话框")
|
||
name: Optional[str] = Field(default="New Dialog", description="对话框名称")
|
||
description: Optional[str] = Field(default="A helpful dialog", description="对话框描述")
|
||
icon: Optional[str] = Field(default="", description="图标")
|
||
top_n: Optional[int] = Field(default=6, description="Top N")
|
||
top_k: Optional[int] = Field(default=1024, description="Top K")
|
||
rerank_id: Optional[str] = Field(default="", description="重排序模型ID")
|
||
similarity_threshold: Optional[float] = Field(default=0.1, description="相似度阈值")
|
||
vector_similarity_weight: Optional[float] = Field(default=0.3, description="向量相似度权重")
|
||
llm_setting: Optional[Dict[str, Any]] = Field(default={}, description="LLM设置")
|
||
meta_data_filter: Optional[Dict[str, Any]] = Field(default={}, description="元数据过滤器")
|
||
prompt_config: Dict[str, Any] = Field(..., description="提示配置")
|
||
kb_ids: Optional[List[str]] = Field(default=[], description="知识库ID列表")
|
||
llm_id: Optional[str] = Field(default=None, description="LLM ID")
|
||
|
||
|
||
class ListDialogsNextQuery(BaseModel):
|
||
"""列出对话框查询参数"""
|
||
keywords: Optional[str] = ""
|
||
page: Optional[int] = 0
|
||
page_size: Optional[int] = 0
|
||
parser_id: Optional[str] = None
|
||
orderby: Optional[str] = "create_time"
|
||
desc: Optional[str] = "true"
|
||
|
||
|
||
class ListDialogsNextBody(BaseModel):
|
||
"""列出对话框请求体"""
|
||
owner_ids: Optional[List[str]] = []
|
||
|
||
|
||
class DeleteDialogRequest(BaseModel):
|
||
"""删除对话框请求"""
|
||
dialog_ids: List[str] = Field(..., description="要删除的对话框ID列表")
|
||
|