85 lines
9.3 KiB
Plaintext
85 lines
9.3 KiB
Plaintext
|
o
|
|||
|
|
<00><>5iJ' <00> @ s<> d Z ddlZddlZddlZddlmZmZmZ ddlm Z m
|
|||
|
|
Z
|
|||
|
|
ddl m Z e<03>
e<0E>Ze G dd<07> d<07><02>Ze G dd <09> d <09><02>Ze G d
|
|||
|
|
d <0B> d <0B><02>Ze G d d
<0A> d
<0A><02>Ze G dd<0F> d<0F><02>Ze G dd<11> d<11><02>ZG dd<13> d<13>Zdadefdd<16>Zdd<18> ZdS )uK
|
|||
|
|
统一配置管理模块
|
|||
|
|
整合所有配置，提供统一的配置接口
|
|||
|
|
<EFBFBD> N)<03>Dict<63>Any<6E>Optional)<02> dataclass<73>asdict)<01>Pathc @ sN e Zd ZU dZdZeed< dZeed< dZ eed< dZ
|
|||
|
|
eed < d
|
|||
|
|
Z eed < d S )
<0A>DatabaseConfigu 数据库配置zRmysql+pymysql://tsp_assistant:password@jeason.online/tsp_assistant?charset=utf8mb4<62>url<72>
|
|||
|
|
<00> pool_size<7A> <00> max_overflow<6F> <00> pool_timeout<75> <00> pool_recycleN) <0C>__name__<5F>
|
|||
|
|
__module__<EFBFBD> __qualname__<5F>__doc__r <00>str<74>__annotations__r <00>intr
r r <00> r r <00>)/root/assist/src/config/unified_config.pyr <00>
|
|||
|
|
r c @ sf e Zd ZU dZdZeed< dZeed< dZeed< dZ eed < d
|
|||
|
|
Z
|
|||
|
|
e ed < d Z e
ed
< dZe
ed< dS )<11> LLMConfigu LLM配置<E9858D>qwen<65>providerz#sk-c0dbefa1718d46eaa897199135066f00<30>api_keyz1https://dashscope.aliyuncs.com/compatible-mode/v1<76>base_urlzqwen-plus-latest<73>modelgffffff<66>?<3F> temperaturei<65> <00>
|
|||
|
|
max_tokensr <00>timeoutN)r r r r r r r r r r! r" <00>floatr# r r$ r r r r r <00>
|
|||
|
|
r c @ sN e Zd ZU dZdZeed< dZeed< dZ eed< dZ
|
|||
|
|
e ed < d
|
|||
|
|
Z eed < d S )
<0A> ServerConfigu 服务器配置z0.0.0.0<EFBFBD>hosti<74> <00>porti=" <00>websocket_portF<74>debug<75>INFO<46> log_levelN)
r r r r r( r r r) r r* r+ <00>boolr- r r r r r' & r r' c @ sf e Zd ZU dZdZeed< dZeed< dZeed< dZ eed< dZ
|
|||
|
|
eed< d Z e ed
|
|||
|
|
< d Z
e ed < d
S )<0E> FeishuConfigu 飞书配置<E9858D> <00>app_id<69>
|
|||
|
|
app_secret<EFBFBD> app_token<65>table_id<69>active<76>statusr
|
|||
|
|
<00>
|
|||
|
|
sync_limitr <00>auto_sync_intervalN)r r r r r1 r r r2 r3 r4 r6 r7 r r8 r r r r r/ / r& r/ c @ sr e Zd ZU dZdZeed< dZeed< dZeed< dZ eed< dZ
|
|||
|
|
eed < d
|
|||
|
|
Z e ed < d
|
|||
|
|
Z
e ed < d
|
|||
|
|
Ze ed
< dS )<0F>AIAccuracyConfigu AI准确率配置gffffff<66>?<3F>auto_approve_thresholdg<64><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>?<3F> use_human_resolution_thresholdg<64><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>?<3F>manual_review_threshold<6C>ai_suggestion_confidence<63>human_resolution_confidenceT<65> prefer_human_when_low_accuracy<63>enable_auto_approval<61>enable_human_fallbackN)r r r r r: r% r r; r< r= r> r? r. r@ rA r r r r r9 : s
|
|||
|
|
r9 c @ sZ e Zd ZU dZdZeed< dZeed< dZ eed< dZ
|
|||
|
|
eed< d Z eed
|
|||
|
|
< dZ eed < d S )
<0A> SystemConfigu 系统配置T<E7BDAE>backup_enabled<65> <00>backup_interval<61> <00>max_backup_files<65>
cache_enabledr <00> cache_ttl<74>monitoring_enabledN)
r r r r rC r. r rE r rG rH rI rJ r r r r rB F s
|
|||
|
|
rB c @ s e Zd ZdZd$defdd<05>Zdefdd<08>Zd d
|
|||
|
|
<EFBFBD> Zd d <0C> Z d
d<0E> Z
|
|||
|
|
defdd<10>Z de ee
f fdd<12>Zde ee
f fdd<14>Zde ee
f fdd<16>Zde ee
f fdd<18>Zde ee
f fdd<1A>Zded e ee
f fd d <1E>Zdefd d <20>Zde ee
f fd!d"<22>Zd#S )%<25>
UnifiedConfigu 统一配置管理器<E79086>config<69>
|
|||
|
|
config_dirc C sT t |<01>| _| jd | _t<03> | _| <00><05> | _t<07> | _t <09> | _
|
|||
|
|
t <0B> | _ t
<0A> | _| <00><0F> d S )Nzunified_config.json)r rM <00> config_filer <00>database<73>_load_default_llm_config<69>llmr' <00>serverr/ <00>feishur9 <00> ai_accuracyrB <00>system<65> load_config)<02>selfrM r r r <00>__init__S s
|
|||
|
|
|
|||
|
|
zUnifiedConfig.__init__<5F>returnc
|
|||
|
|
C sj zddl m} t|j|j|j|j|j|jd<03>W S t y4 } zt
|
|||
|
|
<EFBFBD> d|<02> <00><02> t<02> W Y d}~S d}~ww )u 加载默认LLM配置r )<01>DEFAULT_CONFIG)r r r r! r" r# u5 无法加载默认LLM配置，使用内置默认值: N) <0C>config.llm_configrZ r r r r r! r" r# <00> Exception<6F>logger<65>warning)rW rZ <00>er r r rP b s <08><08><02>z&UnifiedConfig._load_default_llm_configc
|
|||
|
|
C sL z<>| j <00><01> rt| j ddd<03><03>
}t<03>|<01>}W d <00> n1 s w Y d|v r1tdi |d <00><01>| _d|v r?tdi |d <00><01>| _d|v rMt di |d <00><01>| _
|
|||
|
|
d|v r[t di |d <00><01>| _ d |v rit
di |d <00><01>| _d
|
|||
|
|
|v rwtdi |d
|
|||
|
|
<00><01>| _t<11>d <0B> W dS t<11>d <0C> | <00><13> W dS ty<> } zt<11>d
|<03> <00><02> W Y d}~dS d}~ww )u 加载配置文件<E69687>r<>utf-8<><01>encodingNrO rQ rR rS rT rU u 配置文件加载成功u* 配置文件不存在，使用默认配置u 加载配置文件失败: r )rN <00>exists<74>open<65>json<6F>loadr rO r rQ r' rR r/ rS r9 rT rB rU r] <00>info<66> save_configr\ <00>error)rW <00>f<> config_datar_ r r r rV s s0
|
|||
|
|
<1C>
|
|||
|
|
<08><02>zUnifiedConfig.load_configc
|
|||
|
|
C s<> zK| j jdd<02> t| j<03>t| j<04>t| j<05>t| j<06>t| j<07>t| j<08>d<03>}t | j
|
|||
|
|
ddd<06><03>}t j ||ddd <09> W d
|
|||
|
|
<00> n1 s?w Y t
<0A>d <0B> W d
|
|||
|
|
S tyf } zt
<0A>d |<03> <00><02> W Y d
|
|||
|
|
}~d
|
|||
|
|
S d
|
|||
|
|
}~ww )
u 保存配置文件T)<01>exist_ok<6F>rO rQ rR rS rT rU <00>wra rb <00> F)<02>indent<6E> ensure_asciiNu 配置文件保存成功u 保存配置文件失败: )rM <00>mkdirr rO rQ rR rS rT rU re rN rf <00>dumpr] rh r\ rj )rW rl rk r_ r r r ri <00> s" <06> <1C> <08><02>zUnifiedConfig.save_configc C s<> t <00>d<01>r t <00>d<01>| j_t <00>d<02>rt <00>d<02>| j_t <00>d<03>r$t <00>d<03>| j_t <00>d<04>r0t <00>d<04>| j_t <00>d<05>r>tt <00>d<05><01>| j _
|
|||
|
|
t <00>d<06>rJt <00>d<06>| j _ t <00>d<07>rVt <00>d<07>| j _
t <00>d<08>rbt <00>d<08>| j _t <00>d <09>rnt <00>d <09>| j _t <00>d
|
|||
|
|
<EFBFBD>r|t <00>d
|
|||
|
|
<EFBFBD>| j _d S d S ) u 从环境变量加载配置<E9858D> DATABASE_URL<52> LLM_PROVIDER<45> LLM_API_KEY<45> LLM_MODEL<45> SERVER_PORT<52> LOG_LEVEL<45>
FEISHU_APP_ID<49>FEISHU_APP_SECRET<45>FEISHU_APP_TOKEN<45>FEISHU_TABLE_IDN)<11>os<6F>getenvrO r rQ r r r! r rR r) r- rS r1 r2 r3 r4 <00>rW r r r <00>
load_from_env<6E> s*
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
<04>zUnifiedConfig.load_from_envc C s | j jS )u 获取数据库连接URL)rO r r<> r r r <00>get_database_url<72> s z UnifiedConfig.get_database_urlc C <00>
|
|||
|
|
t | j<01>S )u 获取LLM配置)r rQ r<> r r r <00>get_llm_config<69> <00>
|
|||
|
|
z UnifiedConfig.get_llm_configc C r<> )u 获取服务器配置)r rR r<> r r r <00>get_server_config<69> r<> z UnifiedConfig.get_server_configc C r<> )u 获取飞书配置)r rS r<> r r r <00>get_feishu_config<69> r<> z UnifiedConfig.get_feishu_configc C r<> )u 获取AI准确率配置)r rT r<> r r r <00>get_ai_accuracy_config<69> r<> z$UnifiedConfig.get_ai_accuracy_configc C r<> )u 获取系统配置)r rU r<> r r r <00>get_system_config<69> r<> z UnifiedConfig.get_system_config<69>sectionrl c
|
|||
|
|
C s<> ze|dkrt d i |<02><01>| _nH|dkrtd i |<02><01>| _n;|dkr(td i |<02><01>| _n.|dkr5td i |<02><01>| _n!|dkrBtd i |<02><01>| _ n|dkrOt
|
|||
|
|
d i |<02><01>| _ nt d|<01> <00><02><01>| <00>
<0A> t<0E>d|<01> d <09><03> W d S tyz } z t<0E>d
|
|||
|
|
|<03> <00><02> <00> d }~ww )
u 更新配置rO rQ rR rS rT rU u 未知的配置节: u
|
|||
|
|
配置节 u
更新成功u 更新配置失败: Nr )r rO r rQ r' rR r/ rS r9 rT rB rU <00>
|
|||
|
|
ValueErrorri r] rh r\ rj )rW r<> rl r_ r r r <00>
update_config<69> s* <08><02>zUnifiedConfig.update_configc
|
|||
|
|
C s<> z8| j js
t<02>d<01> W dS | jjst<02>d<03> | jjdkr1t | jj
|
|||
|
|
| jj | jj | jj
g<04>s1t<02>d<05> t<02>d<06> W dS tyS } zt<02>d|<01> <00><02> W Y d }~dS d }~ww )
|
|||
|
|
u 验证配置有效性u 数据库URL未配置Fu LLM API密钥未配置r5 u 飞书配置不完整u 配置验证通过Tu 配置验证失败: N)rO r r] rj rQ r r^ rS r6 <00>allr1 r2 r3 r4 rh r\ )rW r_ r r r <00>validate_config<69> s$
|
|||
|
|
|
|||
|
|
<06>
|
|||
|
|
|
|||
|
|
<08><02>z UnifiedConfig.validate_configc C s6 t | j<01>t | j<02>t | j<03>t | j<04>t | j<05>t | j<06>d<01>S )u 获取所有配置rn )r rO rQ rR rS rT rU r<> r r r <00>get_all_config s <06>z UnifiedConfig.get_all_configN)rL )r r r r r rX r rP rV ri r<> r<> r r r<> r<> r<> r<> r<> r<> r. r<> r<> r r r r rK P s rK rY c C s t du r t<01> a t <00><02> t S )u 获取全局配置实例N)<03>_config_instancerK r<> r r r r <00>
|
|||
|
|
get_config s r<> c C s
|
|||
|
|
da t<01> S )u 重新加载配置N)r<> r<> r r r r <00>
reload_config$ s r<> )r r rf <00>logging<6E>typingr r r <00> dataclassesr r <00>pathlibr <00> getLoggerr r] r r r' r/ r9 rB rK r<> r<> r<> r r r r <00><module> s2
|
|||
|
|
|
|||
|
|
|
|||
|
|
K
|