Ti-SrSSKJrJr SSKJrJrJrJrJ r SSK r \ R"\ 5r "SS\5r"SS5rg) uD AI 提供商基类 定义所有 AI 提供商必须实现的接口 )ABCabstractmethod)DictListAnyOptional GeneratorNc8\rSrSrSrS S\S\S\4Sjjr\\S\4S j55r \\S\4S j55r \S\4S j5r \S\4S j5r \S\ \4S j5r\S!S\ \\\4S\ \S\S\S\\\44 Sjj5r\S!S\ \\\4S\ \S\S\S\\\\4SS44 Sjj5rS\\\44SjrS\ \4SjrS\ \4SjrSrS\S\4SjrS\ \\\4S\ \\\44SjrS\ \S\ \4SjrS\S\\\44SjrSrg)"BaseAIProvideruXAI 提供商抽象基类 所有 AI 提供商都必须继承此类并实现抽象方法 Napi_keybase_urlmodelc XlU=(d UR5UlU=(d UR5UlUR SS5UlUR SS5Ul/UlUR S5Ul g)u初始化 AI 提供商 Args: api_key: API 密钥 base_url: API 基础 URL（可选，使用默认值） model: 模型名称（可选，使用默认值） **kwargs: 其他配置参数 timeouti, max_retries log_callbackN) r get_default_base_urlrget_default_modelrgetrr_toolsr)selfr rrkwargss O/Users/yu22x/Desktop/ddd_副本123/ctf/app/services/ai_driver/providers/base.py__init__BaseAIProvider.__init__st  ?D$=$=$? 6d446 zz)S1 !::mQ7 #JJ~6returncg)u返回提供商类型标识Nrs r provider_typeBaseAIProvider.provider_type* rcg)u返回提供商显示名称Nr!r"s r provider_nameBaseAIProvider.provider_name0r%rcg)u返回默认 API Base URLNr!r"s rr#BaseAIProvider.get_default_base_url6 rcg)u返回默认模型名称Nr!r"s rr BaseAIProvider.get_default_model;r+rcg)u返回可用模型列表Nr!r"s rget_available_models#BaseAIProvider.get_available_models@r+rmessagestools temperature max_tokensc g)u执行聊天补全 Args: messages: 消息列表，格式为 [{"role": "user/assistant/system", "content": "..."}] tools: 工具定义列表（用于 Function Calling） temperature: 温度参数 max_tokens: 最大 token 数 **kwargs: 其他参数 Returns: 响应字典，包含: - content: 文本响应内容 - tool_calls: 工具调用列表（如果有） - usage: token 使用统计 - finish_reason: 结束原因 Nr!rr1r2r3r4rs rchatBaseAIProvider.chatEs2 rc g)uR流式聊天补全 Args: 同 chat 方法 Yields: 响应片段字典 Nr!r6s r chat_streamBaseAIProvider.chat_stream`s" rcSSKnUR5nURSSS./SS9n[UR5U- S-5nURS 5(aS S UR3UURS .$S SUS.$![ a=n[ RS[U535 S S[U53SS.sSnA$SnAff=f)u测试 API 连接 Returns: 测试结果字典: - success: 是否成功 - message: 结果消息 - latency: 响应延迟（毫秒） rNuserz"Hi, just testing. Reply with 'OK'.)rolecontent )r1r4ir?Tu连接成功，模型: )successmessagelatencyrFu连接成功但响应为空)rArBrCu测试连接失败: u连接失败: ) timer7intrr Exceptionloggererrorstr)rrD start_timeresponserCes rtest_connectionBaseAIProvider.test_connectionss  Jyy#)6Z[\!H 499;3t;4- .  $*c3h*XtDz4:% 6# 6 6 d38n)= $tCQTH~BV  4: $t*   S#X rr c 4\rSrSrSr\SS\S\S\\\4S\ \S\4 S jj5r \S\4S j5r \S\4S j5r \S\4S j5r \S\4S j5r\S\4Sj5r\S\4Sj5r\S\4Sj5r\S\4Sj5r\S\ \4Sj5rSrg)ToolDefinitionu工具定义辅助类Nname description parametersrequiredrc.SUUSUU=(d /S.S.S.$)u创建工具定义 Args: name: 工具名称 description: 工具描述 parameters: 参数定义 required: 必需参数列表 Returns: 工具定义字典（OpenAI 格式） functionobject)type propertiesr{)rxryrz)rr}r!rxryrzr{s rcreateToolDefinition.creates.&*$", (B  rc J[RSSSSS.SSS.SSS.S .S /S 9$) u运行命令工具定义 run_commandu"在指定目录执行 shell 命令stringu要执行的 shell 命令rryu'命令执行的工作目录（可选）integeru'命令超时时间（秒），默认 60)commandcwdrrrrvrr!rrrToolDefinition.run_commandsU$$<%#> %#L &#L  [#%  rc:[RSSSSSS.0S/S9$)u读取文件工具定义 read_fileu!读取指定路径的文件内容pathr文件的绝对路径rrrr!rrrToolDefinition.read_files<$$;$#: X%  rcD[RSSSSS.SSS.S.SS /S 9$) u写入文件工具定义 write_fileu<将内容写入指定路径的文件（自动创建目录）rrru要写入的文件内容)rr?rr?rrr!rrrToolDefinition.write_file*sK$$V%#: %#= i(%  rc:[RSSSSSS.0S/S9$)u列出目录工具定义list_directoryu*列出指定目录下的文件和子目录rru目录的绝对路径rrrr!rrrToolDefinition.list_directory=s<$$!D$#: X%  rcD[RSSSSS.SSS.S.SS /S 9$) u保存记忆工具定义 save_memoryuN保存关键上下文信息到记忆存储（用于跨阶段保持一致性）ru[记忆键名（建议使用固定key，如 challenge_name, exploit_chain, output_dir 等）ru<要保存的内容（建议结构化、简短、可复用）)keyr?rr?rrr!rrrToolDefinition.save_memoryLsN$$h%$A %#a Y'%  rc:[RSSSSSS.0S/S9$)u读取记忆工具定义 read_memoryu 读取指定 key 的记忆内容rru要读取的记忆键名rrrr!rrrToolDefinition.read_memory_s<$$:$#= W%  rc.[RSS0/S9$)u列出记忆工具定义 list_memoriesu(列出当前已保存的所有记忆 keyrrr!rrrToolDefinition.list_memoriesns($$ B %  rc8[RSSSSSS.0/S9$)u清空记忆工具定义clear_memoriesu;清空所有记忆或按 key 前缀清空（谨慎使用）prefixru1可选：仅清空以该前缀开头的记忆 keyrrrr!rrrToolDefinition.clear_memoriesxs:$$!U$#V %  rc[R5[R5[R5[R 5/$)u,获取 CTF 题目生成所需的所有工具)rvrrrrr!rr get_ctf_toolsToolDefinition.get_ctf_toolss>  & & (  $ $ &  % % '  ) ) +   rr!)N)rmrnrorprq staticmethodrIrrrrrrrrrrrrrrtr!rrrvrvsM #    cN s)     <   ,  t       $  D       $       4    D     4:  rrv)rqabcrrtypingrrrrr logging getLoggerrmrGr rvr!rrrsC $77   8 $PSPfm m r