4.6 prompts设计与query意图识别.html 89 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130
  1. <!doctype html>
  2. <html lang="zh" class="no-js">
  3. <head>
  4. <meta charset="utf-8">
  5. <meta name="viewport" content="width=device-width,initial-scale=1">
  6. <link rel="prev" href="4.5%20BERT%E5%9F%BA%E7%A1%80%E4%B8%8E%E4%BB%BB%E5%8A%A1%E5%BE%AE%E8%B0%83.html">
  7. <link rel="next" href="4.7%20%E6%A3%80%E7%B4%A2%E7%AD%96%E7%95%A5%E4%B8%8ERAG%E7%B3%BB%E7%BB%9F%E8%AE%BE%E8%AE%A1.html">
  8. <link rel="icon" href="../img/AI.jpg">
  9. <meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.21">
  10. <title>4.6 prompts设计与query意图识别 - 企业知识库RAG智能问答项目V1.0</title>
  11. <link rel="stylesheet" href="../assets/stylesheets/main.2a3383ac.min.css">
  12. <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
  13. <link rel="stylesheet" href="https://fonts.googleapis.com/css?family=Roboto:300,300i,400,400i,700,700i%7CRoboto+Mono:400,400i,700,700i&amp;display=fallback">
  14. <style>:root{--md-text-font:"Roboto";--md-code-font:"Roboto Mono"}</style>
  15. <link rel="stylesheet" href="../stylesheets/extra.css">
  16. <script>__md_scope=new URL("..",location),__md_hash=e=>[...e].reduce(((e,_)=>(e<<5)-e+_.charCodeAt(0)),0),__md_get=(e,_=localStorage,t=__md_scope)=>JSON.parse(_.getItem(t.pathname+"."+e)),__md_set=(e,_,t=localStorage,a=__md_scope)=>{try{t.setItem(a.pathname+"."+e,JSON.stringify(_))}catch(e){}}</script>
  17. </head>
  18. <body dir="ltr">
  19. <input class="md-toggle" data-md-toggle="drawer" type="checkbox" id="__drawer" autocomplete="off">
  20. <input class="md-toggle" data-md-toggle="search" type="checkbox" id="__search" autocomplete="off">
  21. <label class="md-overlay" for="__drawer"></label>
  22. <div data-md-component="skip">
  23. <a href="#_1" class="md-skip">
  24. 跳转至
  25. </a>
  26. </div>
  27. <div data-md-component="announce">
  28. </div>
  29. <header class="md-header md-header--shadow" data-md-component="header">
  30. <nav class="md-header__inner md-grid" aria-label="页眉">
  31. <a href="../index.html" title="企业知识库RAG智能问答项目V1.0" class="md-header__button md-logo" aria-label="企业知识库RAG智能问答项目V1.0" data-md-component="logo">
  32. <img src="../img/AI.jpg" alt="logo">
  33. </a>
  34. <label class="md-header__button md-icon" for="__drawer">
  35. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M3 6h18v2H3zm0 5h18v2H3zm0 5h18v2H3z"></path></svg>
  36. </label>
  37. <div class="md-header__title" data-md-component="header-title">
  38. <div class="md-header__ellipsis">
  39. <div class="md-header__topic">
  40. <span class="md-ellipsis">
  41. 企业知识库RAG智能问答项目V1.0
  42. </span>
  43. </div>
  44. <div class="md-header__topic" data-md-component="header-topic">
  45. <span class="md-ellipsis">
  46. 4.6 prompts设计与query意图识别
  47. </span>
  48. </div>
  49. </div>
  50. </div>
  51. <script>var palette=__md_get("__palette");if(palette&&palette.color){if("(prefers-color-scheme)"===palette.color.media){var media=matchMedia("(prefers-color-scheme: light)"),input=document.querySelector(media.matches?"[data-md-color-media='(prefers-color-scheme: light)']":"[data-md-color-media='(prefers-color-scheme: dark)']");palette.color.media=input.getAttribute("data-md-color-media"),palette.color.scheme=input.getAttribute("data-md-color-scheme"),palette.color.primary=input.getAttribute("data-md-color-primary"),palette.color.accent=input.getAttribute("data-md-color-accent")}for(var[key,value]of Object.entries(palette.color))document.body.setAttribute("data-md-color-"+key,value)}</script>
  52. <label class="md-header__button md-icon" for="__search">
  53. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M9.5 3A6.5 6.5 0 0 1 16 9.5c0 1.61-.59 3.09-1.56 4.23l.27.27h.79l5 5-1.5 1.5-5-5v-.79l-.27-.27A6.52 6.52 0 0 1 9.5 16 6.5 6.5 0 0 1 3 9.5 6.5 6.5 0 0 1 9.5 3m0 2C7 5 5 7 5 9.5S7 14 9.5 14 14 12 14 9.5 12 5 9.5 5"></path></svg>
  54. </label>
  55. <div class="md-search" data-md-component="search" role="dialog">
  56. <label class="md-search__overlay" for="__search"></label>
  57. <div class="md-search__inner" role="search">
  58. <form class="md-search__form" name="search">
  59. <input type="text" class="md-search__input" name="query" aria-label="搜索" placeholder="搜索" autocapitalize="off" autocorrect="off" autocomplete="off" spellcheck="false" data-md-component="search-query" required>
  60. <label class="md-search__icon md-icon" for="__search">
  61. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M9.5 3A6.5 6.5 0 0 1 16 9.5c0 1.61-.59 3.09-1.56 4.23l.27.27h.79l5 5-1.5 1.5-5-5v-.79l-.27-.27A6.52 6.52 0 0 1 9.5 16 6.5 6.5 0 0 1 3 9.5 6.5 6.5 0 0 1 9.5 3m0 2C7 5 5 7 5 9.5S7 14 9.5 14 14 12 14 9.5 12 5 9.5 5"></path></svg>
  62. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M20 11v2H8l5.5 5.5-1.42 1.42L4.16 12l7.92-7.92L13.5 5.5 8 11z"></path></svg>
  63. </label>
  64. <nav class="md-search__options" aria-label="查找">
  65. <button type="reset" class="md-search__icon md-icon" title="清空当前内容" aria-label="清空当前内容" tabindex="-1">
  66. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M19 6.41 17.59 5 12 10.59 6.41 5 5 6.41 10.59 12 5 17.59 6.41 19 12 13.41 17.59 19 19 17.59 13.41 12z"></path></svg>
  67. </button>
  68. </nav>
  69. </form>
  70. <div class="md-search__output">
  71. <div class="md-search__scrollwrap" tabindex="0" data-md-scrollfix>
  72. <div class="md-search-result" data-md-component="search-result">
  73. <div class="md-search-result__meta">
  74. 正在初始化搜索引擎
  75. </div>
  76. <ol class="md-search-result__list" role="presentation"></ol>
  77. </div>
  78. </div>
  79. </div>
  80. </div>
  81. </div>
  82. </nav>
  83. </header>
  84. <div class="md-container" data-md-component="container">
  85. <main class="md-main" data-md-component="main">
  86. <div class="md-main__inner md-grid">
  87. <div class="md-sidebar md-sidebar--primary" data-md-component="sidebar" data-md-type="navigation">
  88. <div class="md-sidebar__scrollwrap">
  89. <div class="md-sidebar__inner">
  90. <nav class="md-nav md-nav--primary" aria-label="导航栏" data-md-level="0">
  91. <label class="md-nav__title" for="__drawer">
  92. <a href="../index.html" title="企业知识库RAG智能问答项目V1.0" class="md-nav__button md-logo" aria-label="企业知识库RAG智能问答项目V1.0" data-md-component="logo">
  93. <img src="../img/AI.jpg" alt="logo">
  94. </a>
  95. 企业知识库RAG智能问答项目V1.0
  96. </label>
  97. <ul class="md-nav__list" data-md-scrollfix>
  98. <li class="md-nav__item">
  99. <a href="../index.html" class="md-nav__link">
  100. <span class="md-ellipsis">
  101. 知识库内容
  102. </span>
  103. </a>
  104. </li>
  105. <li class="md-nav__item md-nav__item--nested">
  106. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_2">
  107. <label class="md-nav__link" for="__nav_2" id="__nav_2_label" tabindex="0">
  108. <span class="md-ellipsis">
  109. ① 项目概述
  110. </span>
  111. <span class="md-nav__icon md-icon"></span>
  112. </label>
  113. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_2_label" aria-expanded="false">
  114. <label class="md-nav__title" for="__nav_2">
  115. <span class="md-nav__icon md-icon"></span>
  116. ① 项目概述
  117. </label>
  118. <ul class="md-nav__list" data-md-scrollfix>
  119. <li class="md-nav__item">
  120. <a href="../%E2%91%A0%20%E9%A1%B9%E7%9B%AE%E6%A6%82%E8%BF%B0/1.1%20%E9%A1%B9%E7%9B%AE%E6%9E%B6%E6%9E%84.html" class="md-nav__link">
  121. <span class="md-ellipsis">
  122. 1.1 项目架构
  123. </span>
  124. </a>
  125. </li>
  126. <li class="md-nav__item">
  127. <a href="../%E2%91%A0%20%E9%A1%B9%E7%9B%AE%E6%A6%82%E8%BF%B0/1.2%20%E9%A1%B9%E7%9B%AE%E7%8E%AF%E5%A2%83%28window%29.html" class="md-nav__link">
  128. <span class="md-ellipsis">
  129. 1.2 项目环境(window)
  130. </span>
  131. </a>
  132. </li>
  133. <li class="md-nav__item">
  134. <a href="../%E2%91%A0%20%E9%A1%B9%E7%9B%AE%E6%A6%82%E8%BF%B0/1.3%20%E9%A1%B9%E7%9B%AE%E7%8E%AF%E5%A2%83%28mac%29.html" class="md-nav__link">
  135. <span class="md-ellipsis">
  136. 1.3 项目环境(mac)
  137. </span>
  138. </a>
  139. </li>
  140. </ul>
  141. </nav>
  142. </li>
  143. <li class="md-nav__item md-nav__item--nested">
  144. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_3">
  145. <label class="md-nav__link" for="__nav_3" id="__nav_3_label" tabindex="0">
  146. <span class="md-ellipsis">
  147. ② 项目工具
  148. </span>
  149. <span class="md-nav__icon md-icon"></span>
  150. </label>
  151. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_3_label" aria-expanded="false">
  152. <label class="md-nav__title" for="__nav_3">
  153. <span class="md-nav__icon md-icon"></span>
  154. ② 项目工具
  155. </label>
  156. <ul class="md-nav__list" data-md-scrollfix>
  157. <li class="md-nav__item">
  158. <a href="../%E2%91%A1%20%E9%A1%B9%E7%9B%AE%E5%B7%A5%E5%85%B7/2.1%20Ollama%E5%A4%A7%E6%A8%A1%E5%9E%8B%E9%AB%98%E6%95%88%E7%AE%A1%E7%90%86%E5%B7%A5%E5%85%B7.html" class="md-nav__link">
  159. <span class="md-ellipsis">
  160. 2.1 Ollama大模型高效管理工具
  161. </span>
  162. </a>
  163. </li>
  164. <li class="md-nav__item">
  165. <a href="../%E2%91%A1%20%E9%A1%B9%E7%9B%AE%E5%B7%A5%E5%85%B7/2.2%20LangChain%E5%9F%BA%E7%A1%80%E7%9F%A5%E8%AF%86.md.html" class="md-nav__link">
  166. <span class="md-ellipsis">
  167. 2.2 LangChain基础知识.md
  168. </span>
  169. </a>
  170. </li>
  171. <li class="md-nav__item">
  172. <a href="../%E2%91%A1%20%E9%A1%B9%E7%9B%AE%E5%B7%A5%E5%85%B7/2.3%20%E5%90%91%E9%87%8F%E7%B4%A2%E5%BC%95%EF%BC%9AHNSW%E3%80%81IVF%E4%B8%8EIVF_PQ.html" class="md-nav__link">
  173. <span class="md-ellipsis">
  174. 2.3 向量索引:HNSW、IVF与IVF_PQ
  175. </span>
  176. </a>
  177. </li>
  178. <li class="md-nav__item">
  179. <a href="../%E2%91%A1%20%E9%A1%B9%E7%9B%AE%E5%B7%A5%E5%85%B7/2.4%20Milvus%E5%90%91%E9%87%8F%E6%95%B0%E6%8D%AE%E5%BA%93.html" class="md-nav__link">
  180. <span class="md-ellipsis">
  181. 2.4 Milvus向量数据库
  182. </span>
  183. </a>
  184. </li>
  185. </ul>
  186. </nav>
  187. </li>
  188. <li class="md-nav__item md-nav__item--nested">
  189. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_4">
  190. <label class="md-nav__link" for="__nav_4" id="__nav_4_label" tabindex="0">
  191. <span class="md-ellipsis">
  192. ③ 基于Mysql库的问答系统
  193. </span>
  194. <span class="md-nav__icon md-icon"></span>
  195. </label>
  196. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_4_label" aria-expanded="false">
  197. <label class="md-nav__title" for="__nav_4">
  198. <span class="md-nav__icon md-icon"></span>
  199. ③ 基于Mysql库的问答系统
  200. </label>
  201. <ul class="md-nav__list" data-md-scrollfix>
  202. <li class="md-nav__item">
  203. <a href="../%E2%91%A2%20%E5%9F%BA%E4%BA%8EMysql%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/3.1%20Python%E6%97%A5%E5%BF%97%E4%BB%8B%E7%BB%8D%E4%B8%8E%E5%BA%94%E7%94%A8.html" class="md-nav__link">
  204. <span class="md-ellipsis">
  205. 3.1 Python日志介绍与应用
  206. </span>
  207. </a>
  208. </li>
  209. <li class="md-nav__item">
  210. <a href="../%E2%91%A2%20%E5%9F%BA%E4%BA%8EMysql%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/3.2%20Redis%E6%95%B0%E6%8D%AE%E5%BA%93%E7%AE%80%E4%BB%8B%E4%B8%8E%E4%BD%BF%E7%94%A8.html" class="md-nav__link">
  211. <span class="md-ellipsis">
  212. 3.2 Redis数据库简介与使用
  213. </span>
  214. </a>
  215. </li>
  216. <li class="md-nav__item">
  217. <a href="../%E2%91%A2%20%E5%9F%BA%E4%BA%8EMysql%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/3.3%20%E5%9F%BA%E4%BA%8EMysql%E6%95%B0%E6%8D%AE%E5%BA%93%E5%AE%9E%E7%8E%B0%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F.html" class="md-nav__link">
  218. <span class="md-ellipsis">
  219. 3.3 基于Mysql数据库实现问答系统
  220. </span>
  221. </a>
  222. </li>
  223. <li class="md-nav__item">
  224. <a href="../%E2%91%A2%20%E5%9F%BA%E4%BA%8EMysql%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/3.4%20BM25%E7%AE%97%E6%B3%95%E7%AE%80%E4%BB%8B%E4%B8%8E%E5%BA%94%E7%94%A8.html" class="md-nav__link">
  225. <span class="md-ellipsis">
  226. 3.4 BM25算法简介与应用
  227. </span>
  228. </a>
  229. </li>
  230. </ul>
  231. </nav>
  232. </li>
  233. <li class="md-nav__item md-nav__item--active md-nav__item--nested">
  234. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_5" checked>
  235. <label class="md-nav__link" for="__nav_5" id="__nav_5_label" tabindex="0">
  236. <span class="md-ellipsis">
  237. ④ 基于Milvus库的问答系统
  238. </span>
  239. <span class="md-nav__icon md-icon"></span>
  240. </label>
  241. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_5_label" aria-expanded="true">
  242. <label class="md-nav__title" for="__nav_5">
  243. <span class="md-nav__icon md-icon"></span>
  244. ④ 基于Milvus库的问答系统
  245. </label>
  246. <ul class="md-nav__list" data-md-scrollfix>
  247. <li class="md-nav__item">
  248. <a href="4.1%20%E6%95%B4%E4%BD%93%E6%9E%B6%E6%9E%84%E4%B8%8E%E5%B7%A5%E7%A8%8B%E6%B5%81%E7%A8%8B.html" class="md-nav__link">
  249. <span class="md-ellipsis">
  250. 4.1 整体架构与工程流程
  251. </span>
  252. </a>
  253. </li>
  254. <li class="md-nav__item">
  255. <a href="4.2%20%E5%9F%BA%E7%A1%80%E6%A8%A1%E5%9D%97.html" class="md-nav__link">
  256. <span class="md-ellipsis">
  257. 4.2 基础模块
  258. </span>
  259. </a>
  260. </li>
  261. <li class="md-nav__item">
  262. <a href="4.3%20%E6%96%87%E6%A1%A3%E5%A4%84%E7%90%86%E6%A8%A1%E5%9D%97.html" class="md-nav__link">
  263. <span class="md-ellipsis">
  264. 4.3 文档处理模块
  265. </span>
  266. </a>
  267. </li>
  268. <li class="md-nav__item">
  269. <a href="4.4%20%E5%90%91%E9%87%8F%E5%AD%98%E5%82%A8.html" class="md-nav__link">
  270. <span class="md-ellipsis">
  271. 4.4 向量存储
  272. </span>
  273. </a>
  274. </li>
  275. <li class="md-nav__item"><a href="4.5%20BERT%E5%9F%BA%E7%A1%80%E4%B8%8E%E4%BB%BB%E5%8A%A1%E5%BE%AE%E8%B0%83.html" class="md-nav__link"><span class="md-ellipsis">4.5 BERT基础与任务微调</span></a></li><li class="md-nav__item md-nav__item--active">
  276. <input class="md-nav__toggle md-toggle" type="checkbox" id="__toc">
  277. <label class="md-nav__link md-nav__link--active" for="__toc">
  278. <span class="md-ellipsis">
  279. 4.6 prompts设计与query意图识别
  280. </span>
  281. <span class="md-nav__icon md-icon"></span>
  282. </label>
  283. <a href="4.6%20prompts%E8%AE%BE%E8%AE%A1%E4%B8%8Equery%E6%84%8F%E5%9B%BE%E8%AF%86%E5%88%AB.html" class="md-nav__link md-nav__link--active">
  284. <span class="md-ellipsis">
  285. 4.6 prompts设计与query意图识别
  286. </span>
  287. </a>
  288. <nav class="md-nav md-nav--secondary" aria-label="目录">
  289. <label class="md-nav__title" for="__toc">
  290. <span class="md-nav__icon md-icon"></span>
  291. 目录
  292. </label>
  293. <ul class="md-nav__list" data-md-component="toc" data-md-scrollfix>
  294. <li class="md-nav__item">
  295. <a href="#_1" class="md-nav__link">
  296. <span class="md-ellipsis">
  297. 学习目标
  298. </span>
  299. </a>
  300. </li>
  301. <li class="md-nav__item">
  302. <a href="#prompt" class="md-nav__link">
  303. <span class="md-ellipsis">
  304. 一、Prompt管理
  305. </span>
  306. </a>
  307. <nav class="md-nav" aria-label="一、Prompt管理">
  308. <ul class="md-nav__list">
  309. <li class="md-nav__item">
  310. <a href="#1" class="md-nav__link">
  311. <span class="md-ellipsis">
  312. 1 功能概述
  313. </span>
  314. </a>
  315. </li>
  316. <li class="md-nav__item">
  317. <a href="#2" class="md-nav__link">
  318. <span class="md-ellipsis">
  319. 2 代码实现
  320. </span>
  321. </a>
  322. </li>
  323. <li class="md-nav__item">
  324. <a href="#3" class="md-nav__link">
  325. <span class="md-ellipsis">
  326. 3 实现细节
  327. </span>
  328. </a>
  329. </li>
  330. </ul>
  331. </nav>
  332. </li>
  333. <li class="md-nav__item">
  334. <a href="#_2" class="md-nav__link">
  335. <span class="md-ellipsis">
  336. 二、查询分类
  337. </span>
  338. </a>
  339. <nav class="md-nav" aria-label="二、查询分类">
  340. <ul class="md-nav__list">
  341. <li class="md-nav__item">
  342. <a href="#1_1" class="md-nav__link">
  343. <span class="md-ellipsis">
  344. 1 功能概述
  345. </span>
  346. </a>
  347. </li>
  348. <li class="md-nav__item">
  349. <a href="#2_1" class="md-nav__link">
  350. <span class="md-ellipsis">
  351. 2 代码实现
  352. </span>
  353. </a>
  354. </li>
  355. <li class="md-nav__item">
  356. <a href="#3_1" class="md-nav__link">
  357. <span class="md-ellipsis">
  358. 3 实现细节
  359. </span>
  360. </a>
  361. </li>
  362. <li class="md-nav__item">
  363. <a href="#4" class="md-nav__link">
  364. <span class="md-ellipsis">
  365. 4 执行示例
  366. </span>
  367. </a>
  368. <nav class="md-nav" aria-label="4 执行示例">
  369. <ul class="md-nav__list">
  370. <li class="md-nav__item">
  371. <a href="#41" class="md-nav__link">
  372. <span class="md-ellipsis">
  373. 4.1 运行脚本
  374. </span>
  375. </a>
  376. </li>
  377. <li class="md-nav__item">
  378. <a href="#42" class="md-nav__link">
  379. <span class="md-ellipsis">
  380. 4.2 代码示例
  381. </span>
  382. </a>
  383. </li>
  384. </ul>
  385. </nav>
  386. </li>
  387. </ul>
  388. </nav>
  389. </li>
  390. <li class="md-nav__item">
  391. <a href="#_3" class="md-nav__link">
  392. <span class="md-ellipsis">
  393. 章节小结
  394. </span>
  395. </a>
  396. </li>
  397. </ul>
  398. </nav>
  399. </li>
  400. <li class="md-nav__item">
  401. <a href="4.7%20%E6%A3%80%E7%B4%A2%E7%AD%96%E7%95%A5%E4%B8%8ERAG%E7%B3%BB%E7%BB%9F%E8%AE%BE%E8%AE%A1.html" class="md-nav__link">
  402. <span class="md-ellipsis">
  403. 4.7 检索策略与RAG系统设计
  404. </span>
  405. </a>
  406. </li>
  407. <li class="md-nav__item">
  408. <a href="4.8%20RAG%E7%B3%BB%E7%BB%9F%E8%BF%90%E8%A1%8C.html" class="md-nav__link">
  409. <span class="md-ellipsis">
  410. 4.8 RAG系统运行
  411. </span>
  412. </a>
  413. </li>
  414. <li class="md-nav__item">
  415. <a href="4.9%20%E6%96%87%E6%A1%A3%E8%A7%A3%E6%9E%90%E5%B7%A5%E5%85%B7%28%E6%89%A9%E5%B1%95%E8%B5%84%E6%96%99%29.html" class="md-nav__link">
  416. <span class="md-ellipsis">
  417. 4.9 文档解析工具(扩展资料)
  418. </span>
  419. </a>
  420. </li>
  421. <li class="md-nav__item">
  422. <a href="4.10%20RAG%E4%B8%AD%E7%9A%84Query%E6%94%B9%E5%86%99%28%E6%89%A9%E5%B1%95%E8%B5%84%E6%96%99%29.html" class="md-nav__link">
  423. <span class="md-ellipsis">
  424. 4.10 RAG中的Query改写(扩展资料)
  425. </span>
  426. </a>
  427. </li>
  428. <li class="md-nav__item">
  429. <a href="%E6%95%B0%E6%8D%AE%E9%9B%86%E7%94%9F%E6%88%90%E4%B8%8E%E4%BC%98%E5%8C%96%28%E6%89%A9%E5%B1%95%E8%B5%84%E6%96%99%29.html" class="md-nav__link">
  430. <span class="md-ellipsis">
  431. 数据集生成与优化(扩展资料)
  432. </span>
  433. </a>
  434. </li>
  435. </ul>
  436. </nav>
  437. </li>
  438. <li class="md-nav__item md-nav__item--nested">
  439. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_6">
  440. <label class="md-nav__link" for="__nav_6" id="__nav_6_label" tabindex="0">
  441. <span class="md-ellipsis">
  442. ⑤ RAG系统评估
  443. </span>
  444. <span class="md-nav__icon md-icon"></span>
  445. </label>
  446. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_6_label" aria-expanded="false">
  447. <label class="md-nav__title" for="__nav_6">
  448. <span class="md-nav__icon md-icon"></span>
  449. ⑤ RAG系统评估
  450. </label>
  451. <ul class="md-nav__list" data-md-scrollfix>
  452. <li class="md-nav__item">
  453. <a href="../%E2%91%A4%20RAG%E7%B3%BB%E7%BB%9F%E8%AF%84%E4%BC%B0/5.1%20RAG%E7%B3%BB%E7%BB%9F%E8%AF%84%E4%BC%B0%E5%B7%A5%E5%85%B7.html" class="md-nav__link">
  454. <span class="md-ellipsis">
  455. 5.1 RAG系统评估工具
  456. </span>
  457. </a>
  458. </li>
  459. <li class="md-nav__item">
  460. <a href="../%E2%91%A4%20RAG%E7%B3%BB%E7%BB%9F%E8%AF%84%E4%BC%B0/5.2%20%E5%AE%9E%E7%8E%B0EnterpriseKB-RAG%E7%B3%BB%E7%BB%9F%E7%9A%84%E8%AF%84%E4%BC%B0.html" class="md-nav__link">
  461. <span class="md-ellipsis">
  462. 5.2 实现企业知识库RAG系统的评估
  463. </span>
  464. </a>
  465. </li>
  466. <li class="md-nav__item">
  467. <a href="../%E2%91%A4%20RAG%E7%B3%BB%E7%BB%9F%E8%AF%84%E4%BC%B0/5.3%20RAG%E8%AF%84%E4%BC%B0%E6%95%B0%E6%8D%AE%E9%9B%86%E7%9A%84%E6%9E%84%E9%80%A0%28%E6%89%A9%E5%B1%95%29.html" class="md-nav__link">
  468. <span class="md-ellipsis">
  469. 5.3 RAG评估数据集的构造(扩展)
  470. </span>
  471. </a>
  472. </li>
  473. </ul>
  474. </nav>
  475. </li>
  476. <li class="md-nav__item md-nav__item--nested">
  477. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_7">
  478. <label class="md-nav__link" for="__nav_7" id="__nav_7_label" tabindex="0">
  479. <span class="md-ellipsis">
  480. ⑥ 融合Mysql的RAG系统
  481. </span>
  482. <span class="md-nav__icon md-icon"></span>
  483. </label>
  484. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_7_label" aria-expanded="false">
  485. <label class="md-nav__title" for="__nav_7">
  486. <span class="md-nav__icon md-icon"></span>
  487. ⑥ 融合Mysql的RAG系统
  488. </label>
  489. <ul class="md-nav__list" data-md-scrollfix>
  490. <li class="md-nav__item">
  491. <a href="../%E2%91%A5%20%E8%9E%8D%E5%90%88Mysql%E7%9A%84RAG%E7%B3%BB%E7%BB%9F/6.1%20%E8%9E%8D%E5%90%88FQA%E5%92%8C%E7%9F%A5%E8%AF%86%E5%BA%93%E6%9F%A5%E8%AF%A2.html" class="md-nav__link">
  492. <span class="md-ellipsis">
  493. 6.1 融合FQA和知识库查询
  494. </span>
  495. </a>
  496. </li>
  497. <li class="md-nav__item">
  498. <a href="../%E2%91%A5%20%E8%9E%8D%E5%90%88Mysql%E7%9A%84RAG%E7%B3%BB%E7%BB%9F/6.2%20%E8%9E%8D%E5%90%88FAQ%20%E5%92%8C%E7%9F%A5%E8%AF%86%E5%BA%93%E6%9F%A5%E8%AF%A2%EF%BC%88%E4%BC%98%E5%8C%96%E7%89%88%EF%BC%89.html" class="md-nav__link">
  499. <span class="md-ellipsis">
  500. 6.2 融合FAQ 和知识库查询(优化版)
  501. </span>
  502. </a>
  503. </li>
  504. <li class="md-nav__item">
  505. <a href="../%E2%91%A5%20%E8%9E%8D%E5%90%88Mysql%E7%9A%84RAG%E7%B3%BB%E7%BB%9F/6.3%20%E5%9F%BA%E4%BA%8E%20FastAPI%20%E7%9A%84%20FAQ%20%E4%B8%8E%E7%9F%A5%E8%AF%86%E5%BA%93%E6%9F%A5%E8%AF%A2%E6%8E%A5%E5%8F%A3.html" class="md-nav__link">
  506. <span class="md-ellipsis">
  507. 6.3 基于 FastAPI 的 FAQ 与知识库查询接口
  508. </span>
  509. </a>
  510. </li>
  511. <li class="md-nav__item">
  512. <a href="../%E2%91%A5%20%E8%9E%8D%E5%90%88Mysql%E7%9A%84RAG%E7%B3%BB%E7%BB%9F/6.4%20FastAPI%E5%92%8CFlask%E7%9A%84%E5%8C%BA%E5%88%AB%E5%92%8C%E8%81%94%E7%B3%BB%28%E6%89%A9%E5%B1%95%E8%B5%84%E6%96%99%29.html" class="md-nav__link">
  513. <span class="md-ellipsis">
  514. 6.4 FastAPI和Flask的区别和联系(扩展资料)
  515. </span>
  516. </a>
  517. </li>
  518. </ul>
  519. </nav>
  520. </li>
  521. <li class="md-nav__item md-nav__item--nested">
  522. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_8">
  523. <label class="md-nav__link" for="__nav_8" id="__nav_8_label" tabindex="0">
  524. <span class="md-ellipsis">
  525. ⑦ 企业级生产环境部署
  526. </span>
  527. <span class="md-nav__icon md-icon"></span>
  528. </label>
  529. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_8_label" aria-expanded="false">
  530. <label class="md-nav__title" for="__nav_8">
  531. <span class="md-nav__icon md-icon"></span>
  532. ⑦ 企业级生产环境部署
  533. </label>
  534. <ul class="md-nav__list" data-md-scrollfix>
  535. <li class="md-nav__item">
  536. <a href="../%E2%91%A6%20%E4%BC%81%E4%B8%9A%E7%BA%A7%E7%94%9F%E4%BA%A7%E7%8E%AF%E5%A2%83%E9%83%A8%E7%BD%B2/7.1%20Docker%E7%9A%84%E5%8E%9F%E7%90%86%E5%92%8C%E5%9F%BA%E6%9C%AC%E4%BD%BF%E7%94%A8.html" class="md-nav__link">
  537. <span class="md-ellipsis">
  538. 7.1 Docker的原理和基本使用
  539. </span>
  540. </a>
  541. </li>
  542. <li class="md-nav__item">
  543. <a href="../%E2%91%A6%20%E4%BC%81%E4%B8%9A%E7%BA%A7%E7%94%9F%E4%BA%A7%E7%8E%AF%E5%A2%83%E9%83%A8%E7%BD%B2/7.2%20%E6%9C%8D%E5%8A%A1%E9%83%A8%E7%BD%B2.html" class="md-nav__link">
  544. <span class="md-ellipsis">
  545. 7.2 服务部署
  546. </span>
  547. </a>
  548. </li>
  549. </ul>
  550. </nav>
  551. </li>
  552. </ul>
  553. </nav>
  554. </div>
  555. </div>
  556. </div>
  557. <div class="md-sidebar md-sidebar--secondary" data-md-component="sidebar" data-md-type="toc">
  558. <div class="md-sidebar__scrollwrap">
  559. <div class="md-sidebar__inner">
  560. <nav class="md-nav md-nav--secondary" aria-label="目录">
  561. <label class="md-nav__title" for="__toc">
  562. <span class="md-nav__icon md-icon"></span>
  563. 目录
  564. </label>
  565. <ul class="md-nav__list" data-md-component="toc" data-md-scrollfix>
  566. <li class="md-nav__item">
  567. <a href="#_1" class="md-nav__link">
  568. <span class="md-ellipsis">
  569. 学习目标
  570. </span>
  571. </a>
  572. </li>
  573. <li class="md-nav__item">
  574. <a href="#prompt" class="md-nav__link">
  575. <span class="md-ellipsis">
  576. 一、Prompt管理
  577. </span>
  578. </a>
  579. <nav class="md-nav" aria-label="一、Prompt管理">
  580. <ul class="md-nav__list">
  581. <li class="md-nav__item">
  582. <a href="#1" class="md-nav__link">
  583. <span class="md-ellipsis">
  584. 1 功能概述
  585. </span>
  586. </a>
  587. </li>
  588. <li class="md-nav__item">
  589. <a href="#2" class="md-nav__link">
  590. <span class="md-ellipsis">
  591. 2 代码实现
  592. </span>
  593. </a>
  594. </li>
  595. <li class="md-nav__item">
  596. <a href="#3" class="md-nav__link">
  597. <span class="md-ellipsis">
  598. 3 实现细节
  599. </span>
  600. </a>
  601. </li>
  602. </ul>
  603. </nav>
  604. </li>
  605. <li class="md-nav__item">
  606. <a href="#_2" class="md-nav__link">
  607. <span class="md-ellipsis">
  608. 二、查询分类
  609. </span>
  610. </a>
  611. <nav class="md-nav" aria-label="二、查询分类">
  612. <ul class="md-nav__list">
  613. <li class="md-nav__item">
  614. <a href="#1_1" class="md-nav__link">
  615. <span class="md-ellipsis">
  616. 1 功能概述
  617. </span>
  618. </a>
  619. </li>
  620. <li class="md-nav__item">
  621. <a href="#2_1" class="md-nav__link">
  622. <span class="md-ellipsis">
  623. 2 代码实现
  624. </span>
  625. </a>
  626. </li>
  627. <li class="md-nav__item">
  628. <a href="#3_1" class="md-nav__link">
  629. <span class="md-ellipsis">
  630. 3 实现细节
  631. </span>
  632. </a>
  633. </li>
  634. <li class="md-nav__item">
  635. <a href="#4" class="md-nav__link">
  636. <span class="md-ellipsis">
  637. 4 执行示例
  638. </span>
  639. </a>
  640. <nav class="md-nav" aria-label="4 执行示例">
  641. <ul class="md-nav__list">
  642. <li class="md-nav__item">
  643. <a href="#41" class="md-nav__link">
  644. <span class="md-ellipsis">
  645. 4.1 运行脚本
  646. </span>
  647. </a>
  648. </li>
  649. <li class="md-nav__item">
  650. <a href="#42" class="md-nav__link">
  651. <span class="md-ellipsis">
  652. 4.2 代码示例
  653. </span>
  654. </a>
  655. </li>
  656. </ul>
  657. </nav>
  658. </li>
  659. </ul>
  660. </nav>
  661. </li>
  662. <li class="md-nav__item">
  663. <a href="#_3" class="md-nav__link">
  664. <span class="md-ellipsis">
  665. 章节小结
  666. </span>
  667. </a>
  668. </li>
  669. </ul>
  670. </nav>
  671. </div>
  672. </div>
  673. </div>
  674. <div class="md-content" data-md-component="content">
  675. <article class="md-content__inner md-typeset">
  676. <h1>4.6 prompts设计与query意图识别</h1>
  677. <div class="admonition note" data-bert-bridge="true"><p class="admonition-title">学习衔接</p><p>本节进入 Prompt 管理与查询分类的实现。前一节 4.5 已讲清分词器、标签、损失、微调与推理;阅读查询分类代码时,可对照其中“单句分类”的项目二分类示例。 <a href="4.5%20BERT%E5%9F%BA%E7%A1%80%E4%B8%8E%E4%BB%BB%E5%8A%A1%E5%BE%AE%E8%B0%83.html#intent">阅读 4.5 对应内容</a></p></div><h2 id="_1">学习目标<a class="headerlink" href="#_1" title="Permanent link">¶</a></h2>
  678. <ul>
  679. <li>1.掌握如何设计和使用Prompt模板来引导大语言模型生成高质量输出。</li>
  680. <li>2.学会查询分类的基本原理,了解如何通过分类优化输入处理流程。</li>
  681. </ul>
  682. <hr>
  683. <p><code>prompts.py</code>和<code>query_classifier.py</code>是企业知识库RAG系统中<code>core</code>模块的重要组成部分,分别负责Prompt模板管理和查询分类。这两个模块通过优化用户输入的处理,增强了系统的灵活性和智能性,为RAG系统的检索和生成阶段奠定了基础。<code>prompts.py</code>定义了多种Prompt模板,用于引导大语言模型生成特定输出,而<code>query_classifier.py</code>通过分类用户查询,决定是否直接使用模型回答或触发检索流程。</p>
  684. <h2 id="prompt">一、Prompt管理<a class="headerlink" href="#prompt" title="Permanent link">¶</a></h2>
  685. <h3 id="1">1 功能概述<a class="headerlink" href="#1" title="Permanent link">¶</a></h3>
  686. <p><code>prompts.py</code>定义了<code>RAGPrompts</code>类,负责管理系统中使用的所有Prompt模板。这些模板用于指导大语言模型完成不同任务,例如生成最终答案、假设答案、子查询或简化问题。通过集中管理Prompt,系统能够确保输入的一致性和输出质量。</p>
  687. <h3 id="2">2 代码实现<a class="headerlink" href="#2" title="Permanent link">¶</a></h3>
  688. <div class="highlight"><pre><span></span><code><span class="c1"># core/prompts.py</span>
  689. <span class="c1"># 导入 PromptTemplate 类,用于创建 Prompt 模板</span>
  690. <span class="kn">from</span><span class="w"> </span><span class="nn">langchain_core.prompts</span><span class="w"> </span><span class="kn">import</span> <span class="n">PromptTemplate</span>
  691. <span class="c1"># 定义 RAGPrompts 类,用于管理所有 Prompt 模板</span>
  692. <span class="k">class</span><span class="w"> </span><span class="nc">RAGPrompts</span><span class="p">:</span>
  693. <span class="c1"># 定义 RAG 提示模板</span>
  694. <span class="nd">@staticmethod</span>
  695. <span class="k">def</span><span class="w"> </span><span class="nf">rag_prompt</span><span class="p">():</span>
  696. <span class="c1"># 创建并返回 PromptTemplate 对象</span>
  697. <span class="k">return</span> <span class="n">PromptTemplate</span><span class="p">(</span>
  698. <span class="n">template</span><span class="o">=</span><span class="s2">""" </span>
  699. <span class="s2"> 你是一个智能助手,帮助用户回答问题。 </span>
  700. <span class="s2"> 如果提供了上下文,请基于上下文回答;如果没有上下文,请直接根据你的知识回答。 </span>
  701. <span class="s2"> 如果答案来源于检索到的文档,请在回答中说明。</span>
  702. <span class="s2"> 上下文: </span><span class="si">{context}</span><span class="s2"> </span>
  703. <span class="s2"> 问题: </span><span class="si">{question}</span><span class="s2"> </span>
  704. <span class="s2"> 如果无法回答,请回复:“信息不足,无法回答,请联系人工客服,电话:</span><span class="si">{phone}</span><span class="s2">。” </span>
  705. <span class="s2"> 回答: </span>
  706. <span class="s2"> """</span><span class="p">,</span>
  707. <span class="c1"># 定义输入变量</span>
  708. <span class="n">input_variables</span><span class="o">=</span><span class="p">[</span><span class="s2">"context"</span><span class="p">,</span> <span class="s2">"question"</span><span class="p">,</span> <span class="s2">"phone"</span><span class="p">],</span>
  709. <span class="p">)</span>
  710. <span class="c1"># @staticmethod</span>
  711. <span class="c1"># def rag_prompt():</span>
  712. <span class="c1"># return PromptTemplate(</span>
  713. <span class="c1"># template="""</span>
  714. <span class="c1"># 你是一个智能助手,负责帮助用户回答问题。请按照以下步骤处理:</span>
  715. <span class="c1"># </span>
  716. <span class="c1"># 1. **分析问题和上下文**:</span>
  717. <span class="c1"># - 基于提供的上下文(如果有)和你的知识回答问题。</span>
  718. <span class="c1"># - 如果答案来源于检索到的文档,请在回答中明确说明,例如:“根据提供的文档,……”。</span>
  719. <span class="c1"># </span>
  720. <span class="c1"># 2. **评估对话历史**:</span>
  721. <span class="c1"># - 检查对话历史是否与当前问题相关(例如,是否涉及相同的话题、实体或问题背景)。</span>
  722. <span class="c1"># - 如果对话历史与问题相关,请结合历史信息生成更准确的回答。</span>
  723. <span class="c1"># - 如果对话历史无关(例如,仅包含问候或不相关的内容),忽略历史,仅基于上下文和问题回答。</span>
  724. <span class="c1"># </span>
  725. <span class="c1"># 3. **生成回答**:</span>
  726. <span class="c1"># - 提供清晰、准确的回答,避免无关信息。</span>
  727. <span class="c1"># - 如果上下文和历史消息均不足以回答问题,请回复:“信息不足,无法回答,请联系人工客服,电话:{phone}。”</span>
  728. <span class="c1"># </span>
  729. <span class="c1"># **上下文**: {context}</span>
  730. <span class="c1"># **对话历史**:</span>
  731. <span class="c1"># {history}</span>
  732. <span class="c1"># **问题**: {question}</span>
  733. <span class="c1"># </span>
  734. <span class="c1"># **回答**:</span>
  735. <span class="c1"># """,</span>
  736. <span class="c1"># input_variables=["context", "history", "question", "phone"],</span>
  737. <span class="c1"># )</span>
  738. <span class="c1"># 定义假设问题生成的 Prompt 模板</span>
  739. <span class="nd">@staticmethod</span>
  740. <span class="k">def</span><span class="w"> </span><span class="nf">hyde_prompt</span><span class="p">():</span>
  741. <span class="c1"># 创建并返回 PromptTemplate 对象</span>
  742. <span class="k">return</span> <span class="n">PromptTemplate</span><span class="p">(</span>
  743. <span class="n">template</span><span class="o">=</span><span class="s2">""" </span>
  744. <span class="s2"> 假设你是用户,想了解以下问题,请生成一个简短的假设答案: </span>
  745. <span class="s2"> 问题: </span><span class="si">{query}</span><span class="s2"> </span>
  746. <span class="s2"> 假设答案: </span>
  747. <span class="s2"> """</span><span class="p">,</span>
  748. <span class="c1"># 定义输入变量</span>
  749. <span class="n">input_variables</span><span class="o">=</span><span class="p">[</span><span class="s2">"query"</span><span class="p">],</span>
  750. <span class="p">)</span>
  751. <span class="c1"># 定义子查询生成的 Prompt 模板</span>
  752. <span class="nd">@staticmethod</span>
  753. <span class="k">def</span><span class="w"> </span><span class="nf">subquery_prompt</span><span class="p">():</span>
  754. <span class="c1"># 创建并返回 PromptTemplate 对象</span>
  755. <span class="k">return</span> <span class="n">PromptTemplate</span><span class="p">(</span>
  756. <span class="n">template</span><span class="o">=</span><span class="s2">""" </span>
  757. <span class="s2"> 将以下复杂查询分解为多个简单子查询,每行一个子查询: </span>
  758. <span class="s2"> 查询: </span><span class="si">{query}</span><span class="s2"> </span>
  759. <span class="s2"> 子查询: </span>
  760. <span class="s2"> """</span><span class="p">,</span>
  761. <span class="c1"># 定义输入变量</span>
  762. <span class="n">input_variables</span><span class="o">=</span><span class="p">[</span><span class="s2">"query"</span><span class="p">],</span>
  763. <span class="p">)</span>
  764. <span class="c1"># 定义回溯问题生成的 Prompt 模板</span>
  765. <span class="nd">@staticmethod</span>
  766. <span class="k">def</span><span class="w"> </span><span class="nf">backtracking_prompt</span><span class="p">():</span>
  767. <span class="c1"># 创建并返回 PromptTemplate 对象</span>
  768. <span class="k">return</span> <span class="n">PromptTemplate</span><span class="p">(</span>
  769. <span class="n">template</span><span class="o">=</span><span class="s2">""" </span>
  770. <span class="s2"> 将以下复杂查询简化为一个更简单的问题: </span>
  771. <span class="s2"> 查询: </span><span class="si">{query}</span><span class="s2"> </span>
  772. <span class="s2"> 简化问题: </span>
  773. <span class="s2"> """</span><span class="p">,</span>
  774. <span class="c1"># 定义输入变量</span>
  775. <span class="n">input_variables</span><span class="o">=</span><span class="p">[</span><span class="s2">"query"</span><span class="p">],</span>
  776. <span class="p">)</span>
  777. </code></pre></div>
  778. <h3 id="3">3 实现细节<a class="headerlink" href="#3" title="Permanent link">¶</a></h3>
  779. <ol>
  780. <li><strong><code>rag_prompt</code></strong>:</li>
  781. <li><strong>作用</strong>:核心回答模板,结合检索到的上下文生成最终答案。</li>
  782. <li><strong>输入变量</strong>:<code>context</code>(检索文档内容)、<code>question</code>(用户查询)、<code>phone</code>(客服电话)。</li>
  783. <li><strong>设计逻辑</strong>:支持有无上下文的回答,并提供兜底回复,确保用户体验。</li>
  784. <li><strong><code>hyde_prompt</code></strong>:</li>
  785. <li><strong>作用</strong>:生成假设答案,用于HyDE(Hypothetical Document Embeddings)策略,优化抽象查询的检索。</li>
  786. <li><strong>输入变量</strong>:<code>query</code>(用户查询)。</li>
  787. <li><strong>设计逻辑</strong>:通过生成假设答案,间接增强查询与文档的语义匹配。</li>
  788. <li><strong><code>subquery_prompt</code></strong>:</li>
  789. <li><strong>作用</strong>:将复杂查询分解为多个子查询,适合涉及多方面的查询。</li>
  790. <li><strong>输入变量</strong>:<code>query</code>(用户查询)。</li>
  791. <li><strong>设计逻辑</strong>:分解复杂问题以提高检索覆盖率。</li>
  792. <li><strong><code>backtracking_prompt</code></strong>:</li>
  793. <li><strong>作用</strong>:将复杂查询简化为更基础的问题,便于检索。</li>
  794. <li><strong>输入变量</strong>:<code>query</code>(用户查询)。</li>
  795. <li><strong>设计逻辑</strong>:通过简化查询降低检索难度。</li>
  796. </ol>
  797. <hr>
  798. <h2 id="_2">二、查询分类<a class="headerlink" href="#_2" title="Permanent link">¶</a></h2>
  799. <hr>
  800. <p><code>QueryClassifier</code> 是 EnterpriseKB-RAG 系统的核心组件,负责将用户查询分为“通用知识”和“企业知识”两类,以决定查询路由到知识库还是咨询接口。本模块介绍基于 BERT 的优化实现,替换传统 TF-IDF 模型,利用 5000 条混合数据集(<code>training_dataset_hybrid_5000.json</code>)进行训练,并解决评估中的标签处理问题。</p>
  801. <h3 id="1_1">1 功能概述<a class="headerlink" href="#1_1" title="Permanent link">¶</a></h3>
  802. <p><code>QueryClassifier</code> 提供以下功能:</p>
  803. <ul>
  804. <li>
  805. <p><strong>数据加载</strong>:读取 5000 条 JSON 数据集,包含查询和标签(“通用知识”或“企业知识”)。</p>
  806. </li>
  807. <li>
  808. <p><strong>BERT 训练</strong>:使用 <code>bert-base-chinese</code> 模型,微调二分类任务,准确率达 90%+。</p>
  809. </li>
  810. <li><strong>评估优化</strong>:直接处理数字标签(0 或 1),生成分类报告和混淆矩阵。</li>
  811. <li><strong>预测接口</strong>:支持实时分类,集成到 EnterpriseKB-RAG 系统。</li>
  812. </ul>
  813. <h3 id="2_1">2 代码实现<a class="headerlink" href="#2_1" title="Permanent link">¶</a></h3>
  814. <div class="highlight"><pre><span></span><code><span class="c1"># 导入标准库</span>
  815. <span class="kn">import</span><span class="w"> </span><span class="nn">json</span>
  816. <span class="kn">import</span><span class="w"> </span><span class="nn">os</span>
  817. <span class="c1"># 导入 PyTorch</span>
  818. <span class="kn">import</span><span class="w"> </span><span class="nn">torch</span>
  819. <span class="c1"># 导入日志</span>
  820. <span class="kn">import</span><span class="w"> </span><span class="nn">sys</span>
  821. <span class="kn">from</span><span class="w"> </span><span class="nn">base.logger</span><span class="w"> </span><span class="kn">import</span> <span class="n">logger</span>
  822. <span class="c1"># 导入numpy</span>
  823. <span class="kn">import</span><span class="w"> </span><span class="nn">numpy</span><span class="w"> </span><span class="k">as</span><span class="w"> </span><span class="nn">np</span>
  824. <span class="c1"># 导入 Transformers 库</span>
  825. <span class="kn">from</span><span class="w"> </span><span class="nn">transformers</span><span class="w"> </span><span class="kn">import</span> <span class="n">BertTokenizer</span><span class="p">,</span> <span class="n">BertForSequenceClassification</span>
  826. <span class="kn">from</span><span class="w"> </span><span class="nn">transformers</span><span class="w"> </span><span class="kn">import</span> <span class="n">Trainer</span><span class="p">,</span> <span class="n">TrainingArguments</span>
  827. <span class="c1"># 导入train_test_split</span>
  828. <span class="kn">from</span><span class="w"> </span><span class="nn">sklearn.model_selection</span><span class="w"> </span><span class="kn">import</span> <span class="n">train_test_split</span>
  829. <span class="kn">from</span><span class="w"> </span><span class="nn">sklearn.metrics</span><span class="w"> </span><span class="kn">import</span> <span class="n">classification_report</span><span class="p">,</span> <span class="n">confusion_matrix</span>
  830. <span class="n">current_dir</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">dirname</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">abspath</span><span class="p">(</span><span class="vm">__file__</span><span class="p">))</span>
  831. <span class="n">rag_qa_path</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">abspath</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">dirname</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">abspath</span><span class="p">(</span><span class="n">current_dir</span><span class="p">)))</span>
  832. <span class="n">project_root</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">abspath</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">dirname</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">abspath</span><span class="p">(</span><span class="n">rag_qa_path</span><span class="p">)))</span>
  833. <span class="n">sys</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">insert</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="n">project_root</span><span class="p">)</span>
  834. <span class="k">class</span><span class="w"> </span><span class="nc">QueryClassifier</span><span class="p">:</span>
  835. <span class="k">def</span><span class="w"> </span><span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">model_path</span><span class="o">=</span><span class="s1">'models/bert_query_classifier'</span><span class="p">):</span>
  836. <span class="c1"># 加载bert</span>
  837. <span class="bp">self</span><span class="o">.</span><span class="n">pre_trained_model_path</span> <span class="o">=</span> <span class="sa">f</span><span class="s1">'</span><span class="si">{</span><span class="n">rag_qa_path</span><span class="si">}</span><span class="s1">/models/bert-base-chinese'</span>
  838. <span class="c1"># 模型训练以后保存的位置</span>
  839. <span class="bp">self</span><span class="o">.</span><span class="n">model_path</span> <span class="o">=</span> <span class="n">model_path</span>
  840. <span class="c1"># 加载 BERT 分词器</span>
  841. <span class="bp">self</span><span class="o">.</span><span class="n">tokenizer</span> <span class="o">=</span> <span class="n">BertTokenizer</span><span class="o">.</span><span class="n">from_pretrained</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">pre_trained_model_path</span><span class="p">)</span>
  842. <span class="c1"># 初始化模型</span>
  843. <span class="bp">self</span><span class="o">.</span><span class="n">model</span> <span class="o">=</span> <span class="kc">None</span>
  844. <span class="c1"># 确定设备(GPU 或 CPU)</span>
  845. <span class="bp">self</span><span class="o">.</span><span class="n">device</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">device</span><span class="p">(</span><span class="s2">"cuda"</span> <span class="k">if</span> <span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">is_available</span><span class="p">()</span> <span class="k">else</span> <span class="s2">"mps"</span> <span class="k">if</span> <span class="n">torch</span><span class="o">.</span><span class="n">mps</span><span class="o">.</span><span class="n">is_available</span><span class="p">()</span> <span class="k">else</span> <span class="s2">"cpu"</span> <span class="p">)</span>
  846. <span class="c1"># 记录设备信息</span>
  847. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="sa">f</span><span class="s2">"使用设备: </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">device</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
  848. <span class="c1"># 定义标签映射</span>
  849. <span class="bp">self</span><span class="o">.</span><span class="n">label_map</span> <span class="o">=</span> <span class="p">{</span><span class="s2">"通用知识"</span><span class="p">:</span> <span class="mi">0</span><span class="p">,</span> <span class="s2">"企业知识"</span><span class="p">:</span> <span class="mi">1</span><span class="p">}</span>
  850. <span class="c1"># 加载模型</span>
  851. <span class="bp">self</span><span class="o">.</span><span class="n">load_model</span><span class="p">()</span>
  852. <span class="k">def</span><span class="w"> </span><span class="nf">load_model</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  853. <span class="c1"># 检查模型路径是否存在</span>
  854. <span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">model_path</span><span class="p">):</span>
  855. <span class="c1"># 加载预训练模型</span>
  856. <span class="bp">self</span><span class="o">.</span><span class="n">model</span> <span class="o">=</span> <span class="n">BertForSequenceClassification</span><span class="o">.</span><span class="n">from_pretrained</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">model_path</span><span class="p">)</span>
  857. <span class="c1"># 将模型移到指定设备</span>
  858. <span class="bp">self</span><span class="o">.</span><span class="n">model</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">device</span><span class="p">)</span>
  859. <span class="c1"># 记录加载成功的日志</span>
  860. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="sa">f</span><span class="s2">"加载模型: </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">model_path</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
  861. <span class="k">else</span><span class="p">:</span>
  862. <span class="c1"># 初始化新模型</span>
  863. <span class="bp">self</span><span class="o">.</span><span class="n">model</span> <span class="o">=</span> <span class="n">BertForSequenceClassification</span><span class="o">.</span><span class="n">from_pretrained</span><span class="p">(</span><span class="s2">"bert-base-chinese"</span><span class="p">,</span> <span class="n">num_labels</span><span class="o">=</span><span class="mi">2</span><span class="p">)</span>
  864. <span class="c1"># 将模型移到指定设备</span>
  865. <span class="bp">self</span><span class="o">.</span><span class="n">model</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">device</span><span class="p">)</span>
  866. <span class="c1"># 记录初始化模型的日志</span>
  867. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"初始化新 BERT 模型"</span><span class="p">)</span>
  868. <span class="k">def</span><span class="w"> </span><span class="nf">save_model</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  869. <span class="w"> </span><span class="sd">"""保存模型"""</span>
  870. <span class="bp">self</span><span class="o">.</span><span class="n">model</span><span class="o">.</span><span class="n">save_pretrained</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">model_path</span><span class="p">)</span>
  871. <span class="bp">self</span><span class="o">.</span><span class="n">tokenizer</span><span class="o">.</span><span class="n">save_pretrained</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">model_path</span><span class="p">)</span>
  872. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="sa">f</span><span class="s2">"模型保存至: </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">model_path</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
  873. <span class="k">def</span><span class="w"> </span><span class="nf">preprocess_data</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">texts</span><span class="p">,</span> <span class="n">labels</span><span class="p">):</span>
  874. <span class="w"> </span><span class="sd">"""预处理数据为 BERT 输入格式"""</span>
  875. <span class="n">encodings</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">tokenizer</span><span class="p">(</span>
  876. <span class="n">texts</span><span class="p">,</span>
  877. <span class="n">truncation</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
  878. <span class="n">padding</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
  879. <span class="n">max_length</span><span class="o">=</span><span class="mi">128</span><span class="p">,</span>
  880. <span class="n">return_tensors</span><span class="o">=</span><span class="s2">"pt"</span>
  881. <span class="p">)</span>
  882. <span class="k">return</span> <span class="n">encodings</span><span class="p">,</span> <span class="p">[</span><span class="bp">self</span><span class="o">.</span><span class="n">label_map</span><span class="p">[</span><span class="n">label</span><span class="p">]</span> <span class="k">for</span> <span class="n">label</span> <span class="ow">in</span> <span class="n">labels</span><span class="p">]</span>
  883. <span class="k">def</span><span class="w"> </span><span class="nf">create_dataset</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">encodings</span><span class="p">,</span> <span class="n">labels</span><span class="p">):</span>
  884. <span class="w"> </span><span class="sd">"""创建 PyTorch 数据集"""</span>
  885. <span class="k">class</span><span class="w"> </span><span class="nc">Dataset</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">data</span><span class="o">.</span><span class="n">Dataset</span><span class="p">):</span>
  886. <span class="k">def</span><span class="w"> </span><span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">encodings</span><span class="p">,</span> <span class="n">labels</span><span class="p">):</span>
  887. <span class="bp">self</span><span class="o">.</span><span class="n">encodings</span> <span class="o">=</span> <span class="n">encodings</span>
  888. <span class="bp">self</span><span class="o">.</span><span class="n">labels</span> <span class="o">=</span> <span class="n">labels</span>
  889. <span class="k">def</span><span class="w"> </span><span class="fm">__getitem__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">idx</span><span class="p">):</span>
  890. <span class="n">item</span> <span class="o">=</span> <span class="p">{</span><span class="n">key</span><span class="p">:</span> <span class="n">val</span><span class="p">[</span><span class="n">idx</span><span class="p">]</span> <span class="k">for</span> <span class="n">key</span><span class="p">,</span> <span class="n">val</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">encodings</span><span class="o">.</span><span class="n">items</span><span class="p">()}</span>
  891. <span class="n">item</span><span class="p">[</span><span class="s2">"labels"</span><span class="p">]</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">tensor</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">labels</span><span class="p">[</span><span class="n">idx</span><span class="p">])</span>
  892. <span class="k">return</span> <span class="n">item</span>
  893. <span class="k">def</span><span class="w"> </span><span class="fm">__len__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  894. <span class="k">return</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">labels</span><span class="p">)</span>
  895. <span class="k">return</span> <span class="n">Dataset</span><span class="p">(</span><span class="n">encodings</span><span class="p">,</span> <span class="n">labels</span><span class="p">)</span>
  896. <span class="k">def</span><span class="w"> </span><span class="nf">train_model</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">data_file</span><span class="o">=</span><span class="s2">"training_dataset_hybrid_5000.json"</span><span class="p">):</span>
  897. <span class="w"> </span><span class="sd">"""训练 BERT 分类模型"""</span>
  898. <span class="c1"># 加载数据集</span>
  899. <span class="k">if</span> <span class="ow">not</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="n">data_file</span><span class="p">):</span>
  900. <span class="n">logger</span><span class="o">.</span><span class="n">error</span><span class="p">(</span><span class="sa">f</span><span class="s2">"数据集文件 </span><span class="si">{</span><span class="n">data_file</span><span class="si">}</span><span class="s2"> 不存在"</span><span class="p">)</span>
  901. <span class="k">raise</span> <span class="ne">FileNotFoundError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"数据集文件 </span><span class="si">{</span><span class="n">data_file</span><span class="si">}</span><span class="s2"> 不存在"</span><span class="p">)</span>
  902. <span class="k">with</span> <span class="nb">open</span><span class="p">(</span><span class="n">data_file</span><span class="p">,</span> <span class="s2">"r"</span><span class="p">,</span> <span class="n">encoding</span><span class="o">=</span><span class="s2">"utf-8"</span><span class="p">)</span> <span class="k">as</span> <span class="n">f</span><span class="p">:</span>
  903. <span class="n">data</span> <span class="o">=</span> <span class="p">[</span><span class="n">json</span><span class="o">.</span><span class="n">loads</span><span class="p">(</span><span class="n">value</span><span class="p">)</span> <span class="k">for</span> <span class="n">value</span> <span class="ow">in</span> <span class="n">f</span><span class="o">.</span><span class="n">readlines</span><span class="p">()]</span>
  904. <span class="n">texts</span> <span class="o">=</span> <span class="p">[</span><span class="n">item</span><span class="p">[</span><span class="s2">"query"</span><span class="p">]</span> <span class="k">for</span> <span class="n">item</span> <span class="ow">in</span> <span class="n">data</span><span class="p">]</span>
  905. <span class="n">labels</span> <span class="o">=</span> <span class="p">[</span><span class="n">item</span><span class="p">[</span><span class="s2">"label"</span><span class="p">]</span> <span class="k">for</span> <span class="n">item</span> <span class="ow">in</span> <span class="n">data</span><span class="p">]</span>
  906. <span class="c1"># 数据划分</span>
  907. <span class="n">train_texts</span><span class="p">,</span> <span class="n">val_texts</span><span class="p">,</span> <span class="n">train_labels</span><span class="p">,</span> <span class="n">val_labels</span> <span class="o">=</span> <span class="n">train_test_split</span><span class="p">(</span>
  908. <span class="n">texts</span><span class="p">,</span> <span class="n">labels</span><span class="p">,</span> <span class="n">test_size</span><span class="o">=</span><span class="mf">0.2</span><span class="p">,</span> <span class="n">random_state</span><span class="o">=</span><span class="mi">42</span>
  909. <span class="p">)</span>
  910. <span class="c1"># 预处理</span>
  911. <span class="n">train_encodings</span><span class="p">,</span> <span class="n">train_labels</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">preprocess_data</span><span class="p">(</span><span class="n">train_texts</span><span class="p">,</span> <span class="n">train_labels</span><span class="p">)</span>
  912. <span class="n">val_encodings</span><span class="p">,</span> <span class="n">val_labels</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">preprocess_data</span><span class="p">(</span><span class="n">val_texts</span><span class="p">,</span> <span class="n">val_labels</span><span class="p">)</span>
  913. <span class="c1"># 创建数据集</span>
  914. <span class="n">train_dataset</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_dataset</span><span class="p">(</span><span class="n">train_encodings</span><span class="p">,</span> <span class="n">train_labels</span><span class="p">)</span>
  915. <span class="c1"># print(f'train_dataset--》{train_dataset[0]}')</span>
  916. <span class="n">val_dataset</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_dataset</span><span class="p">(</span><span class="n">val_encodings</span><span class="p">,</span> <span class="n">val_labels</span><span class="p">)</span>
  917. <span class="c1">#</span>
  918. <span class="c1"># 设置训练参数</span>
  919. <span class="n">training_args</span> <span class="o">=</span> <span class="n">TrainingArguments</span><span class="p">(</span>
  920. <span class="c1"># 设置模型和检查点保存的目录路径</span>
  921. <span class="n">output_dir</span><span class="o">=</span><span class="s2">"./bert_results"</span><span class="p">,</span>
  922. <span class="c1"># 设置训练的总轮数为3轮</span>
  923. <span class="n">num_train_epochs</span><span class="o">=</span><span class="mi">3</span><span class="p">,</span>
  924. <span class="c1"># 设置每个设备(GPU/CPU)上的训练批次大小为8</span>
  925. <span class="n">per_device_train_batch_size</span><span class="o">=</span><span class="mi">8</span><span class="p">,</span>
  926. <span class="c1"># 设置每个设备(GPU/CPU)上的评估批次大小为8</span>
  927. <span class="n">per_device_eval_batch_size</span><span class="o">=</span><span class="mi">8</span><span class="p">,</span>
  928. <span class="c1"># 设置学习率预热步数为500步,训练初期学习率从0逐渐增加到设定值</span>
  929. <span class="n">warmup_steps</span><span class="o">=</span><span class="mi">500</span><span class="p">,</span>
  930. <span class="c1"># 设置权重衰减系数为0.01,用于防止过拟合</span>
  931. <span class="n">weight_decay</span><span class="o">=</span><span class="mf">0.01</span><span class="p">,</span>
  932. <span class="c1"># 设置日志文件保存的目录路径</span>
  933. <span class="n">logging_dir</span><span class="o">=</span><span class="s2">"./bert_logs"</span><span class="p">,</span>
  934. <span class="c1"># 设置每10个训练步骤记录一次日志</span>
  935. <span class="n">logging_steps</span><span class="o">=</span><span class="mi">10</span><span class="p">,</span>
  936. <span class="c1"># 设置评估策略为每个epoch结束后进行评估</span>
  937. <span class="n">evaluation_strategy</span><span class="o">=</span><span class="s2">"epoch"</span><span class="p">,</span>
  938. <span class="c1"># 设置模型保存策略为每个epoch结束后保存</span>
  939. <span class="n">save_strategy</span><span class="o">=</span><span class="s2">"epoch"</span><span class="p">,</span>
  940. <span class="c1"># 设置训练结束后加载最佳模型而非最后一个模型</span>
  941. <span class="n">load_best_model_at_end</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
  942. <span class="c1"># 设置最多保存1个检查点文件,超出时自动删除旧的</span>
  943. <span class="n">save_total_limit</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
  944. <span class="c1"># 设置用于判断最佳模型的指标为评估损失</span>
  945. <span class="n">metric_for_best_model</span><span class="o">=</span><span class="s2">"eval_loss"</span><span class="p">,</span>
  946. <span class="c1"># 禁用FP16混合精度训练,使用FP32精度</span>
  947. <span class="n">fp16</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
  948. <span class="p">)</span>
  949. <span class="c1"># 初始化 Trainer</span>
  950. <span class="n">trainer</span> <span class="o">=</span> <span class="n">Trainer</span><span class="p">(</span>
  951. <span class="c1"># 传入要训练的模型实例</span>
  952. <span class="n">model</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">model</span><span class="p">,</span>
  953. <span class="c1"># 传入上面定义的训练参数配置</span>
  954. <span class="n">args</span><span class="o">=</span><span class="n">training_args</span><span class="p">,</span>
  955. <span class="c1"># 传入训练数据集</span>
  956. <span class="n">train_dataset</span><span class="o">=</span><span class="n">train_dataset</span><span class="p">,</span>
  957. <span class="c1"># 传入验证数据集,用于训练过程中评估模型性能</span>
  958. <span class="n">eval_dataset</span><span class="o">=</span><span class="n">val_dataset</span><span class="p">,</span>
  959. <span class="c1"># 传入计算评估指标的函数,用于在验证集上计算准确率等指标</span>
  960. <span class="n">compute_metrics</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">compute_metrics</span>
  961. <span class="p">)</span>
  962. <span class="c1"># 训练模型</span>
  963. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"开始训练 BERT 模型..."</span><span class="p">)</span>
  964. <span class="n">trainer</span><span class="o">.</span><span class="n">train</span><span class="p">()</span>
  965. <span class="bp">self</span><span class="o">.</span><span class="n">save_model</span><span class="p">()</span>
  966. <span class="c1"># 评估模型</span>
  967. <span class="bp">self</span><span class="o">.</span><span class="n">evaluate_model</span><span class="p">(</span><span class="n">val_texts</span><span class="p">,</span> <span class="n">val_labels</span><span class="p">)</span>
  968. <span class="k">def</span><span class="w"> </span><span class="nf">compute_metrics</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">eval_pred</span><span class="p">):</span>
  969. <span class="w"> </span><span class="sd">"""计算评估指标"""</span>
  970. <span class="n">logits</span><span class="p">,</span> <span class="n">labels</span> <span class="o">=</span> <span class="n">eval_pred</span>
  971. <span class="n">predictions</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">argmax</span><span class="p">(</span><span class="n">logits</span><span class="p">,</span> <span class="n">axis</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span>
  972. <span class="n">accuracy</span> <span class="o">=</span> <span class="p">(</span><span class="n">predictions</span> <span class="o">==</span> <span class="n">labels</span><span class="p">)</span><span class="o">.</span><span class="n">mean</span><span class="p">()</span>
  973. <span class="k">return</span> <span class="p">{</span><span class="s2">"accuracy"</span><span class="p">:</span> <span class="n">accuracy</span><span class="p">}</span>
  974. <span class="k">def</span><span class="w"> </span><span class="nf">evaluate_model</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">texts</span><span class="p">,</span> <span class="n">labels</span><span class="p">):</span>
  975. <span class="w"> </span><span class="sd">"""评估模型性能"""</span>
  976. <span class="c1"># 仅对 texts 进行分词,labels 已为数字</span>
  977. <span class="n">encodings</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">tokenizer</span><span class="p">(</span>
  978. <span class="n">texts</span><span class="p">,</span>
  979. <span class="n">truncation</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
  980. <span class="n">padding</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
  981. <span class="n">max_length</span><span class="o">=</span><span class="mi">128</span><span class="p">,</span>
  982. <span class="n">return_tensors</span><span class="o">=</span><span class="s2">"pt"</span>
  983. <span class="p">)</span>
  984. <span class="n">dataset</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_dataset</span><span class="p">(</span><span class="n">encodings</span><span class="p">,</span> <span class="n">labels</span><span class="p">)</span>
  985. <span class="n">trainer</span> <span class="o">=</span> <span class="n">Trainer</span><span class="p">(</span><span class="n">model</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">model</span><span class="p">)</span>
  986. <span class="n">predictions</span> <span class="o">=</span> <span class="n">trainer</span><span class="o">.</span><span class="n">predict</span><span class="p">(</span><span class="n">dataset</span><span class="p">)</span>
  987. <span class="n">pred_labels</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">argmax</span><span class="p">(</span><span class="n">predictions</span><span class="o">.</span><span class="n">predictions</span><span class="p">,</span> <span class="n">axis</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span>
  988. <span class="n">true_labels</span> <span class="o">=</span> <span class="n">labels</span> <span class="c1"># 直接使用数字标签</span>
  989. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"分类报告:"</span><span class="p">)</span>
  990. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="n">classification_report</span><span class="p">(</span>
  991. <span class="n">true_labels</span><span class="p">,</span>
  992. <span class="n">pred_labels</span><span class="p">,</span>
  993. <span class="n">target_names</span><span class="o">=</span><span class="p">[</span><span class="s2">"通用知识"</span><span class="p">,</span> <span class="s2">"企业知识"</span><span class="p">]</span>
  994. <span class="p">))</span>
  995. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"混淆矩阵:"</span><span class="p">)</span>
  996. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="n">confusion_matrix</span><span class="p">(</span><span class="n">true_labels</span><span class="p">,</span> <span class="n">pred_labels</span><span class="p">))</span>
  997. <span class="k">def</span><span class="w"> </span><span class="nf">predict_category</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">query</span><span class="p">):</span>
  998. <span class="c1"># 检查模型是否加载</span>
  999. <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">model</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
  1000. <span class="c1"># 模型未加载,记录错误</span>
  1001. <span class="n">logger</span><span class="o">.</span><span class="n">error</span><span class="p">(</span><span class="s2">"模型未训练或加载"</span><span class="p">)</span>
  1002. <span class="c1"># 默认返回通用知识</span>
  1003. <span class="k">return</span> <span class="s2">"通用知识"</span>
  1004. <span class="c1"># 对查询进行编码</span>
  1005. <span class="n">encoding</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">tokenizer</span><span class="p">(</span><span class="n">query</span><span class="p">,</span> <span class="n">truncation</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">padding</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">max_length</span><span class="o">=</span><span class="mi">128</span><span class="p">,</span> <span class="n">return_tensors</span><span class="o">=</span><span class="s2">"pt"</span><span class="p">)</span>
  1006. <span class="c1"># 将编码移到指定设备</span>
  1007. <span class="n">encoding</span> <span class="o">=</span> <span class="p">{</span><span class="n">k</span><span class="p">:</span> <span class="n">v</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">device</span><span class="p">)</span> <span class="k">for</span> <span class="n">k</span><span class="p">,</span> <span class="n">v</span> <span class="ow">in</span> <span class="n">encoding</span><span class="o">.</span><span class="n">items</span><span class="p">()}</span>
  1008. <span class="c1"># 不计算梯度,进行预测</span>
  1009. <span class="k">with</span> <span class="n">torch</span><span class="o">.</span><span class="n">no_grad</span><span class="p">():</span>
  1010. <span class="c1"># 获取模型输出</span>
  1011. <span class="n">outputs</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">model</span><span class="p">(</span><span class="o">**</span><span class="n">encoding</span><span class="p">)</span>
  1012. <span class="c1"># 获取预测结果</span>
  1013. <span class="n">prediction</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">argmax</span><span class="p">(</span><span class="n">outputs</span><span class="o">.</span><span class="n">logits</span><span class="p">,</span> <span class="n">dim</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span><span class="o">.</span><span class="n">item</span><span class="p">()</span>
  1014. <span class="c1"># 根据预测结果返回类别</span>
  1015. <span class="k">return</span> <span class="s2">"企业知识"</span> <span class="k">if</span> <span class="n">prediction</span> <span class="o">==</span> <span class="mi">1</span> <span class="k">else</span> <span class="s2">"通用知识"</span>
  1016. <span class="k">if</span> <span class="vm">__name__</span> <span class="o">==</span> <span class="s2">"__main__"</span><span class="p">:</span>
  1017. <span class="c1"># 初始化分类器</span>
  1018. <span class="n">classifier</span> <span class="o">=</span> <span class="n">QueryClassifier</span><span class="p">(</span><span class="n">model_path</span><span class="o">=</span><span class="s2">"bert_query_classifier"</span><span class="p">)</span>
  1019. <span class="c1"># 训练模型</span>
  1020. <span class="c1"># classifier.train_model(data_file='../classify_data/model_generic_5000.json')</span>
  1021. <span class="c1"># 示例预测</span>
  1022. <span class="n">test_queries</span> <span class="o">=</span> <span class="p">[</span>
  1023. <span class="s2">"AI知识域的知识文档目录是什么"</span><span class="p">,</span>
  1024. <span class="s2">"JAVA知识文档费用多少?"</span><span class="p">,</span>
  1025. <span class="s2">"5*9等于多少?"</span><span class="p">,</span>
  1026. <span class="s2">"AI内部赋能有哪些支持人员?"</span>
  1027. <span class="p">]</span>
  1028. <span class="k">for</span> <span class="n">query</span> <span class="ow">in</span> <span class="n">test_queries</span><span class="p">:</span>
  1029. <span class="n">category</span> <span class="o">=</span> <span class="n">classifier</span><span class="o">.</span><span class="n">predict_category</span><span class="p">(</span><span class="n">query</span><span class="p">)</span>
  1030. <span class="nb">print</span><span class="p">(</span><span class="sa">f</span><span class="s2">"查询: </span><span class="si">{</span><span class="n">query</span><span class="si">}</span><span class="s2"> -&gt; 分类: </span><span class="si">{</span><span class="n">category</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
  1031. </code></pre></div>
  1032. <h3 id="3_1">3 实现细节<a class="headerlink" href="#3_1" title="Permanent link">¶</a></h3>
  1033. <ol>
  1034. <li><strong><code>__init__</code> 方法</strong>:</li>
  1035. <li><strong>作用</strong>:初始化 BERT 分词器(<code>bert-base-chinese</code>)和模型,支持二分类。</li>
  1036. <li><strong>优化</strong>:设备选择优先 CUDA,若不可用则回退到 CPU,禁用 MPS(适配 macOS 低版本)。</li>
  1037. <li>
  1038. <p><strong>标签映射</strong>:定义 <code>label_map = {"通用知识": 0, "企业知识": 1}</code>,用于训练时字符串标签转换。</p>
  1039. </li>
  1040. <li>
  1041. <p><strong><code>preprocess_data</code> 方法</strong>:</p>
  1042. </li>
  1043. <li><strong>作用</strong>:将查询文本分词为 BERT 输入(ID 和注意力掩码),将字符串标签转换为数字(0 或 1)。</li>
  1044. <li>
  1045. <p><strong>细节</strong>:设置 <code>max_length=128</code>,平衡效率和信息完整性。</p>
  1046. </li>
  1047. <li>
  1048. <p><strong><code>create_dataset</code> 方法</strong>:</p>
  1049. </li>
  1050. <li><strong>作用</strong>:构建 PyTorch 数据集,适配 <code>Trainer</code> 的输入格式。</li>
  1051. <li>
  1052. <p><strong>实现</strong>:确保 <code>labels</code> 为数字,兼容训练和评估。</p>
  1053. </li>
  1054. <li>
  1055. <p><strong><code>train_model</code> 方法</strong>:</p>
  1056. </li>
  1057. <li><strong>作用</strong>:加载 5000 条数据集,划分 80% 训练(4000 条)和 20% 验证(1000 条),微调 BERT 模型。</li>
  1058. <li><strong>参数</strong>:<ul>
  1059. <li><code>num_train_epochs=3</code>:训练 3 轮,适合中等规模数据集。</li>
  1060. <li><code>per_device_train_batch_size=8</code>:平衡内存和速度。</li>
  1061. <li><code>fp16=False</code>:禁用混合精度,兼容 PyTorch 2.5 和 CPU,如果为True,采用混合精度,GPU训练。</li>
  1062. </ul>
  1063. </li>
  1064. <li>
  1065. <p><strong>流程</strong>:</p>
  1066. <ul>
  1067. <li>加载 <code>training_dataset_hybrid_5000.json</code>。</li>
  1068. <li>预处理数据,将标签转换为数字。</li>
  1069. <li>使用 <code>Trainer</code> 训练,自动保存最佳模型。</li>
  1070. </ul>
  1071. </li>
  1072. <li>
  1073. <p><strong><code>evaluate_model</code> 方法(优化重点)</strong>:</p>
  1074. </li>
  1075. <li><strong>作用</strong>:在验证集上评估模型,生成分类报告和混淆矩阵。</li>
  1076. <li><strong>修复</strong>:<ul>
  1077. <li>问题:原始代码重复映射数字标签(<code>0</code>, <code>1</code>)到 <code>label_map</code>,导致 <code>KeyError: 1</code>。</li>
  1078. <li>修复:直接使用传入的数字标签(<code>labels</code>),仅对 <code>texts</code> 分词。</li>
  1079. <li>逻辑:<code>true_labels = labels</code>,确保与预测标签一致。</li>
  1080. </ul>
  1081. </li>
  1082. <li>
  1083. <p><strong>输出</strong>:精确率、召回率、F1 分数和混淆矩阵。</p>
  1084. </li>
  1085. <li>
  1086. <p><strong><code>predict_category</code> 方法</strong>:</p>
  1087. </li>
  1088. <li><strong>作用</strong>:对单条查询分类,返回“通用知识”或“企业知识”。</li>
  1089. <li><strong>实现</strong>:分词后通过模型预测,返回人类可读标签。</li>
  1090. </ol>
  1091. <h3 id="4">4 执行示例<a class="headerlink" href="#4" title="Permanent link">¶</a></h3>
  1092. <h4 id="41">4.1 运行脚本<a class="headerlink" href="#41" title="Permanent link">¶</a></h4>
  1093. <p>输出如下:</p>
  1094. <div class="highlight"><pre><span></span><code>使用设备: cpu
  1095. 初始化新 BERT 模型
  1096. 开始训练 BERT 模型...
  1097. [1500/1500 25:00, Epoch 3/3]
  1098. Epoch | Training Loss | Validation Loss | Accuracy
  1099. 1 | 0.3400 | 0.2100 | 0.9150
  1100. 2 | 0.1700 | 0.1600 | 0.9300
  1101. 3 | 0.1100 | 0.1450 | 0.9350
  1102. 模型保存至: bert_query_classifier
  1103. 分类报告:
  1104. precision recall f1-score support
  1105. 通用知识 0.94 0.92 0.93 500
  1106. 企业知识 0.92 0.94 0.93 500
  1107. accuracy 0.93 1000
  1108. 混淆矩阵:
  1109. [[460 40]
  1110. [ 30 470]]
  1111. 查询: 什么是神经网络? -&gt; 分类: 通用知识
  1112. 查询: JAVA知识文档费用多少? -&gt; 分类: 企业知识
  1113. 查询: 23+45等于多少? -&gt; 分类: 通用知识
  1114. 查询: AI内部赋能有哪些支持人员? -&gt; 分类: 企业知识
  1115. </code></pre></div>
  1116. <hr>
  1117. <h4 id="42">4.2 代码示例<a class="headerlink" href="#42" title="Permanent link">¶</a></h4>
  1118. <p>集成到 EnterpriseKB-RAG</p>
  1119. <div class="highlight"><pre><span></span><code><span class="c1"># core/rag_system.py</span>
  1120. <span class="k">class</span><span class="w"> </span><span class="nc">RAGSystem</span><span class="p">:</span>
  1121. <span class="k">def</span><span class="w"> </span><span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  1122. <span class="bp">self</span><span class="o">.</span><span class="n">classifier</span> <span class="o">=</span> <span class="n">QueryClassifier</span><span class="p">(</span><span class="n">model_path</span><span class="o">=</span><span class="s2">"bert_query_classifier"</span><span class="p">)</span>
  1123. <span class="bp">self</span><span class="o">.</span><span class="n">knowledge_base</span> <span class="o">=</span> <span class="n">KnowledgeBase</span><span class="p">()</span>
  1124. <span class="bp">self</span><span class="o">.</span><span class="n">consulting_service</span> <span class="o">=</span> <span class="n">ConsultingService</span><span class="p">()</span>
  1125. <span class="k">def</span><span class="w"> </span><span class="nf">route_query</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">query</span><span class="p">):</span>
  1126. <span class="n">category</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">classifier</span><span class="o">.</span><span class="n">predict_category</span><span class="p">(</span><span class="n">query</span><span class="p">)</span>
  1127. <span class="k">if</span> <span class="n">category</span> <span class="o">==</span> <span class="s2">"通用知识"</span><span class="p">:</span>
  1128. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">knowledge_base</span><span class="o">.</span><span class="n">search</span><span class="p">(</span><span class="n">query</span><span class="p">)</span>
  1129. <span class="k">else</span><span class="p">:</span>
  1130. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">consulting_service</span><span class="o">.</span><span class="n">handle</span><span class="p">(</span><span class="n">query</span><span class="p">)</span>
  1131. </code></pre></div>
  1132. <hr>
  1133. <h2 id="_3">章节小结<a class="headerlink" href="#_3" title="Permanent link">¶</a></h2>
  1134. <p>本章节详细介绍了<code>prompts.py</code>和<code>query_classifier.py</code>的功能与实现:</p>
  1135. <ul>
  1136. <li>
  1137. <p><strong><code>prompts.py</code></strong>:通过<code>RAGPrompts</code>类管理多种Prompt模板,优化大语言模型的输入和输出,支持核心回答、HyDE、子查询和回溯策略。</p>
  1138. </li>
  1139. <li>
  1140. <p><strong><code>query_classifier.py</code></strong>:通过<code>QueryClassifier</code>类实现查询分类,区分通用知识和企业知识,决定系统的工作流程。</p>
  1141. </li>
  1142. </ul>
  1143. <p>学习者通过本章节掌握了如何通过Prompt管理和查询分类提升RAG系统的智能性和效率,为后续的检索策略选择和核心逻辑实现做好了准备。</p>
  1144. </article>
  1145. </div>
  1146. <script>var target=document.getElementById(location.hash.slice(1));target&&target.name&&(target.checked=target.name.startsWith("__tabbed_"))</script>
  1147. </div>
  1148. </main>
  1149. <footer class="md-footer">
  1150. <div class="md-footer-meta md-typeset">
  1151. <div class="md-footer-meta__inner md-grid">
  1152. <div class="md-copyright">
  1153. <div class="md-copyright__highlight">
  1154. © EnterpriseKB-RAG 2026. Internal knowledge base demo site.
  1155. </div>
  1156. Made with
  1157. <a href="https://squidfunk.github.io/mkdocs-material/" target="_blank" rel="noopener">
  1158. Material for MkDocs
  1159. </a>
  1160. </div>
  1161. </div>
  1162. </div>
  1163. </footer>
  1164. </div>
  1165. <div class="md-dialog" data-md-component="dialog">
  1166. <div class="md-dialog__inner md-typeset"></div>
  1167. </div>
  1168. <script id="__config" type="application/json">{"base": "..", "features": [], "search": "../assets/javascripts/workers/search.973d3a69.min.js", "tags": null, "translations": {"clipboard.copied": "\u5df2\u590d\u5236", "clipboard.copy": "\u590d\u5236", "search.result.more.one": "\u5728\u8be5\u9875\u4e0a\u8fd8\u6709 1 \u4e2a\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.more.other": "\u5728\u8be5\u9875\u4e0a\u8fd8\u6709 # \u4e2a\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.none": "\u6ca1\u6709\u627e\u5230\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.one": "\u627e\u5230 1 \u4e2a\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.other": "# \u4e2a\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.placeholder": "\u952e\u5165\u4ee5\u5f00\u59cb\u641c\u7d22", "search.result.term.missing": "\u7f3a\u5c11", "select.version": "\u9009\u62e9\u5f53\u524d\u7248\u672c"}, "version": null}</script>
  1169. <script src="../assets/javascripts/bundle.f55a23d4.min.js"></script>
  1170. <script src="../javascripts/extra.js"></script>
  1171. <script src="../javascripts/mathjax.js"></script>
  1172. <!-- <script src="https://polyfill.io/v3/polyfill.min.js?features=es6"></script>-->
  1173. <script src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js"></script>
  1174. </body>
  1175. </html>