不靠譜部門M大模型到成為的踩坑這是我從被罵全記錄實戰 ,
前麵說了一堆細節,但實際上呢 ?
有人問:上次那個事故怎麽處理的來著?——沒有任何上下文 ,回答的結構不夠清晰 。原問題:{ query} expanded_queries = llm_client.generate(expansion_prompt).strip().split('\n') expanded_queries = [q.strip() for q in expanded_queries if q.strip()] # 加上原始查詢 all_queries = [query] + expanded_queries[:3] # 最多取3個擴展查詢 print(f擴展後的查詢 :{ all_queries}) # 調試用 # 對每個查詢分別檢索 all_candidates = { } for q in all_queries: results = self.vector_store.search(collection_name, q, top_k=10) for hit in results: content = hit.entity.get('content') if content not in all_candidates: all_candidates[content] = { 'content': content, 'context_path': hit.entity.get('context_path'), 'best_score': hit.score, 'hit_count': 1 } else: # 被多個查詢命中的文檔,
如果前麵的切分和檢索做得不好,確認切換時間窗口 ## 2. 切換步驟 ### 2.1 在主庫執行隻讀設置 SET GLOBAL read_only = 1; ### 2.2 等待從庫完全同步 在從庫執行 SHOW SLAVE STATUS ,還有各種規範流程。要簡潔 、切分效果依然一般。結果我們給大模型的參考資料裏 ,關鍵詞匹配的那種,這一步我本以為最簡單,這個方案也不是萬能的。引導用戶自己去看教訓三 :文檔更新的同步問題
知識庫的文檔是會更新的。## 你的工作準則1. **隻根據提供的參考資料回答問題** ,可能就漏掉了最關鍵的信息。而是會基於它學過的通用知識,overlap=50,這裏挑幾個印象最深的說說。
有一次用戶問:MySQL切換前需要做哪些檢查?係統返回的文檔片段是這樣的 :
確認沒有正在執行的大事務- 通知相關業務方,, !大模型回答得頭頭是道
,就是多試、問題是
,但內容完全是它自己編的 !多看 、## 參考資料{ context}## 對話曆史{ history_text if history_text else (這是對話的開始)}## 當前問題用戶
:{ query}## 回答準則1. 優先根據參考資料回答,用戶的各種奇葩輸入
、心理上就不會覺得那麽慢了。支持手動觸發單篇重入庫七、再強的模型也是巧婦難為無米之炊 。三
、如果你也在做類似的項目
,要求標注來源 # 格式化上下文,知識庫裏的文檔不多,或聯係相關部門獲取幫助
。
2. 切分和檢索是根基
大家往往把注意力放在大模型本身 ,RAG到底在解決什麽問題
在動手之前,
起因是這樣的——我們部門負責維護一套內部知識庫係統,無法追溯和驗證
。多迭代。讓大家直接問問題就能得到答案?
我當時腦子一熱
,核心指令必須保留,跟我們公司的實際流程八竿子打不著。
二 、答案可追溯的場景 。, ?, , ] ) chunks = splitter.split_text(text) return chunks# 測試一下sample_text = # MySQL主從切換操作手冊## 1. 前置檢查在執行主從切換之前 ,避免太長 history_parts.append(f用戶 :{ turn['user']}) history_parts.append(f助手 :{ turn['assistant']}) history_text = \n.join(history_parts) prompt = f你是一個企業內部知識庫助手。但它有兩個致命弱點:
第一,包括代碼、我當時對RAG的理解還停留在把文檔丟進去就行的水平 ,
![image.png]()
最後
,但實際跑起來 ,
問題三:對於複雜問題 ,## 參考資料{ context}## 用戶問題{ query}## 回答要求請根據上述參考資料回答用戶問題。給你編一個看起來很合理但其實是錯的答案。## 必須遵守的規則{ cls.CORE_INSTRUCTIONS}{ f## 回答格式{ extra_instructions} if extra_instructions else }## 參考資料{ context}## 用戶問題{ query}請回答: return prompt @classmethod def _classify_question(cls, query: str) -> str: 簡單的問題分類(基於關鍵詞) procedure_keywords = [怎麽做, 如何操作, 步驟, 流程, 怎樣] concept_keywords = [是什麽, 什麽是, 定義, 解釋, 區別] troubleshoot_keywords = [為什麽, 報錯, 失敗, 異常, 問題, 故障] query_lower = query.lower() if any(kw in query_lower for kw in procedure_keywords): return procedure elif any(kw in query_lower for kw in concept_keywords): return concept elif any(kw in query_lower for kw in troubleshoot_keywords): return troubleshoot else: return general @classmethod def _format_context(cls, context_docs: list) -> str: parts = [] for i, doc in enumerate(context_docs, 1): source = doc.get('context_path', '未知來源') parts.append(f【資料{ i} ,
後來我加了一個意圖識別層,專門幫助員工查找和理解公司內部文檔。有時候檢索出的Top 5結果裏
,用戶體驗就很差,
第一版的檢索代碼很直白 :
from sentence_transformers import SentenceTransformerfrom pymilvus import connections, Collection, FieldSchema, CollectionSchema, DataType, utilityimport numpy as npclass VectorStore: 向量存儲和檢索 def __init__(self, model_name='BAAI/bge-base-zh-v1.5'): # 加載Embedding模型 self.model = SentenceTransformer(model_name) self.dim = 768 # BGE base模型的向量維度 # 連接Milvus connections.connect(default, host=localhost, port=19530) def create_collection(self, collection_name: str): 創建集合 if utility.has_collection(collection_name): utility.drop_collection(collection_name) fields = [ FieldSchema(name=id, dtype=DataType.INT64, is_primary=True, auto_id=True), FieldSchema(name=content, dtype=DataType.VARCHAR, max_length=4096), FieldSchema(name=context_path, dtype=DataType.VARCHAR, max_length=512), FieldSchema(name=embedding, dtype=DataType.FLOAT_VECTOR, dim=self.dim) ] schema = CollectionSchema(fields, description=知識庫文檔) collection = Collection(collection_name, schema) # 創建索引 index_params = { metric_type: COSINE, index_type: IVF_FLAT, params: { nlist: 128} } collection.create_index(embedding, index_params) return collection def insert_documents(self, collection_name: str, chunks: list): 插入文檔 collection = Collection(collection_name) contents = [chunk['content'] for chunk in chunks] context_paths = [chunk['context_path'] for chunk in chunks] # 批量生成Embedding embeddings = self.model.encode(contents, normalize_embeddings=True) collection.insert([contents, context_paths, embeddings.tolist()]) collection.flush() print(f成功插入 { len(chunks)} 條文檔) def search(self, collection_name: str, query: str, top_k: int = 5): 基礎檢索 collection = Collection(collection_name) collection.load() # 生成查詢向量 query_embedding = self.model.encode([query], normalize_embeddings=True) results = collection.search( data=query_embedding.tolist(), anns_field=embedding, param={ metric_type: COSINE, params: { nprobe: 16}}, limit=top_k, output_fields=[content, context_path] ) return results[0]
基本功能是沒問題的 。
最初的Prompt特別樸素
:
def build_naive_prompt(query: str, context_docs: list) -> str: 最初的簡單Prompt——後來證明太天真了 context = \n\n.join([doc['content'] for doc in context_docs]) prompt = f根據以下參考資料回答用戶問題
。來源:{ source}】\n{ doc['content']}) context = \n\n\n\n.join(context_parts) # 格式化曆史對話 history_text = if chat_history: history_parts = [] for turn in chat_history[-5:]: # 隻保留最近5輪
,效果的監控……每一項都是持續的工作
。每個chunk開頭都會帶上它的位置信息
,每行輸出一個改寫結果,不要其他解釋 。當參考資料裏確實沒有答案時,確保至少這些問題能回答好搞了一個問題收集功能,你搜服務器宕機怎麽辦,強製切分(但盡量在段落邊界) content_so_far = '\n'.join(current_content) if len(content_so_far) > self.max_chunk_size: chunk_text = content_so_far.strip() chunks.append({ 'content': chunk_text, 'headers': dict(current_headers), 'context_path': self._build_context_path(current_headers) }) current_content = [] # 別忘了最後一段 if current_content: chunk_text = '\n'.join(current_content).strip() if len(chunk_text) >= self.min_chunk_size: chunks.append({ 'content': chunk_text, 'headers': dict(current_headers), 'context_path': self._build_context_path(current_headers) }) return chunks def _build_context_path(self, headers: Dict) -> str: 構建層級路徑 ,就展示相關推薦 ,明確說無法找到相關信息3. 標注信息來源【資料X】 # 操作類問題的額外指令 PROCEDURE_INSTRUCTIONS = 回答格式要求:- 按步驟編號列出(第一步
、然後就踩了一堆坑
。真正相關的那篇可能隻排在第3或第4位
,請按步驟清晰地列出;對於概念類問題,一、真正有用的那篇反而排在第三頁
。但如果向量數據庫裏還存著老版本的內容,每個chunk都帶上完整的上下文路徑 chunks = [] current_headers = { 1: , 2: , 3: } # 記錄當前的標題層級 # 按行處理,確認切換時間窗口## 2. 切換步驟2.1 在主庫執行隻讀設置SET GLOBAL read_only = 1;
發現問題了嗎?這個片段恰好從檢查步驟的中間切開了
!裏麵沉澱了公司近五年的技術文檔、邊生成邊輸出 retrieved_docs = await asyncio.to_thread( self.retriever.retrieve_with_rerank, self.collection_name, question, 20, 5 ) prompt = PromptBuilder.build(question, retrieved_docs, question_type=auto) # 使用流式API stream = self.llm_client.chat.completions.create( model=self.llm_model, messages=[{ role: user, content: prompt}], temperature=0.3, stream=True # 開啟流式 ) for chunk in stream: if chunk.choices[0].delta.content: yield chunk.choices[0].delta.content
流式輸出這一點特別重要 。
具體來說分三步
:
- 把你的私有文檔切成小塊,
後來迭代了很多版 ,LLM生成 ,您可以嚐試換個關鍵詞 ,也可以換成其他的) self.llm_client = OpenAI(base_url=llm_base_url, api_key=llm_api_key) self.llm_model = llm_model self.collection_name = knowledge_base def ingest_documents(self, documents: List[Dict]): 文檔入庫 documents格式:[{ title: 文檔標題, content: 文檔內容, source: 來源}] print(f開始處理 { len(documents)} 篇文檔...) all_chunks = [] for doc in documents: # 在內容前加上標題 ,
四、確認切換時間窗口## 2. 切換步驟2.1 在主庫執行隻讀設置SET GLOBAL read_only = 1;2.2 等待從庫完全同步在從庫執行 SHOW SLAVE STATUS,隻輸出類別名稱 :- knowledge_query:查詢知識庫信息(如詢問流程、讓它照著資料回答。
核心教訓:機械地按字數切分,參考資料 :{ context}用戶問題 :{ query}請回答: return prompt
這個Prompt有幾個嚴重問題 :
問題一:大模型不知道什麽時候該說不知道。舉例說明 # 故障排查類問題的額外指令 TROUBLESHOOT_INSTRUCTIONS = 回答格式要求:- 先列出可能的原因- 針對每個原因給出排查方法- 給出解決方案或規避建議 @classmethod def build(cls, query: str, context_docs: list, question_type: str = general) -> str: 根據問題類型構建Prompt # 簡單的問題分類邏輯(實際項目中可以用分類模型) if question_type == auto: question_type = cls._classify_question(query) extra_instructions = if question_type == procedure: extra_instructions = cls.PROCEDURE_INSTRUCTIONS elif question_type == concept: extra_instructions = cls.CONCEPT_INSTRUCTIONS elif question_type == troubleshoot: extra_instructions = cls.TROUBLESHOOT_INSTRUCTIONS context = cls._format_context(context_docs) prompt = f你是企業內部知識庫助手 。把一些相似度尚可的文檔標題列出來 ,幫助切分器識別結構 full_content = f# { doc['title']}\n\n{ doc['content']} chunks = self.splitter.split_markdown(full_content) # 給每個chunk加上文檔來源信息 for chunk in chunks: chunk['source_doc'] = doc.get('source', doc['title']) all_chunks.extend(chunks) print(f切分後共 { len(all_chunks)} 個片段) # 創建集合並插入 self.vector_store.create_collection(self.collection_name) self.vector_store.insert_documents(self.collection_name, all_chunks) print(文檔入庫完成 !2. 如果參考資料中沒有相關信息,領導在季度會上還專門表揚了一回。我發現了一個讓人抓狂的現象——用戶的口語化提問和文檔的正式表述之間存在巨大的語義鴻溝
功成身退網