1. 先聊聊我为什么放弃"全家桶式云API",转投本地推理
这事得从半年前说起。当时我手头有好几个项目都依赖云端大模型API,多模态理解、代码生成、长文本摘要全往里塞,月底一看账单,心里拔凉拔凉的。更要命的是,只要上游服务一限流或者某个模型突然下架,我的业务就得跟着抖动。做过在线服务的人都知道,"依赖第三方推理能力"这件事,本质上是把系统的可用性和成本控制权交到了别人手里。
所以当AI进入本地时代这个趋势越来越明显的时候,我几乎是第一时间就决定把推理能力收归己有。自己组一台机器,把主流模型跑在本地,不光是省钱的考虑,更重要的是能拿到完整的控制权——数据不出内网、请求不排队、模型随意换。而且随着开源模型生态的爆发,本地推理早就不是那种"能跑但很勉强"的玩具状态了,很多场景下的效果已经逼近甚至追平云端旗舰模型。
但"跑通一个模型"和"搭建一个推理平台"是两码事。前者用一条命令行就能搞定,后者要把多模型管理、Agent调度、Workflow编排这些工程问题全部串起来。我开源的这个推理平台,就是在这半年里被这些问题逼出来的产物:它把多模型接入、Agent执行、Workflow编排做成了一个开箱即用的本地服务,16G显存级别的消费级显卡也能搭起来。
这篇文章我会从架构设计讲到落地的踩坑细节,动手能力强的朋友可以直接照着搭一套,不感兴趣代码的也能从里面看到本地推理平台在工程层面到底要解决哪些问题。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 平台的整体架构:多模型、Agent、Workflow 是怎么在同一套系统里协作的
2.1 核心定位:不是一个"模型启动器",而是一个"推理操作系统"
先给这个平台定个性。市面上有不少工具能做模型加载和简单对话,但它们大多数停留在"单一模型启动"的层面,离"平台"还差一个量级。我对自己这个开源项目的定位是:它应该像操作系统管理进程一样管理模型,像消息队列一样调度请求,像工作流引擎一样编排任务。
整个系统从上到下分成了四层:
- 接入层:统一HTTP接口,兼容OpenAI风格的API格式,方便迁移。
- 调度层:负责多模型路由、并发控制、请求排队。
- 执行层:跑的是Agent和Workflow运行时,也是这个平台最有价值的部分。
- 推理层:对接各类推理后端,管理显存、进程和模型生命周期。
为什么一定要做多层拆分?因为模型本身的推理和业务逻辑的执行是两套完全不同频率的事情。模型推理一次可能花几秒,而一个Agent循环可能要调几十次模型;Workflow里还会夹着普通的代码执行、外部接口调用。如果全揉在一起,出问题的时候根本没法定位是模型挂了还是逻辑挂了。
2.2 多模型管理:不是"加载多个模型",而是"聪明地分配模型"
多模型支持听起来简单,实际上有不少坑要趟。第一个问题是显存。单卡16G显存,要同时驻留一个大尺寸模型和一个小尺寸模型,就得严格控制每个模型的显存占用。我的方案是把模型按尺寸分为"常驻级"和"按需级",常驻的占住显存不动,按需的用完之后立刻释放。虽然频繁加载模型会增加延迟,但在多模型场景下,资源冲突的概率会小很多。
第二个问题是路由策略。平台内置了几种路由方式,最基础的是模型名匹配,即请求里声明要用哪个模型就路由到哪个模型;进阶一点的是按任务类型自动路由,比如意图识别类的任务默认走小模型,复杂推理默认走大模型;还有一种是按队列长度动态分配,哪个模型的积压任务少就往哪个模型上送。
我们可以看一个实际的路由配置示例,这里展示按任务类型自动路由的设置方式:
yaml复制model_router:
default: qwen2.5-7b-instruct
task_routing:
intent_detection: qwen2.5-3b-instruct
code_generation: deepseek-coder-6.7b
visual_understanding: llava-v1.6-7b
这样的设计带来一个很直接的好处:整体吞吐上去了,而GPU的利用率也平滑了很多,不会出现一个模型闲着、另一个模型排队排到冒烟的情况。
2.3 Agent运行时:让模型具备"感知—决策—行动"的闭环能力
如果说多模型管理是平台的地基,那Agent运行时就是平台的灵魂。我的Agent设计参考了业界主流的ReAct模式,核心循环是:模型接收任务,观察当前环境,推理下一步行动,选择一个工具执行,再观察结果,循环往复,直到任务完成。
这个循环里最难的地方在于工具调用的可靠性。模型输出的JSON格式稍微错一个引号,整个解析就崩了。我的做法是双重保障,第一,在Prompt里给出严格的工具调用格式规范,强制模型按固定schema输出;第二,在解析层做容错处理,即使JSON不完整,也能通过正则和括号配对进行局部修复。这两层叠加之后,工具调用的成功率从最初的八成五左右提升到了九成七以上。
这里是一个Agent工具的注册代码片段,展示了自定义工具接入的基本方式:
python复制@agent.tool("search_codebase")
def search_codebase(query: str) -> str:
"""在本地代码库中搜索相关代码片段"""
results = codebase_index.search(query, top_k=5)
return "\n---\n".join(str(r) for r in results)
Agent的另一个关键设计是状态管理。Agent执行期间会产生很多中间变量和上下文片段,平台把这些问题都塞进独立的上下文管理器里,每个任务有自己独立的上下文空间,互不干扰。这样多个Agent并行跑的时候,不会出现上下文串号的诡异问题。
2.4 Workflow编排:用可视化逻辑把复杂任务拆成可复用的管道
Agent适合处理"自由发挥"的任务,但有些业务需要的是固定路径的执行,比如:先做文本清洗,再做实体识别,然后调用翻译服务,最后汇总成报告。这种流程化的需求,就该上Workflow。
Workflow模块的设计灵感来自业界成熟的流式处理框架,每个节点是一个独立的执行单元,节点之间通过有向图连接,数据以参数形式在节点间传递。平台支持顺序执行、条件分支、并行分支和循环四种基本结构,基本能覆盖大多数业务逻辑。
下面是一个Workflow定义示例,展示一个"日报生成"流程的节点配置方式:
json复制{
"workflow": "daily_report",
"nodes": [
{"id": "collect", "type": "code", "action": "git_log_collector", "params": {"days": 1}},
{"id": "summarize", "type": "agent", "prompt": "根据以下commit信息生成每日摘要", "deps": ["collect"]},
{"id": "format", "type": "code", "action": "markdown_renderer", "deps": ["summarize"]}
]
}
Workflow相比裸写Agent脚本有什么优势?最直观的是可观测性。每个节点的输入输出都会被记录到执行日志里,哪一步耗时最长、哪一步输出异常,一眼就能看到。排错效率比直接读千百行Agent代码高太多了。
3. 实践落地:16G显存级别的本地部署实操记录
3.1 硬件选型的心路历程:为什么16G显存是一个"黄金甜点"
先说硬件。要做本地推理,GPU显存是绕不开的第一道门槛。24G显存的卡固然很香,但价格劝退了一大波人;8G显存也不是不行,但能跑的模型尺寸被卡得很死,多模态模型基本没戏。综合来看,16G显存是目前性价比最平衡的档位——既跑得动7B~13B级别的模型,包括一些多模态模型,也够塞进一个Agent运行时加几个小模型。
我的实际配置是单张RTX 4080 16G,配合32G内存和1T NVMe固态。为什么内存要32G?因为模型在加载过程中需要先经过CPU内存再拷入显存,内存小了,大模型加载的时候直接OOM。这个配置基本能满足我日常所有需求:同时常驻一个7B对话模型和一个3B分类模型,再用剩余显存跑一个6.7B的代码模型。
3.2 推理后端的选型对比:vLLM、llama.cpp、Ollama 哪个才是最优解?
推理后端是整个平台的性能底座,不同的推理后端在不同场景下的表现差异很大。我把市面上主流的几个方案都试了一遍,收获不少。
- llama.cpp:CPU推理优化得很彻底,但GPU场景的优势不明显。适合没有独立显卡的机器备用。
- Ollama:上手极快,用户体验一流,但底层定制空间有限。做原型验证很爽,做成生产平台缺一点灵活性。
- vLLM:吞吐怪兽,支持连续批处理、PagedAttention,每秒钟能处理的请求数是普通推理框架的数倍。但部署配置相对复杂,对显存的利用也比较激进。
我的选择是拿vLLM作为主力推理后端,因为它的吞吐性能在服务多请求场景下表现最好,尤其适合我这种需要同时服务多个Agent并发请求的平台。只有跑一些vLLM暂不支持的模型时,我才退回到llama.cpp作为兼容后端。平台在推理层做了一层抽象,后端可以随时切换,这也是一个关键的设计决策。
用Docker跑一个vLLM推理服务比较简单,参数如下:
bash复制docker run -d --gpus all \
-p 8000:8000 \
vllm/vllm-openai:latest \
--model Qwen/Qwen2.5-7B-Instruct \
--gpu-memory-utilization 0.85 \
--max-model-len 32768 \
--enforce-eager
这里--gpu-memory-utilization 0.85的意思是让vLLM最多占用85%的显存,留出一部分余量给其他模型或Agent处理的中间数据。--enforce-eager则是禁用CUDA Graph以换取更平稳的显存占用,这在多模型共存的场景下非常实用。
3.3 模型加载策略:常驻、按需、动态驱逐
模型的加载和卸载策略决定了平台的资源利用效率。我的方案是建立一个模型生命周期管理组件,把模型分为三个状态:常驻、休眠、已卸载。
设置常驻模型的依据是调用频率。只要一个模型每小时被调用的次数超过了阈值,就把它常驻在显存里,省去反复加载的时间。休眠状态是指模型暂时空闲,但不卸载,超过一定时间没人调用再转入已卸载状态。每个模型在平台里都有一个独立的最小化进程包装,任何模型的崩溃都不应该拖垮整个平台的运行。
这里有个非常有用的实际操作。平台启动的时候,不要把所有模型一次性加载到显存,而是采用"懒加载"策略——只有请求第一次到达时才初始化模型。否则,你可能等半天平台才启动完成,结果发现大多数时间只有一两个模型在工作。懒加载配合动态驱逐策略,可以让16G显存跑出20G显存的效果。
3.4 请求调度与并发控制:如何在资源不够时优雅地排队
并发控制是我踩坑最多的地方。大模型的推理延迟是秒级的,一个模型同时只能处理有限的并发,如果所有请求一股脑灌进来,显存会瞬间透支或者进程直接崩溃。
解决思路是引入两层限流。第一层是模型级别限流:每个模型维护一个请求队列,队列长度上限可配置,超过上限的请求直接返回HTTP 503或者加入缓存等待。第二层是全局并发控制:平台记录所有模型的实时并发数,当总并发达到硬件上限时,新请求进入全局等待队列。这个设计保证了在极端情况下,平台是"变慢"而不是"崩溃"。
平台在调度层内置了基于令牌桶的流量整形,可以针对不同客户端设置不同的调用速率。比如管理员可以设置"某个Agent每秒最多调用模型10次",防止有异常的Agent脚本陷入死循环疯狂消耗资源。
4. 搭建过程中踩过的那些坑:从显存爆炸到Agent死循环
4.1 显存为什么会突然爆炸?排查链路记录
这个坑几乎每个做本地推理的人都会遇到。现象是平台运行一切正常,突然某一次请求之后,显存使用率飙升到100%,然后进程被OOM Killer干掉。最诡异的是复现起来非常随机,完全找不出规律。
排查的过程分了三步走。第一步查现象,用nvidia-smi和nvitop轮流观察显存变化,发现显存是"阶梯式上升"的,每次请求之后都会高一点,但是回落不到原来的水平。这说明有显存泄漏。第二步定位模块,我怀疑是vLLM的KV Cache没有正确释放,于是对不同的模型后端分别做压力测试,结果发现llama.cpp后端非常稳定,vLLM后端存在泄漏现象。第三步查根因,翻了vLLM的GitHub Issues,发现这确实是一个已知问题,频繁加载/卸载LoRA或切换模型时,PagedAttention的显存块管理器没有完全回收。解决办法是在模型空闲超过阈值时彻底销毁进程并重新拉起,而不是单纯从显存中卸载权重。这个方案很粗暴,但有效。
4.2 Agent执行超时:为什么30次工具调用是最优解?
Agent死循环是另一个高频事故。模型在两三个工具之间来回调用,每次都有一点新东西,但就是不收敛到最终答案。这不但浪费算力,还占用着一整个Agent并发配额,把其他正常任务都堵住了。
解决办法不是"尽量让模型聪明一点",而是从工程上加护栏。我给每个Agent执行周期设定了最大工具调用次数,默认是30次。这个数字选得很有讲究,少于15次,一些复杂的多步任务会做不完;多于50次,等待时间太长,用户会失去耐心。30次是一个比较合理的甜点值,既给了Agent充分的执行空间,又能在失控时及时止损。
平台还支持单步超时,即任何一次工具调用的执行时间超过60秒就主动中断,让模型重新规划方案。这两个超时策略的组合,让Agent任务的成功率提升了好几个百分点,用户的等待体验也好多了。
4.3 Workflow调试的"黑盒"困境:日志重要性胜过一切
Workflow刚开始上线的时候,最让人崩溃的是:节点A的输出明明看着不对,但你去查节点B的输入,发现它拿到的却是"据预期值"。这种问题用肉眼看代码根本找不出来,因为问题不在逻辑,而在数据传递。
后来我强制规定平台里所有Workflow节点在执行前和执行后都要做一次数据快照,把输入输出的结构信息和前128字节内容记录到日志里。一旦出现异常,翻日志就能定位是哪个节点的输出格式不符合下一个节点的预期。后来我又加了一个JSON Schema校验器,每个节点的输出都先做一轮结构校验,再决定是往下游传还是直接中断报错。实践下来,这个提前校验的动作减少了至少60%以上的数据链条问题。
另外,在写Workflow时尽量让节点保持"纯函数"特性——同样的输入永远产生同样的输出,不要依赖全局状态。比如需要读取时间、随机数、环境变量的逻辑,不要直接散落在节点的执行代码里,而是通过Workflow的上下文对象显式传入。这一点对排查和复现问题至关重要。
4.4 多模型并发导致的推理性能雪崩
当同时有多个Agent在跑不同的模型时,平台会面临一个特殊的性能问题:显存总占用没超限,但推理变得异常慢,每个请求的延迟都翻倍甚至更多。表面上看是"算力不够",实际原因是显存带宽被多个模型同时访问给拖垮了。
GPU的显存访问带宽是有限资源,多个模型并发推理时,权重参数和KV Cache的读取会在带宽上打仗,导致每个模型的effective throughput都严重下降。解决方案有两个方向:一是任务挤堆,把同一类模型的小请求合并成batch走vLLM的continuous batching,提高单个模型的计算密度;二是错峰调度,在调度层做时间片轮转,把对不同模型的推理请求稍微打散,避免同时挤爆带宽。两种方法都实际测试过,后者的代码改动更小,收益也很明显。
5. 从"能跑"到"可用":这半年里我学到的工程经验
5.1 用OpenAI兼容接口是为了什么?
平台对外暴露的接口完全兼容OpenAI的/v1/chat/completions格式。这个决定在最初做的成本评估里看似多此一举,但后续证明这是性价比最高的一个设计。第一,市面上几乎所有的Agent框架和AI应用开发工具都原生支持OpenAI接口,我不用为任何生态做适配;第二,团队里已有的代码迁移到本地时,只需要改一个base_url,逻辑完全不用动。
事实上,我后来把好几个原本用云端服务的应用切到本平台时,整个过程就是改个环境变量的事。这种"零改造迁移"带来的便利,远超当初实现兼容层的那些工作量。
5.2 日志和可观测性是平台的"安全气囊"
做本地推理平台,最容易忽视的就是可观测性。因为"模型能跑"给人的错觉太大了,你以为服务正常,其实显存已经告急、队列已经堆积、模型已经悄悄降智了。
我在平台里做了三层可观测性:第一层是系统指标,包括GPU显存占用、模型请求延迟、队列长度,全部通过Prometheus格式暴露出来;第二层是请求追踪,每一个从入口到模型再到Agent工具的完整调用链,都记录一条trace;第三层是业务日志,记录Agent每一步的思考、调用、返回。有了这三层数据,系统出任何问题,都能在五分钟内定位个大概。
给所有准备搭平台的朋友一个建议:宁可功能少一点,也要把日志做厚。很多问题在没有日志的时候可能要排查一整天,有了日志之后可能一眼就看穿了。
5.3 安全边界:本地平台也得有"门禁"
虽然平台默认跑在内网,但安全问题不能完全忽略。至少要做到三层防护。
认证是必须的。平台内置了API Key机制,调用方必须在请求头里携带密钥。如果你要把平台暴露到公网,建议在前面再套一层反向代理做IP白名单。
还要提防Prompt注入。只要Agent具备工具调用能力,恶意的Prompt就可能诱导它执行危险操作,比如读取本机敏感文件、调用删除接口等。我的办法是在工具层做权限标注,高风险的工具有单独的授权开关,而且Agent默认没有权限执行系统级的写操作。
最后是资源配额。每个客户端的并发数、Token消耗量、队列长度都要做上限约束,防止一个异常的接入方打垮整个平台。安全设计宁可做得过度,也不能留短板。
5.4 下一步的计划:多模态增强、高可用和更聪明的Agent记忆
这个开源项目目前已经能稳定支撑我日常的开发需求,但还有几个明显的改进方向。第一个方向是多模态增强,现在虽然已经支持视觉类模型,但在音视频理解、多模态混合推理这些方向还有不少路要走。第二个方向是高可用,目前是单机部署,下一步计划加入多机分布式调度,把不同机器的GPU资源统一管理和分配。第三个方向是Agent记忆,我想在平台内引入向量数据库做长期记忆存储,让Agent能记住历史交互中的关键信息,而不是每次对话都从零开始。
这些功能不是一次性就能做完的,我打算按里程碑逐步推进,每一个版本都会尽量保证向后兼容。有兴趣一起搞的朋友,可以在GitHub上提Issue或者直接提PR。
6. 开源地址与快速上手指引
最后是干货环节。整个项目代码已经推到GitHub上,仓库名就不在这里刷存在感了,直接说怎么跑。
快速搭建步骤如下:
- 克隆项目代码。
- 按官方文档里的说明安装依赖,推荐Python 3.10+,CUDA 12.x。
- 修改
config.yaml里的后端配置,选择vLLM或llama.cpp。 - 运行启动命令,打开管理界面,在界面上配置你要使用的模型。
- 调用本地API接口开始使用,默认地址是
http://localhost:8090/v1。
项目自带的examples/目录下有几个现成的示例,包括一个简单的Agent工具调用演示和一个多步骤Workflow定义,照着跑一遍基本能理解整个系统的设计思路。
按照我个人的使用体验,从零开始搭建到稳定运行,大概需要一到两个周末的时间。如果你之前用过Ollama这类工具,上手速度会更快。只要过了第一道门槛,后面接自己的业务场景就只是写配置和写工具函数的事。
这篇稿子断断续续写了两天,回头再看,这几年AI开发工具链最缺的就是一个好用的本地底座,固化自己的工程经验、开源出去让更多人少走弯路,是这个项目最朴素的愿望。有任何部署或者使用上的问题,欢迎在GitHub的Discussions区交流,我看到都会回复。
