《AI大模型应用开发实战从入门到精通共60篇》059、完整项目实战：构建一个“嵌入式知识库问答机器人”

059、完整项目实战构建一个“嵌入式知识库问答机器人”昨晚调一个RAG的embedding对齐问题到凌晨三点发现罪魁祸首是tokenizer的padding策略没统一——这种坑文档里永远不会写。今天把整个项目从零到部署的完整过程拆开揉碎代码里该踩的坑我都替你踩过了。项目背景为什么是嵌入式知识库客户现场的设备手册、芯片datasheet、调试日志这些资料动辄几百页PDF工程师查个寄存器定义要翻半天。我们要做的就是把这些非结构化文档喂给大模型让它能像资深FAE一样回答问题。技术选型上我放弃了LangChain全家桶——太重调试链路太长。改用轻量级方案HuggingFace Embedding ChromaDB FastAPI 本地部署的Qwen2.5-7B。别问我为什么不用GPT-4客户现场没网而且7B模型在RTX 4090上跑推理延迟能压到200ms以内。第一步文档解析——PDF里的文字不是你想的那样# 别用PyPDF2它对扫描件和表格的处理简直是灾难fromlangchain_community.document_loadersimportPyMuPDFLoaderimportredefload_embedded_docs(pdf_path):loaderPyMuPDFLoader(pdf_path)docsloader.load()# 这里踩过坑PDF解析出来的文本经常有奇怪的换行符# 比如寄存\n器这种不处理的话embedding会碎成渣fordocindocs:doc.page_contentre.sub(r\n(?[a-z]),,doc.page_content)# 小写字母前的换行去掉doc.page_contentre.sub(r(?[。])\s*\n,\n\n,doc.page_content)# 句号后保留分段returndocs注意如果你的PDF里有表格PyMuPDFLoader会把它当成连续文本。我后来加了个启发式规则——检测连续空格超过3个的行单独提取成表格片段。这部分代码不贴了太脏。第二步文本分块——chunk_size不是越大越好fromlangchain.text_splitterimportRecursiveCharacterTextSplitter# 别这样写text_splitter RecursiveCharacterTextSplitter(chunk_size1000, chunk_overlap200)# 嵌入式文档里寄存器描述经常只有一两句话1000的chunk会把不同寄存器混在一起text_splitterRecursiveCharacterTextSplitter(chunk_size512,# 实测512对技术文档最友好chunk_overlap64,# 重叠太少会丢失上下文太多会引入噪声separators[\n\n,\n,。,,, ],# 中文优先按句号切length_functionlen,)chunkstext_splitter.split_documents(docs)这里有个血泪教训chunk_overlap设成200时检索出来的片段经常包含两个不相关的话题。后来改成64召回率反而提升了12%。原因可能是嵌入式文档的段落边界清晰不需要太多重叠。第三步Embedding与向量库——选对模型比调参重要fromsentence_transformersimportSentenceTransformerimportchromadbfromchromadb.configimportSettings# 别用text-embedding-ada-002本地部署要命# 推荐BAAI/bge-small-zh-v1.5768维速度比bge-large快3倍embedding_modelSentenceTransformer(BAAI/bge-small-zh-v1.5,devicecuda# 别忘记指定设备默认CPU慢到哭)# ChromaDB的持久化路径一定要用绝对路径别问我怎么知道的clientchromadb.PersistentClient(path/data/embeddings/embedded_kb,settingsSettings(anonymized_telemetryFalse)# 关掉遥测客户现场敏感)collectionclient.get_or_create_collection(nameembedded_qa,metadata{hnsw:space:cosine}# 余弦距离比L2更适合语义检索)# 批量写入别一条一条add会慢到怀疑人生batch_size64foriinrange(0,len(chunks),batch_size):batchchunks[i:ibatch_size]texts[doc.page_contentfordocinbatch]embeddingsembedding_model.encode(texts,normalize_embeddingsTrue)# 归一化归一化归一化collection.add(ids[fchunk_{j}forjinrange(i,ilen(batch))],embeddingsembeddings.tolist(),documentstexts,metadatas[{source:doc.metadata.get(source,)}fordocinbatch])归一化这一步我忘了两次。第一次是余弦距离没归一化检索结果全是乱序第二次是归一化后忘了更新索引导致新写入的向量和旧向量不在同一个空间。ChromaDB不会报错但结果就是错的。第四步检索增强生成——RAG的核心是检索defretrieve_context(query,top_k5):# 查询向量也要归一化和写入时保持一致query_embeddingembedding_model.encode(query,normalize_embeddingsTrue)resultscollection.query(query_embeddings[query_embedding.tolist()],n_resultstop_k,include[documents,distances])# 这里有个trick过滤掉距离大于0.7的结果避免噪声# 0.7这个阈值是我在STM32手册上试出来的不同文档可能需要微调filtered_docs[]fordoc,distinzip(results[documents][0],results[distances][0]):ifdist0.7:filtered_docs.append(doc)# 如果过滤后为空返回原始结果中距离最小的那个ifnotfiltered_docs:filtered_docs[results[documents][0][0]]return\n\n.join(filtered_docs)距离阈值这个参数我一开始设0.5结果很多相关文档被过滤掉了。后来改成0.7在测试集上F1分数从0.72涨到0.85。但要注意如果文档质量差比如扫描件OCR错误多阈值要适当放宽。第五步Prompt模板——别让模型自由发挥# 这个prompt我迭代了7版最终版长这样SYSTEM_PROMPT你是一个嵌入式系统专家请基于以下参考资料回答问题。 要求 1. 如果参考资料中没有相关信息直接说根据现有资料无法回答不要编造 2. 回答时引用具体的寄存器名称、地址或参数值 3. 如果涉及配置步骤按顺序列出 4. 不要添加参考资料中没有的假设 参考资料 {context} 用户问题{question} 回答defgenerate_answer(question,context):promptSYSTEM_PROMPT.format(contextcontext,questionquestion)# 这里用Qwen2.5-7B的chat模板别自己拼字符串messages[{role:system,content:你是一个嵌入式系统专家。},{role:user,content:prompt}]# 推理参数temperature设低别让模型发散responsemodel.chat(messages,temperature0.1,max_tokens512,top_p0.9)returnresponsetemperature设0.1是因为我发现当模型不确定时高temperature会编造寄存器地址。有一次它告诉我USART1的基地址是0x40013800实际上是0x40011000——这种错误在嵌入式领域是致命的。第六步FastAPI部署——异步处理是必须的fromfastapiimportFastAPI,HTTPExceptionfrompydanticimportBaseModelimportuvicornfromthreadingimportLock appFastAPI()# 模型加载是重量级操作用单例模式别每次请求都加载classModelSingleton:_instanceNone_lockLock()def__new__(cls):ifcls._instanceisNone:withcls._lock:ifcls._instanceisNone:# 这里用4bit量化显存占用从16G降到6GfromtransformersimportAutoModelForCausalLM,AutoTokenizer modelAutoModelForCausalLM.from_pretrained(Qwen/Qwen2.5-7B-Instruct,device_mapauto,load_in_4bitTrue,# 别用8bit速度慢一倍bnb_4bit_compute_dtypefloat16)tokenizerAutoTokenizer.from_pretrained(Qwen/Qwen2.5-7B-Instruct)cls._instance(model,tokenizer)returncls._instanceclassQueryRequest(BaseModel):question:strtop_k:int5app.post(/ask)asyncdefask_question(request:QueryRequest):try:contextretrieve_context(request.question,request.top_k)answergenerate_answer(request.question,context)return{answer:answer,context:context}exceptExceptionase:# 别把原始错误返回给客户端记录日志就好raiseHTTPException(status_code500,detailInternal error)if__name____main__:uvicorn.run(app,host0.0.0.0,port8000,workers1)# workers1避免模型重复加载workers设1是因为多进程下每个worker都会加载一份模型显存直接爆炸。如果并发要求高用Ray Serve或者vLLM做模型推理服务化。第七步性能优化——从2秒到200毫秒上线后发现首轮响应要2秒用户反馈太慢了。排查发现瓶颈在embedding和模型推理。# 1. embedding缓存相同问题不重复计算fromfunctoolsimportlru_cachelru_cache(maxsize128)defget_embedding(text):returnembedding_model.encode(text,normalize_embeddingsTrue)# 2. 模型推理用vLLM替代原生transformers# 安装pip install vllmfromvllmimportLLM,SamplingParams llmLLM(modelQwen/Qwen2.5-7B-Instruct,tensor_parallel_size1,gpu_memory_utilization0.9)sampling_paramsSamplingParams(temperature0.1,max_tokens512)defgenerate_answer_vllm(prompt):outputsllm.generate([prompt],sampling_params)returnoutputs[0].outputs[0].textvLLM的PagedAttention机制让显存利用率提升3倍推理延迟从800ms降到150ms。但注意vLLM不支持所有模型Qwen系列是官方支持的。踩坑记录那些文档里没有的ChromaDB的持久化路径相对路径在重启后可能丢失必须用绝对路径。embedding归一化写入和查询必须一致否则余弦距离计算错误。chunk_size与文档类型技术文档用512小说类用1024代码文档用256。模型量化4bit量化损失约3%的准确率但显存占用降低60%值得。并发控制单模型实例下用asyncio.Lock控制并发避免显存OOM。个人经验别追求完美先跑起来这个项目从构思到上线用了两周第一周全在调embedding和分块参数。后来发现与其花时间调参不如先把流程跑通然后根据用户反馈迭代。比如距离阈值0.7一开始我设0.5用户说有些问题答不上来。改成0.7后召回率上去了但偶尔会引入噪声。后来加了个后处理规则如果检索结果中有多个文档按距离排序后只取前3个再过滤掉距离大于0.7的。这样既保证了召回又控制了噪声。最后说一句别迷信RAG的黄金参数。不同领域、不同文档格式最优参数天差地别。我的做法是准备一个测试集50个问题标准答案每次改参数后跑一遍看F1分数变化。自动化调参可以用Optuna但手动调几轮也能找到不错的点。项目代码已上传到GitHub搜索embedded-qa-robot就能找到。有问题评论区见我每天睡前会看。