39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
||
from typing import Literal
|
||
|
||
import httpx
|
||
|
||
RAW_RESPONSE_HEADER = "X-Stainless-Raw-Response"
|
||
OVERRIDE_CAST_TO_HEADER = "____stainless_override_cast_to"
|
||
|
||
# default timeout is 10 minutes
|
||
DEFAULT_TIMEOUT = httpx.Timeout(timeout=600.0, connect=5.0)
|
||
DEFAULT_MAX_RETRIES = 2
|
||
DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=1000, max_keepalive_connections=100)
|
||
|
||
INITIAL_RETRY_DELAY = 0.5
|
||
MAX_RETRY_DELAY = 8.0
|
||
|
||
EMBEDDING_MODEL: str = "bge-large-zh-v1.5"
|
||
HTTPX_TIMEOUT: float = 10.0
|
||
API_BASE_URI: str = 'http://127.0.0.1:7861/'
|
||
|
||
# 知识库相关
|
||
"""知识库中单段文本长度(不适用MarkdownHeaderTextSplitter)"""
|
||
CHUNK_SIZE: int = 250
|
||
"""知识库中相邻文本重合长度(不适用MarkdownHeaderTextSplitter)"""
|
||
OVERLAP_SIZE: int = 50
|
||
"""是否开启中文标题加强，以及标题增强的相关配置"""
|
||
ZH_TITLE_ENHANCE: bool = False
|
||
"""知识库匹配向量数量"""
|
||
VECTOR_SEARCH_TOP_K: int = 3 # TODO: 与 tool 配置项重复
|
||
"""知识库匹配相关度阈值，取值范围在0-2之间，SCORE越小，相关度越高，取到2相当于不筛选，建议设置在0.5左右"""
|
||
SCORE_THRESHOLD: float = 0.4
|
||
"""默认向量库/全文检索引擎类型"""
|
||
VS_TYPE: Literal["faiss", "milvus", "zilliz", "pg", "es", "relyt", "chromadb"] = "faiss"
|
||
# llm
|
||
TEMPERATURE: float = 0.7
|
||
LLM_MODEL = "chatglm-6b"
|
||
MAX_TOKENS = 2048
|
||
|