n+j d Z ddlmZ ddlZddlmZmZ ddlmZ ej dej
Z G d de Z G d d
Z
G d d ZdS )
u LLM 客户端:协议 + litellm 实现(DeepSeek / MiniMax)+ 测试用脚本桩。
面向接口编程——上层只依赖 LLM 协议,换模型/换厂商只改注入(litellm 已抽象多厂商)。
) annotationsN)CallableProtocol)configz.*?\s*c e Zd Zd dZdS )
LLMmessages
list[dict]kwargsobjectreturnstrc dS )u7 传 [{'role','content'}...],返回模型文本输出。N selfr r s -/root/genesis-world/src/genesis/llm/client.pycompletez LLM.complete s Nr r
r r r
r )__name__
__module__ __qualname__r r r r r r s( r r c 0 e Zd ZdZ ddddddd
ZddZdS )
LiteLLMClientuO litellm 后端,按 config.LLM_PROVIDER 路由到 MiniMax(默认)或 DeepSeek。Ng?)api_keyapi_base temperaturemodel
str | Noner r r floatr
Nonec < dd l }|| _ t j dk r=|pdt j | _ |p t j | _ |p t j | _ n<|pdt j
| _ |p t j | _ |p t j | _ || _
d S )Nr minimaxzopenai/z deepseek/)litellm_litellmr LLM_PROVIDER
MINIMAX_MODELr MINIMAX_API_KEYr MINIMAX_BASE_URLr DEEPSEEK_MODELDEEPSEEK_API_KEYDEEPSEEK_BASE_URLr )r r r r r r% s r __init__zLiteLLMClient.__init__ s
) + + B"BF,@"B"BDJ"str 动态应答。N) responder responseslist[str] | NonerN "Callable[[list[dict]], str] | Noner
r" c N t |pg | _ || _ g | _ d S rI )list
_responses
_respondercalls)r rO rN s r r. zScriptedLLM.__init__L s) yB//#')
r r r
r r r c | j | | j | | S | j r| j d S t d )Nr u ScriptedLLM 响应已耗尽)rV appendrU rT r= AssertionErrorr s r r zScriptedLLM.completeV sa
(### ? &??8,, , ? *?&&q)) ) :;;;r rI )rO rP rN rQ r
r" r rJ r r r rM rM I s[ bb '+* 9= * * * * * *< < < < < r` s
# " " " " " % % % % % % % % rz2BD99 ( 05 05 05 05 05 05 05 05f< < < < < < < < <