55< head >
66 < meta charset ="utf-8 " />
77 < meta name ="viewport " content ="width=device-width, initial-scale=1.0 " />
8- < title > pythainlp.chat.core — PyThaiNLP 82e1d9f documentation</ title >
8+ < title > pythainlp.chat.core — PyThaiNLP 2e0acf8 documentation</ title >
99 < link rel ="stylesheet " type ="text/css " href ="../../../_static/pygments.css?v=03e43079 " />
1010 < link rel ="stylesheet " type ="text/css " href ="../../../_static/css/theme.css?v=9edc463e " />
1111 < link rel ="stylesheet " type ="text/css " href ="../../../_static/copybutton.css?v=76b2166b " />
1414
1515 < script src ="../../../_static/jquery.js?v=5d32c60e "> </ script >
1616 < script src ="../../../_static/_sphinx_javascript_frameworks_compat.js?v=2cd50e6c "> </ script >
17- < script src ="../../../_static/documentation_options.js?v=d324fe00 "> </ script >
17+ < script src ="../../../_static/documentation_options.js?v=04cc7e5f "> </ script >
1818 < script src ="../../../_static/doctools.js?v=fd6eb6e6 "> </ script >
1919 < script src ="../../../_static/sphinx_highlight.js?v=6ffebe34 "> </ script >
2020 < script src ="../../../_static/clipboard.min.js?v=a7894cd8 "> </ script >
@@ -114,7 +114,7 @@ <h1>Source code for pythainlp.chat.core</h1><div class="highlight"><pre>
114114< span class ="c1 "> # SPDX-License-Identifier: Apache-2.0</ span >
115115< span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> __future__</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> annotations</ span >
116116
117- < span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> typing</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> TYPE_CHECKING</ span > < span class ="p "> ,</ span > < span class ="n "> Optional</ span >
117+ < span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> typing</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> TYPE_CHECKING</ span > < span class ="p "> ,</ span > < span class ="n "> Any </ span > < span class =" p " > , </ span > < span class =" n " > Optional</ span >
118118
119119< span class ="k "> if</ span > < span class ="n "> TYPE_CHECKING</ span > < span class ="p "> :</ span >
120120 < span class ="kn "> import</ span > < span class ="w "> </ span > < span class ="nn "> torch</ span >
@@ -123,6 +123,9 @@ <h1>Source code for pythainlp.chat.core</h1><div class="highlight"><pre>
123123< div class ="viewcode-block " id ="ChatBotModel ">
124124< a class ="viewcode-back " href ="../../../api/chat.html#pythainlp.chat.ChatBotModel "> [docs]</ a >
125125< span class ="k "> class</ span > < span class ="w "> </ span > < span class ="nc "> ChatBotModel</ span > < span class ="p "> :</ span >
126+ < span class ="n "> history</ span > < span class ="p "> :</ span > < span class ="nb "> list</ span > < span class ="p "> [</ span > < span class ="nb "> tuple</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span > < span class ="nb "> str</ span > < span class ="p "> ]]</ span >
127+ < span class ="n "> model</ span > < span class ="p "> :</ span > < span class ="n "> Any</ span >
128+
126129< div class ="viewcode-block " id ="ChatBotModel.__init__ ">
127130< a class ="viewcode-back " href ="../../../api/chat.html#pythainlp.chat.ChatBotModel.__init__ "> [docs]</ a >
128131 < 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 ="o "> -></ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
@@ -167,7 +170,7 @@ <h1>Source code for pythainlp.chat.core</h1><div class="highlight"><pre>
167170 < span class ="k "> if</ span > < span class ="n "> model_name</ span > < span class ="o "> ==</ span > < span class ="s2 "> "wangchanglm"</ span > < span class ="p "> :</ span >
168171 < span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> pythainlp.generate.wangchanglm</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> WangChanGLM</ span >
169172
170- < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> model</ span > < span class ="o "> =</ span > < span class ="n "> WangChanGLM</ span > < span class ="p "> ()</ span >
173+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> model</ span > < span class =" p " > : </ span > < span class =" n " > Any </ span > < span class ="o "> =</ span > < span class ="n "> WangChanGLM</ span > < span class ="p "> ()</ span >
171174 < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> model</ span > < span class ="o "> .</ span > < span class ="n "> load_model</ span > < span class ="p "> (</ span >
172175 < span class ="n "> model_path</ span > < span class ="o "> =</ span > < span class ="s2 "> "pythainlp/wangchanglm-7.5B-sft-en-sharded"</ span > < span class ="p "> ,</ span >
173176 < span class ="n "> return_dict</ span > < span class ="o "> =</ span > < span class ="n "> return_dict</ span > < span class ="p "> ,</ span >
@@ -218,7 +221,7 @@ <h1>Source code for pythainlp.chat.core</h1><div class="highlight"><pre>
218221 < span class ="p "> )</ span >
219222 < span class ="n "> _bot</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> model</ span > < span class ="o "> .</ span > < span class ="n "> gen_instruct</ span > < span class ="p "> (</ span > < span class ="n "> _temp</ span > < span class ="p "> )</ span >
220223 < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> history</ span > < span class ="o "> .</ span > < span class ="n "> append</ span > < span class ="p "> ((</ span > < span class ="n "> text</ span > < span class ="p "> ,</ span > < span class ="n "> _bot</ span > < span class ="p "> ))</ span >
221- < span class ="k "> return</ span > < span class ="n "> _bot</ span > < span class =" c1 " > # type: ignore[no-any-return] </ span > </ div >
224+ < span class ="k "> return</ span > < span class ="n "> _bot</ span > </ div >
222225</ div >
223226
224227</ pre > </ div >
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?v=4&size=40){kind=avatar}
0 commit comments