+ ^|i]^RIt^RIt^RIt^RIHt^RIHtHtHtH t H t H t ^RI H t Ht^RIt^RIt^RIHt^RIHt^RIHt^RIHtHt^R IHt^R IHt^R IH t ^R I!H"t"H#t#H$t$^R I%H&t&^RI'H(t(^RI)H*t*RRlt+!RR]"4t,!RR]4t-RRlt.RRlt/RRlt0RRlt1RRlt2R R!lt3R"R#lt4R$R%lt5R&R'lt6R(R)lt7R*R+lt8R,R-lt9R.R/lt:R0R1lt;]]]-R3,tR6R7lt?R#)8N)StringIO)ListOptionalUnionTupleClassVarAny)Callable GeneratorBuffer KeyPressEvent)named_commands)AutoSuggestFromHistory SuggestionDocument)History PromptSession) ProcessorTransformationTransformationInput) get_ipython)generate_tokens) pass_throughc$V^8dQhR\/#)documentr)formats"I/usr/lib/python3/dist-packages/IPython/terminal/shortcuts/auto_suggest.py __annotate__r#s888c<VPVP,#N)linescursor_position_row)r s&r" _get_queryr)s >>(66 77r$c\a]tRt^!toRtRtR V3RlRlltV3RlRltV3RltRt Vt R #) AppendAutoSuggestionInAnyLinea Append the auto suggestion to lines other than the last (appending to the last line is natively supported by the prompt toolkit). This has a private `_debug` attribute that can be set to True to display debug information as virtual suggestion on the end of any line. You can do so with: >>> from IPython.terminal.shortcuts.auto_suggest import AppendAutoSuggestionInAnyLine >>> AppendAutoSuggestionInAnyLine._debug = True Fc$<V^8dQhRS[RR/#)rstylereturnN)str)r! __classdict__s"r"r#*AppendAutoSuggestionInAnyLine.__annotate__1scr$cWnR#r&r-)selfr-s&&r"__init__&AppendAutoSuggestionInAnyLine.__init__1s r$c&<V^8dQhRS[RS[/#)rtir.)rr)r!r0s"r"r#r14s i6i6':i6~i6r$c aaSPP^, pSPV8HpVV3RlpSPP^8Xd V!R4#SPPV8XdV'd V!R4#SPSPP8d V!R4#SPP pVP 'dSPP'g V!R4#SPSPP, pVP PP4p\V4^8Xd V!R4#\PR8d\V4^8do\PR8dZSPSPP8Xd\SPR.,R7#\SPR7#\V4^8XdWSPSPP8Xd2\SPSPV^,3.,R7#\SPR7#V^8Xd4V^,p\SPSPV3.,R7#V'Ed<V\V48d/R\V4V, R 2p\SPV3.4#\V4p \!\V44FSp SP#W*, 4p R P%R V 44P'4p V 'dM V ^,p KU V 'd\SPR V R 23.4#\SP#V\V4, ^,4SP('dSPR3.,4#.,4#V\V48d!Wv,p\SPV3.4#SP\V4, ^,p \SP#V 44#)a Apply transformation to the line that is currently being edited. This is a variation of the original implementation in prompt toolkit that allows to not only append suggestions to any line, but also to show multi-line suggestions. As transformation are applied on a line-by-line basis; we need to trick a bit, and elide any line that is after the line we are currently editing, until we run out of completions. We cannot shift the existing lines There are multiple cases to handle: The completions ends before the end of the buffer: We can resume showing the normal line, and say that some code may be hidden. The completions ends at the end of the buffer We can just say that some code may be hidden. And separately: The completions ends beyond the end of the buffer We need to both say that some code may be hidden, and that some lines are not shown. c<\SPSPSP'd RV,MR3.,R7#)  fragments)rr>r-_debug)textr4r8s&r"DAppendAutoSuggestionInAnyLine.apply_transformation..Ts1Nlltzz3:RT&U%VV r$z noop:onelineznoop:last line and cursorznoop:before cursorz noop:not eolznoop: no suggestionsr=u… rest of suggestion (z lines) and code hiddenr<c32"TF q^,xK R#5i)N).0ls& r" EAppendAutoSuggestionInAnyLine.apply_transformation..s.2aqTT2su… z line(s) hiddenzshift-last-line)r1)redzD(Cannot show multiline suggestion; requires prompt_toolkit > 3.0.49))r line_countlinenor(buffer_controlbuffer suggestionis_cursor_at_the_end_of_liner@ splitlineslenprompt_toolkitVERSIONrr>r-rangeget_linejoinstripr?)r4r8last_line_number is_last_linenooprPdelta suggestionsrQn_elidedillelshiftsff r"apply_transformation2AppendAutoSuggestionInAnyLine.apply_transformation4s:;;11A5yy$44   ;; ! !Q &' ' ;; * *.> ><34 4 99r{{66 6,- -""))    (P(P(P' ' BKK;;;'',,779 { q ./ /  ! !J .;!#(>(>(K99 ? ??)"$,,#*BLLAA[!Q&99 ? ??)"$,,4::{1~2N1O"O"BLL9 9 A:$QJ!BLLTZZsgr$cj<\SV`4^Vn.VnRVnRVn^VnR#rN)superr5 skip_lines_connected_apps_llm_provider_instance_init_llm_provider_request_number)r4 __class__s&r"r5(NavigableAutoSuggestFromHistory.__init__s5 !&*#"& r$c$<V^8dQhRS[RR/#)r_r.Nr )r!r0s"r"r#,NavigableAutoSuggestFromHistory.__annotate__s4r$c^VnR#rx)rz)r4rs&&r"reset_history_position6NavigableAutoSuggestFromHistory.reset_history_positions r$c<V^8dQhRR/#rr.NrE)r!r0s"r"r#rsJJDJr$cVP4VPF4pVPPpVP VP 4K6 R#r&)_cancel_running_llm_taskr{default_bufferon_text_insertremove_handlerr)r4pt_apptext_insert_events& r" disconnect*NavigableAutoSuggestFromHistory.disconnectsD %%'**F & 5 5 D D   , ,T-H-H I+r$c$<V^8dQhRS[RR/#)rrr.Nr)r!r0s"r"r#rsTTmTTr$cVPPV4VPPP VP 4VPP P VP4R#r&)r{appendrr add_handlerron_cursor_position_changed_dismiss)r4rs&&r"connect'NavigableAutoSuggestFromHistory.connectsV ##F+ ,,889T9TU88DDT]]Sr$c<<V^8dQhRS[RS[RS[S[,/#)rrPr r.)r rrr)r!r0s"r"r#rs*   (0  *  r$c\V4pVP4'd;VPW0PVP4FwrE\ V4u# R#r&)r)rZ_find_next_matchrzhistoryr)r4rPr r@rQrs&&& r"get_suggestion.NavigableAutoSuggestFromHistory.get_suggestionsO(# ::<> %~~""$$!%--/I!% V &r$c VPf(VP'dVP4VnVP#)zLazy-initialized instance of the LLM provider. Do not use in the constructor, as `_init_llm_provider` can trigger slow side-effects. )r|r})r4s&r" _llm_provider-NavigableAutoSuggestFromHistory._llm_providerDs8  & & .43J3J3J*.*A*A*CD '***r$c<V^8dQhRR/#rrE)r!r0s"r"r#rNs  D r$c ba"^RIpSP'g\P!R4R#Vf\P!R4SP 4V3Rlp\ P!V!V44SnSPGRjxL R# \dRpLi;iL5i)z This will ask the current llm provider a suggestion for the current buffer. If there is a currently running llm task, it will cancel it. N5No LLM provider found, cannot trigger LLM completionsz;LLM Completion requires `jupyter_ai_magics` to be installedc<"SPV4GRjxL R#L \d+p\4PP RT4hRp?ii;i5i)zi This catches and log any errors, as otherwise this is just lost in the void of the future running task. Nzerror %s)_trigger_llm_core Exceptionrlogerror)rPer4s& r" error_catcherCNavigableAutoSuggestFromHistory._trigger_llm..error_catcherasI  ,,V444  !!'' A6 s0A" "A" A%AAA) jupyter_ai_magicsModuleNotFoundErrorrrrrasyncio create_taskr)r4rPrrsf& r" _trigger_llm,NavigableAutoSuggestFromHistory._trigger_llmNs % $!!! MMQ R   $ MMW X %%' !,,]6-BCnn/# % $  %. s9B/BB/A8B/B-B/ B*'B/)B**B/c <V^8dQhRS[/#rrPr )r!r0s"r"r#rpsFFfFr$c "^RIHuHpV'g \ R4hVP 'g \ R4hVP PPpVPV4p\4PPRV4V;P^, un VPpVPVWAPP ,VPP"RRRRRR7pVP P%V4RjxL pVPV8wdR#\'WrP(4'd\+VP,P.4^8d \ R 4h\1VP,P.^,P24VnVP6P94K\'WrP:4'gK\1VP<P24VnVP6P94EK \dRpEL3i;iEL'ED)R#5i) a This is the core of the current llm request. Here we build a compatible `InlineCompletionRequest` and ask the llm provider to stream it's response back to us iteratively setting it as the suggestion on the current buffer. Unlike with JupyterAi, as we do not have multiple cells, the cell id is always set to `None`. We set the prefix to the current cell content, but could also insert the rest of the history or even just the non-fail history. In the same way, we do not have cell id. LLM provider may return multiple suggestion stream, but for the time being we only support one. Here we make the assumption that the provider will have stream_inline_completions, I'm not sure it is the case for all providers. Nzjupyter-ai is not installedrz prefix: %sz text/x-pythonTpython)numberprefixsuffixmimestreampathlanguagecell_idzBTerminal IPython cannot deal with multiple LLM suggestions at once)jupyter_ai.completions.models completionsmodelsr ValueErrorrrshellhistory_manager _llm_prefixerrrdebugr~InlineCompletionRequestr text_before_cursortext_after_cursorstream_inline_completions isinstanceInlineCompletionReplyrTritemsr insertTextrQon_suggestion_setfireInlineCompletionStreamChunkresponse)r4rP jai_modelshmrrequest_numberrequestreply_and_chunkss&& r"r1NavigableAutoSuggestFromHistory._trigger_llm_coreps.  > >:; ;!!!TU U ^^ ! ! 1 1##B'  f5 !--44!OO>>>??44 5  '+&8&8&R&R '  0 0"##~5*,L,LMM',,223a7$\%/$))//2==%!((--/,.T.TUU$./?/H/H/S/S$T!((--/Y# J 6 0' $ s^I  H0I I C'I III!C I /AI 0 I=I II II c<V^8dQh/S[;R&S[S[,;R&S[PR,;R&S[R,;R&S[R,;R&S[;R&#)rrzr{Nrr}r|r)intrrrTaskr r )r!r0s"r"r#rshO-((||d")"!4'#& $J&'(5)r$)r{r}r|rr~rz)rirjrkrlrmrrr5rrrrrrrrrrrpropertyrrrrnrorp __classcell__)rr0s@@r"rrrrs&*I6M!JJ TT  !!''RKK     "" ++  DFFar$rrc$V^8dQhR\/#reventr)r!s"r"r#r#s66M6r$c"^p\4Pp\V\4'gR#VPP p\ ^WP, VP,4p\V4F!pVPPRRRR7K# VPVP4GRjxL R#L5i)z Ask the AutoSuggester from history to delegate to ask an LLM for completion This will first make sure that the current buffer have _MIN_LINES (7) available lines to insert the LLM completion Provisional as of 8.32, may change without warnings N F) move_cursor fire_event) r auto_suggestrrrcurrent_bufferr maxrMr(rW insert_textr)r _MIN_LINESproviderdoclines_to_insertrs& r"llm_autosuggestionr!sJ}))H h ? @ @    ' 'C!Z..83;R;RRSO ? # ((5U(S$    4 4 555sCC C C c$V^8dQhR\/#rr)r!s"r"r#r#s   r$cpVPpVPpVPVPRpVP R4pV^,P 4pVP pVe7VP'd%VR8XdVPVP4R#\P!V4R#)z,Apply autosuggestion or jump to end of line.Nrr<) rr r@cursor_positionsplitrZrQrnc end_of_line)rrPd after_cursorr'end_of_current_linerQs& r"accept_or_jump_to_endr+s  ! !FA66!++-.L   t $E(..*""JZ___;NRT;T:??+ ur$c$V^8dQhR\/#rr)r!s"r"r#r#s-r$cVPpVPpV'dVPVP4R#\P !V4R#)zAccept autosuggestionN)rrQrr@r& forward_char)rrPrQs& r"acceptr/s9  ! !F""J:??+ r$c$V^8dQhR\/#rr)r!s"r"r#r#s=r$c,VPpRVnR#)zDiscard autosuggestionN)rrQ)rrPs& r"discardr2s  ! !FFr$c$V^8dQhR\/#rr)r!s"r"r#r#s}r$cVPpVPpV'dF\P!RVP4pVP \ RV4R44R#\P!V4R#)z#Fill partial autosuggestion by wordz(\S+\s+)c38"TFq'gK VxK R#5ir&rE)rFrus& r"rHaccept_word..s 3AqAs  r<N) rrQrer%r@rnextr& forward_word)rrPrQts& r" accept_wordr;sV  ! !F""J HH[*// 24 3A 3R89 r$c$V^8dQhR\/#rr)r!s"r"r#r#s**M*r$cVPpVPpV'd9VP'd%VPVP^,4R#R#R#)z(Fill partial autosuggestion by characterN)rrQr@r)rbrQs& r"accept_characterr?s= AJjooo jooa()&zr$c$V^8dQhR\/#rr)r!s"r"r#r#s..-.r$cVPpVPpVPpV'd$VPVP4W!nR#R#)z.Accept autosuggestion and keep cursor in placeN)rr$rQrr@)rrP old_positionrQs& r"accept_and_keep_cursorrCsC  ! !F))L""J:??+!-r$c$V^8dQhR\/#rr)r!s"r"r#r# s}r$cH\V4\P!V4R#)z3Accept autosuggestion and move cursor left in placeN)rCr& backward_charrs&r"accept_and_move_cursor_leftrH s5!Ur$c$V^8dQhR\/#rr )r!s"r"r#r#s'''r$cVP'd.VPPWP4pWnR#R#r&)rrr rQ)rPrQs& r" _update_hintrKs3 ((77P &r$c$V^8dQhR\/#rr)r!s"r"r#r#s'']'r$c\\P!V4\VP4R#)z4Resume autosuggestions after deleting last characterN)r&backward_delete_charrKrrGs&r"backspace_and_resume_hintrOsE"%%&r$c$V^8dQhR\/#rr)r!s"r"r#r#s''-'r$c\\P!V4\VP4R#)zResume autosuggestionsN)rreplyrKrrGs&r"resume_hintingrSs u%%&r$c$V^8dQhR\/#rr)r!s"r"r#r#$s!!m!r$clVPpVPVPR7\V4R#)zGo up and update hintcountN)rauto_upargrKrrs& r"up_and_update_hintr[$s*))N+ r$c$V^8dQhR\/#rr)r!s"r"r#r#,s!! !r$clVPpVPVPR7\V4R#)zGo down and update hintrVN)r auto_downrYrKrZs& r"down_and_update_hintr_,s*))N599- r$c$V^8dQhR\/#rr)r!s"r"r#r#4s'' 'r$cVPpVPpV'Ed\VP4pW2P,p.ROpR.p^p\ \ V4P4FpVP\P8Xd \V4p M)VPV^,\VR,44p VPVRV 4VR,p V PV4'gKV^8XdE\V 4\V48d,V \V4RV^&VPV 4V^, pV^,WW&V^8XdM V^, pK V^,'dZVR,p V^,'d!\V^,4^8Xd VR,p V \V4Rp VPV 4R#\ P"!V4R#)z$Fill partial autosuggestion by tokenNr<)NNNr)rrQr)r r@rrreadlinetypetokenizeNEWLINErTindexrrrr&r9) rr>rQrr@tokens substringsratokenrgtokenized_so_farr to_inserts & r" accept_tokenrm4s AJzAJJ''&8T $Xd^%<%<=EzzX---D  58SB-@A   d6El +)"~ **6226c"23c&kA 0V ?F1I%%&67FA!!H 6Q>" !99$R.KayyS^q0(n #CKM2I MM) $ OOEr$c<V^8dQhR\R\R\/#)rrPrdirection_method)r rrr )r!s"r"r#r#as&'' '-''r$cVPpV'gR#\VP4pWCP,pV!WEVPR7VP WP4pW`nR#)a We skip most recent history entry (in either direction) if it equals the current autosuggestion because if user cycles when auto-suggestion is shown they most likely want something else than what was suggested (otherwise they would have accepted the suggestion). N)rrr)rQr)r r@rr)rPrrorQrcurrentnew_suggestions&&& r"_swap_autosuggestionrsasW""J  v 'Eoo%G5fnnM,,V__EN&r$c$V^8dQhR\/#rr)r!s"r"r#r#ys  - r$c\4pVPp\V\4'gR#\ VP W"P R7#)z%Get next autosuggestion from history.NrPrro)rrrrrrsrrrrrs& r"swap_autosuggestion_uprxysA ME!!H h ? @ @ ##h r$c$V^8dQhR\/#rr)r!s"r"r#r#s  M r$c\4pVPp\V\4'gR#\ VP VVP R7#)z)Get previous autosuggestion from history.Nrv)rrrrrrsrrrws& r"swap_autosuggestion_downr{sE ME!!H h ? @ @ ##! r$)@r7rreiortypingrrrrrr collections.abcr r rrUprompt_toolkit.bufferr prompt_toolkit.key_bindingr#prompt_toolkit.key_binding.bindingsrr&prompt_toolkit.auto_suggestrrprompt_toolkit.documentrprompt_toolkit.historyrprompt_toolkit.shortcutsr prompt_toolkit.layout.processorsrrrIPython.core.getipythonrIPython.utils.tokenutilrfiltersrr)r+rrr!r+r/r2r;r?rCrHrKrOrSr[r_rmProviderrsrxr{rEr$r"rs >>/(4DJ,*2 03!8|6I|6~V&<Vr6,  *. ' ' '!!'T ')H$N O'0  r$