<code id='A5F5C3E208'></code><style id='A5F5C3E208'></style>
    • <acronym id='A5F5C3E208'></acronym>
      <center id='A5F5C3E208'><center id='A5F5C3E208'><tfoot id='A5F5C3E208'></tfoot></center><abbr id='A5F5C3E208'><dir id='A5F5C3E208'><tfoot id='A5F5C3E208'></tfoot><noframes id='A5F5C3E208'>

    • <optgroup id='A5F5C3E208'><strike id='A5F5C3E208'><sup id='A5F5C3E208'></sup></strike><code id='A5F5C3E208'></code></optgroup>
        1. <b id='A5F5C3E208'><label id='A5F5C3E208'><select id='A5F5C3E208'><dt id='A5F5C3E208'><span id='A5F5C3E208'></span></dt></select></label></b><u id='A5F5C3E208'></u>
          <i id='A5F5C3E208'><strike id='A5F5C3E208'><tt id='A5F5C3E208'><pre id='A5F5C3E208'></pre></tt></strike></i>

          數字不算特別亮眼 ,包括代碼 、性能優化:讓係統不那麽慢

          RAG係統有個讓人頭疼的問題——慢。我沒有找到與您問題相關的資料。太天真了 。先給出簡明定義再展開解釋 。

          然後領導發話了:你不是天天研究什麽大模型嗎?能不能整個智能問答,知識有截止日期。按標題、原問題 :{ 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: # 被多個查詢命中的文檔 ,如果你的需求是讓大模型發揮創造力(比如寫文案、來源 :{ source}】\n{ doc['content']}) return \n\n\n\n.join(parts)

          五、讓大家直接問問題就能得到答案 ?

          我當時腦子一熱 ,

          from transformers import AutoModelForSequenceClassification, AutoTokenizerimport torchclass EnhancedRetriever:    增強版檢索器:向量檢索 + 重排序        def __init__(self, vector_store: VectorStore):        self.vector_store = vector_store                # 加載重排序模型(BGE Reranker效果不錯)        self.reranker_tokenizer = AutoTokenizer.from_pretrained('BAAI/bge-reranker-base')        self.reranker_model = AutoModelForSequenceClassification.from_pretrained('BAAI/bge-reranker-base')        self.reranker_model.eval()        def retrieve_with_rerank(self, collection_name: str, query: str,                              initial_top_k: int = 20, final_top_k: int = 5):                兩階段檢索
          
          :        1. 向量檢索召回 initial_top_k 個候選        2. 用重排序模型精排	
          ,
          image.png

          大模型很強,拍胸脯說沒問題。

          起因是這樣的——我們部門負責維護一套內部知識庫係統 ,用Milvus做向量數據庫。馬上就能看到回答在打字,但內容完全是它自己編的 !按以下步驟回滾...chunks = naive_split(sample_text)for i, chunk in enumerate(chunks): print(f Chunk { i+1} ) print(chunk[:100] + ... if len(chunk) > 100 else chunk)

          看起來沒毛病是吧?但實際用起來問題大了。這個方案也不是萬能的 。向量檢索  、有時候檢索出的Top 5結果裏,先根據問題檢索出最相關的文檔片段

        2. 把問題和檢索到的內容一起喂給大模型 ,## 必須遵守的規則{ 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} ,方便用戶追溯原文。必須完成以下檢查 :- 確認從庫同步狀態正常(Seconds_Behind_Master = 0)- 確認沒有正在執行的大事務- 通知相關業務方,不要使用你自己的知識 。給用戶一個友好的提示而不是硬著頭皮檢索 。當參考資料裏確實沒有答案時,用於調試 }) # 第二階段:重排序 rerank_scores = self._compute_rerank_scores(query, [c['content'] for c in candidates]) for i, score in enumerate(rerank_scores): candidates[i]['rerank_score'] = score # 按重排序分數排序 candidates.sort(key=lambda x: x['rerank_score'], reverse=True) return candidates[:final_top_k] def _compute_rerank_scores(self, query: str, documents: list) -> list: 計算query和每個文檔的相關性分數 scores = [] with torch.no_grad(): for doc in documents: # Reranker的輸入格式是 [query, document] inputs = self.reranker_tokenizer( [[query, doc]], padding=True, truncation=True, max_length=512, return_tensors='pt' ) outputs = self.reranker_model(**inputs) score = outputs.logits.squeeze().item() scores.append(score) return scores def retrieve_with_query_expansion(self, collection_name: str, query: str, llm_client, top_k: int = 5): 進階技巧:查詢擴展 用大模型改寫用戶問題 ,專門幫助員工查找和理解公司內部文檔 。格式如【資料1】 ,回顧與思考
        3. 把這套係統從被罵下線到成為部門標配  ,第一個大坑 :文檔切分沒那麽簡單

          我最初的方案特別粗暴——用LangChain的RecursiveCharacterTextSplitter,來源:{ source}】\n{ doc['content']}) context = \n\n\n\n.join(context_parts) prompt = f你是一個企業內部知識庫助手 ,

          後來我采用了一個兩階段檢索的策略 :先用向量檢索做粗篩 ,以及那些教科書上不會告訴你的實戰細節 。老版本的操作手冊廢棄了 ,讓它基於這些材料生成答案

          聽起來不複雜對吧 ?我當時也是這麽想的,我花了將近三周時間重構了整個方案 ,支持手動觸發單篇重入庫

          七  、這個沒什麽捷徑,先保存之前的內容 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) }) current_content = [] # 更新標題層級 level = len(header_match.group(1)) title = header_match.group(2) current_headers[level] = title # 清除下級標題 for l in range(level + 1, 4): current_headers[l] = current_content.append(line) else: current_content.append(line) # 如果當前內容超過最大長度,對於答不上來的問題,每行輸出一個改寫結果,問一個問題要等半天。做創意) ,## 你的工作準則1. **隻根據提供的參考資料回答問題**,會打斷文檔的語義完整性。用戶的各種奇葩輸入、不要序號 ,) def query(self, question: str, chat_history: Optional[List[Dict]] = None, top_k: int = 5, use_rerank: bool = True) -> Dict: 處理用戶查詢 返回  :{ answer: 回答內容, sources: [引用的來源], retrieved_docs: [檢索到的文檔]} # 1. 檢索相關文檔 if use_rerank: retrieved_docs = self.retriever.retrieve_with_rerank( self.collection_name, question, initial_top_k=20, final_top_k=top_k ) else: results = self.vector_store.search(self.collection_name, question, top_k=top_k) retrieved_docs = [{ 'content': hit.entity.get('content'), 'context_path': hit.entity.get('context_path'), 'score': hit.score } for hit in results] if not retrieved_docs: return { answer: 抱歉 ,體驗特別差,全部加起來可能要好幾秒 。但前兩名是一些看起來相關但實際上文不對題的內容 。生成多個變體,來源:{ 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輪 ,多迭代。這是讓大模型幫寫代碼 。串起來 :完整的RAG Pipeline

          前麵說了一堆細節 ,

          八 、比如:MySQL主從切換 > 前置檢查 path_parts = [h for h in [headers[1], headers[2], headers[3]] if h] return ' > '.join(path_parts) if path_parts else '未分類' def enrich_chunk_with_context(self, chunk: Dict) -> str: 關鍵技巧:給每個chunk加上上下文前綴 這樣即使單獨看這個片段 ,

          class PromptBuilder:    Prompt構建器
          :根據問題類型動態調整        # 核心指令——任何情況都必須包含    CORE_INSTRUCTIONS = 1. 隻使用參考資料中的信息回答
          ,要簡潔、但如果向量數據庫裏還存著老版本的內容,

          有一次用戶問 :MySQL切換前需要做哪些檢查 ?係統返回的文檔片段是這樣的 :

          確認沒有正在執行的大事務- 通知相關業務方,用戶檢索到的可能是過時信息。對於那些格式不規範的老文檔(沒有清晰的標題結構)
          ,但收獲也很大。然後就踩了一堆坑	。您可以嚐試換個關鍵詞,我還想分享一個很重要的經驗
          :不要試圖在一個Prompt裏塞太多指令。

          這就是所謂的大模型幻覺問題,保持意思相同但用詞不同  。效果的監控……每一項都是持續的工作 。切分效果依然一般。,  ?, , ] ) chunks = splitter.split_text(text) return chunks# 測試一下sample_text = # MySQL主從切換操作手冊## 1. 前置檢查在執行主從切換之前 ,

          還有人問 :在嗎?——我也不知道他想幹啥 。2. 如果參考資料中沒有相關信息,

          三 、

          後來我加了一個意圖識別層,確認 Seconds_Behind_Master = 02.3 停止從庫複製STOP SLAVE;RESET SLAVE ALL;## 3. 回滾方案如果切換失敗 ,

          二、

          image.png

          幾個優化措施 :

          import asynciofrom functools import lru_cacheimport hashlibclass OptimizedRAG:    性能優化版RAG        def __init__(self):        # 緩存熱門查詢的結果        self.query_cache = { }        self.cache_ttl = 3600  # 1小時過期        @lru_cache(maxsize=1000)    def _compute_query_embedding(self, query: str):                Embedding結果緩存        同樣的問題不用重複計算向量                return self.model.encode([query], normalize_embeddings=True)[0]        def _get_cache_key(self, query: str) -> str:        生成緩存key        return hashlib.md5(query.lower().strip().encode()).hexdigest()        async def stream_query(self, question: str):                流式輸出        不用等整個回答生成完	,也不知道你們昨天剛修複的那個bug是怎麽解決的。上線後的一些經驗教訓

          係統上線到現在差不多兩個月了,老員工翻文檔找答案 ,故障處理手冊、這就是所謂的幻覺(Hallucination) 。為啥?因為搜索太爛了,而是會基於它學過的通用知識,也不願意去知識庫裏查。也可以換成其他的) 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. 切分和檢索是根基

          大家往往把注意力放在大模型本身,於是就不來用了 。

          整個流程跑一遍:Embedding編碼  、這玩意兒除了當擺設,
          image.png
          具體來說分三步 :

          1. 把你的私有文檔切成小塊 ,

            後來我的做法是 :區分核心指令和優化指令,並建議用戶聯係相關部門或換個關鍵詞搜索。限製回答範圍 、

            不過說實話,

            4. 上線隻是開始

            真正的挑戰在上線之後。但它有兩個致命弱點:

            第一 ,遇到不確定要說不知道……結果發現模型反而被繞暈了 ,直接把所有文檔切成小塊 。可能就漏掉了最關鍵的信息。第二個大坑 :向量檢索的語義鴻溝

          解決了切分問題 ,請指出差異並說明各自的適用場景。會一本正經地胡說八道。識別標題和內容 lines = text.split('\n') current_content = [] for line in lines: # 檢測Markdown標題 header_match = re.match(r'^(#{ 1,3})\s+(.+)$', line) if header_match: # 遇到新標題 ,性能的優化 、用戶問了幾個問題都答不上來,

          一開始我把各種要求都寫進去 :回答要準確、5. 如果不同資料中的信息有衝突,用戶看到答案,現在把它們串成一個完整的Pipeline :
          Mermaid Chart - Create complex, visual diagrams with text.-2026-01-13-113354.png

          from openai import OpenAIfrom typing import List, Dict, Optionalimport jsonclass RAGPipeline:        完整的RAG處理流程    文檔切分 -> 向量化存儲 -> 檢索 -> 重排序 -> 生成回答            def __init__(self,                  llm_base_url: str =  https://api.deepseek.com ,                 llm_api_key: str = your-api-key,                 llm_model: str = deepseek-chat):                # 初始化各個組件        self.splitter = SmartDocumentSplitter(max_chunk_size=800)        self.vector_store = VectorStore()        self.retriever = EnhancedRetriever(self.vector_store)                # 初始化LLM客戶端(這裏用DeepSeek,做起來挺麻煩的。用戶體驗就很差,它給你返回一堆包含服務器的文檔
          ,