从 500MB 到 50MB:我用 Go 写了个边缘 AI 记忆体,让你的设备多跑一个大模型

一、边缘开发者的崩溃瞬间

做边缘 AI 开发的朋友,应该都遇到过这种场景:

你在一块 RK3588 或者树莓派上部署 AI Agent,满心欢喜地准备跑个大模型推理。结果光是启动个 InfluxDB 或者 PostgreSQL,内存就被吃掉了 500MB+,留给模型的算力所剩无几。

换用 SQLite 吧,数据一多就锁死,断网续传时数据直接堆积卡死。想做个本地 RAG(检索增强生成)?抱歉,向量存储还得额外装插件。

这不是你的问题,是工具选错了。

二、解决方案:sfsAI 边缘数据底座

为了解决这个痛点,我用纯 Go 写了一个专为边缘 AI 设计的数据底座 —— sfsAI。它不做通用数据库,只聚焦三件事:

核心能力 说明
记忆存取 原生支持向量存储,为 AI Agent 提供持久化记忆
隐私安全 AES-256-GCM 透明加密,设备绑定密钥,数据不出设备
就近服务 Sidecar 模式部署,毫秒级响应,零网络延迟

极致轻量是它的灵魂

  • 二进制体积:8MB
  • 常驻内存:50MB(峰值)
  • 启动时间:< 100ms

三、架构设计:小而美的典范

sfsAI 采用经典的三层架构:

AI Agent (Python/Go/C++)
      │ HTTP/JSON
      ▼
┌─────────────┐   Sidecar 进程,伴生运行
│   sfsAI     │   聚焦 AI 记忆 API
└──────┬──────┘
       │ import
       ▼
┌─────────────┐
│   sfsDb     │   LSM-Tree 嵌入式引擎
└─────────────┘

核心模块

1. Memory 子系统 — AI 记忆体核心

  • MemoryUnitmemory_id + session_id + content + embedding + metadata
  • Conversation Stream(session_id, created_at) 复合索引,毫秒级时序扫描
  • 语义搜索:余弦相似度向量检索,原生支持 RAG 场景

2. Security 子系统 — 零信任安全

  • AES-256-GCM:透明加密,默认启用
  • 设备绑定密钥:密钥与设备 ID 绑定,硬盘拆卸也无法解密
  • 被遗忘权:物理删除,配合密钥销毁实现真正遗忘

3. Sidecar 子系统 — 生命周期管理

  • 常驻进程,优雅关闭(SIGINT/SIGTERM)
  • 健康检查,让 Agent 随时感知服务状态

四、硬核性能对比

在 RK3588(4GB 内存)上的实测数据:

指标 sfsAI SQLite InfluxDB
空闲内存 ~50MB ~15MB 500MB+
10万条写入 < 10s ~30s ~25s
向量检索(1000条) < 50ms - -
部署复杂度 单文件 简单 复杂

五、5 分钟上手

1. 启动服务

go run ./cmd/sfsai -db /data/my-device -addr :8630

2. 写入记忆

import "sfsAI/pkg/sdk"

client := sdk.NewClient("http://localhost:8630")

// 写入一条记忆
result, _ := client.InsertMemory(
    "user-1",                              // session_id
    "用户说:帮我打开客厅空调",              // content
    []float32{0.1, 0.2, 0.3},              // embedding(可选)
    map[string]interface{}{"source": "voice"}, // metadata
)

3. 语义搜索

// 根据向量查找相似记忆
queryVector := []float32{0.15, 0.25, 0.35} // 来自 embedding 模型
results, _ := client.SemanticSearch("user-1", queryVector, 5)

for _, mem := range results.Memories {
    fmt.Println(mem.Content)
}

4. 被遗忘权

// 擦除用户所有记忆(物理删除)
client.WipeMemories("user-1")

六、能帮你做什么?

1. 硬件降本

如果你的设备原本因为数据库吃内存而选配 4GB 内存,换上 sfsAI 后,或许可以降级到 2GB,单台 BOM 成本立省几十块

2. 断网续传神器

采用 LSM-Tree 架构,专为高频写入优化。工厂网络再差,数据也能稳稳存在本地,绝不丢失。

3. AI 记忆原生支持

不管是做边缘 RAG 还是 AI Agent 持久化记忆,一个 SDK 全搞定,无需额外部署向量数据库。

七、适用场景

  • 边缘 AI 盒子:智能家居中控、工业巡检仪
  • 工业网关:数据采集、本地存储、断点续传
  • AI Agent:记忆持久化、上下文管理、RAG 检索
  • 车载设备:本地日志、传感器数据、离线推理

八、获取方式

sfsAI 已开源在 GitHub,采用 MIT 协议,完全免费使用:

git clone https://github.com/liaoran123/sfsAI.git
cd sfsAI
go run ./cmd/sfsai

如果你是做边缘 AI 或者工业网关的开发者,欢迎拿去实测! 如果你是硬件方案商,也欢迎联系我探讨预装合作。


写在最后:在边缘计算越来越重要的今天,我们需要的不是又一个通用数据库,而是真正为边缘场景优化的专用工具。sfsAI 就是这样一个尝试 —— 小而美,但足够强大。

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐