基于大模型的 MCU 代码生成器

把芯片数据手册变成可检索的本地知识库,让大模型照着寄存器定义生成 MCU 驱动代码。 面向嵌入式开发场景,避免「模型不熟悉这颗芯片的外设寄存器」这个最常见的问题。

• PDF 数据手册导入并切分入库(pypdf 解析,放入独立进程执行,避免与 Qt 界面冲突)
• 向量检索(ChromaDB + all-MiniLM-L6-v2 嵌入模型),按提问召回相关寄存器段落
• 结合召回片段构造提示词,调用 OpenAI 兼容接口生成驱动代码
• 桌面界面(PyQt5 + QScintilla 代码编辑器,C/C++、JSON 语法高亮)
• 多进程 PDF 导入并回传进度,API 配置通过 .env 管理

main.py 程序入口、API 配置与主窗口装配
app_ui.py 界面组件与设置弹窗
knowledge_base.py ChromaDB 知识库与检索
pdf_import_worker.py 独立进程的 PDF 导入工作器
test_chromadb.py 向量库连通性测试

pip install -r requirements.txt
cp .env.example .env # 填入自己的 API Key
python main.py

仓库不包含数据手册原文、本地向量库与打包产物。密钥只从 .env 读取,请勿写进源码。