]j]8SSKrSSKrSSKrSSKJr SSKJrJrJrJ r J r J r SSK J r Jr SSKrSSKrSSKJr SSKJr SSKJr SSKJrJr SS KJr SS KJr SS KJ r SS K!J"r"J#r#J$r$ SS K%J&r& SSK'J(r( SSK)J*r* S\4Sjr+"SS\"5r,"SS\5r-S\4Sjr.S\4Sjr/S\4Sjr0S\4Sjr1S\4Sjr2S\4Sjr3S\4Sjr4S\4Sjr5S \4S!jr6S\4S"jr7S\4S#jr8S\4S$jr9S\4S%jr:S\4S&jr;\\\-S4rS\4S+jr?g),N)StringIO)ListOptionalUnionTupleClassVarAny)Callable Generator)Buffer) KeyPressEvent)named_commands)AutoSuggestFromHistory Suggestion)Document)History) PromptSession) ProcessorTransformationTransformationInput) get_ipython)generate_tokens) pass_throughdocumentc4URUR$N)linescursor_position_row)rs d/home/mande/repo/quber/.venv/lib/python3.13/site-packages/IPython/terminal/shortcuts/auto_suggest.py _get_queryr!s >>(66 77cV\rSrSr%SrSr\\\S'S S\ SS4Sjjr S \ S\ 4S jr S rg) AppendAutoSuggestionInAnyLine!a 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 F_debugstylereturnNcXlgrr')selfr's r __init__&AppendAutoSuggestionInAnyLine.__init__1s r"tic^^TRRS- nTRU:HnUU4SjnTRRS:XaU"S5$TRRU:XaU(aU"S5$TRTRR:aU"S5$TRR nUR (aTRR(dU"S5$TRTRR- nUR RR5n[U5S:XaU"S5$[RS :a[U5S:ab[RS :aNTRTRR:Xa[TRS /-S 9$[TRS 9$[U5S:XaJTRTRR:Xa&[TRTRUS4/-S 9$[TRS 9$US:Xa(USn[TRTRU4/-S 9$U(GaU[U5:a*S [U5U- S 3n[TRU4/5$[U5n [![U55HJn TR#X*- 5n SR%SU 55R'5n U (a O U S-n ML U (a[TRSU S34/5$[TR#U[U5- S-5TR((aTRS4/-5$/-5$U[U5:aXvn[TRU4/5$TR[U5- S-n [TR#U 55$)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. rcv>[TRTRTR(aSU-OS4/-S9$)N  fragments)rr4r'r&)textr+r.s r DAppendAutoSuggestionInAnyLine.apply_transformation..Ts/Nlltzz3:RT&U%VV r"z noop:onelineznoop:last line and cursorznoop:before cursorz noop:not eolrznoop: no suggestions)r1)redzD(Cannot show multiline suggestion; requires prompt_toolkit > 3.0.49)r3u… rest of suggestion (z lines) and code hiddenr2c3*# UH oSv M g7f)rN).0ls r EAppendAutoSuggestionInAnyLine.apply_transformation..s.2aqT2su… z line(s) hiddenzshift-last-line)r line_countlinenorbuffer_controlbuffer suggestionis_cursor_at_the_end_of_liner5 splitlineslenprompt_toolkitVERSIONrr4r'rangeget_linejoinstripr&)r+r.last_line_number is_last_linenooprDdelta suggestionsrEn_elidedillelshifts`` r apply_transformation2AppendAutoSuggestionInAnyLine.apply_transformation4s:;;11A5yy$44   ;; ! !Q &' ' ;; * *.> ><34 4 99r{{66 6,- -""))   (P(P' ' BKK;;;'',,779 { q ./ /  ! !J .;!#(>(>(K99 ? ??)"$,,#*BLLAA[!Q&99 ? ??)"$,,4::{1~2N1O"O"BLL9 9 A:$QJ!BLLTZZsgr" _llm_prefixerch>[TU]5 SUl/UlSUlSUlSUlgNr)superr,rgrhrkrj_request_number)r+ __class__s r r,(NavigableAutoSuggestFromHistory.__init__s5 !&*#"& r"_r(cSUlgrr)rg)r+rws r reset_history_position6NavigableAutoSuggestFromHistory.reset_history_positions r"cUR5 URH4nURRnUR UR 5 M6 gr)_cancel_running_llm_taskrhdefault_bufferon_text_insertremove_handlerry)r+pt_apptext_insert_events r disconnect*NavigableAutoSuggestFromHistory.disconnectsD %%'**F & 5 5 D D   , ,T-H-H I+r"rcURRU5 URRR UR 5 URR R UR5 gr)rhappendr}r~ add_handlerryon_cursor_position_changed_dismiss)r+rs r connect'NavigableAutoSuggestFromHistory.connectsV ##F+ ,,889T9TU88DDT]]Sr"rDrc[U5nUR5(a:URX0RUR5HupE[ U5s $ gr)r!rN_find_next_matchrghistoryr)r+rDrr5rErws r get_suggestion.NavigableAutoSuggestFromHistory.get_suggestionsO(# ::<> %~~""$$!%--/I!% V &r"cURc&UR(aUR5UlUR$)zLazy-initialized instance of the LLM provider. Do not use in the constructor, as `_init_llm_provider` can trigger slow side-effects. )rkrj)r+s r _llm_provider-NavigableAutoSuggestFromHistory._llm_providerDs5  & & .43J3J*.*A*A*CD '***r"cX^# SSKnTR(d[R"S5 gUc[R"S5 TR 5 U4Sjn[ R"U"U55TlTRIShvN g![a SnNf=fN7f)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. rN5No LLM provider found, cannot trigger LLM completionsz;LLM Completion requires `jupyter_ai_magics` to be installedc># TRU5IShvN gN![a*n[5RR SU5 eSnAff=f7f)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)rDer+s r error_catcherCNavigableAutoSuggestFromHistory._trigger_llm..error_catcherasI  ,,V444  !!'' A6 s0A!!A! A%AAA) jupyter_ai_magicsModuleNotFoundErrorrrrr|asyncio create_taskri)r+rDrrs` r _trigger_llm,NavigableAutoSuggestFromHistory._trigger_llmNs % $!! MMQ R   $ MMW X %%' !,,]6-BCnn/# % $  %. s4B*BBB*B(B* B%"B*$B%%B*c f# SSKJs Jn U(d [ S5eUR (d [ S5eUR RRnURU5n[5RRSU5 U=RS- sl URnURUXARR -URR"SSSS SS 9nUR R%U5ShvN nURU:wa g['XrR(5(a{[+UR,R.5S:a [ S 5e[1UR,R.SR25UlUR6R95 M['XrR:5(dM[1UR<R25UlUR6R95 GM ![a SnGNf=fGN g7f) 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. rNzjupyter-ai is not installedrz prefix: %srz text/x-pythonTpython)numberprefixsuffixmimestreampathlanguagecell_idzBTerminal IPython cannot deal with multiple LLM suggestions at once)jupyter_ai.completions.models completionsmodelsr ValueErrorrrshellhistory_managerrprrdebugrtInlineCompletionRequestrtext_before_cursortext_after_cursorstream_inline_completions isinstanceInlineCompletionReplyrHritemsr insertTextrEon_suggestion_setfireInlineCompletionStreamChunkresponse)r+rD 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"##~5*,L,LMM',,223a7$\%/$))//2==%!((--/,.T.TUU$./?/H/H/S/S$T!((--/Y# J 6 0' $ sTH1 HC?H1 H/H,H/CH1AH1 H)%H1(H))H1,H//H1)rhrjrkrirtrg)r(N)+r[r\r]r^r_intrarrrirTaskr r rpr,r ryrrrrrrrrbrrr`r rrrrrrr|propertyrrrrc __classcell__)rus@r reres O-(( &*Iw||d")!4'$J&5M85!4J TmTT  (0  *  !''%*'5<'HL' 5e$dD0 1'RKK%*K5<K 5e$dD0 1K   % '    w 4 "#3T" ++ DFfFFr"reeventc~# Sn[5Rn[U[5(dgURR n[ SXR- UR-5n[U5HnURRSSSS9 M URUR5IShvN gN7f)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 Nr F) move_cursor fire_event) r auto_suggestrrecurrent_bufferrmaxrArrK insert_textr)r _MIN_LINESproviderdoclines_to_insertrws r llm_autosuggestionrsJ}))H h ? @ @    ' 'C!Z..83;R;RRSO ? # ((5U(S$    4 4 555sB3B=5B;6B=c^URnURnURURSnUR S5nUSR 5nUR nUb3UR(a"US:XaURUR5 g[R"U5 g)z,Apply autosuggestion or jump to end of line.Nrrr2) rrr5cursor_positionsplitrNrErnc end_of_line)rrDd after_cursorrend_of_current_linerEs r accept_or_jump_to_endr s  ! !FA66!++-.L   t $E(..*""JZ__;NRT;T:??+ ur"cURnURnU(aURUR5 g[R "U5 g)zAccept autosuggestionN)rrErr5r forward_char)rrDrEs r acceptr s9  ! !F""J:??+ r"c*URnSUlg)zDiscard autosuggestionN)rrE)rrDs r discardrs  ! !FFr"cURnURnU(aD[R"SUR5nUR [ SU5S55 g[R"U5 g)z#Fill partial autosuggestion by wordz(\S+\s+)c36# UHo(dM Uv M g7frr<)r=rns r r?accept_word..s 3AqAs  r2N) rrErerr5rnextr forward_word)rrDrEts r accept_wordrsV  ! !F""J HH[*// 24 3A 3R89 r"cURnURnU(a1UR(aURURS5 ggg)z(Fill partial autosuggestion by characterrN)rrEr5r)rbrEs r accept_characterrs; AJjoo jooa()&zr"cURnURnURnU(a"URUR5 X!lgg)z.Accept autosuggestion and keep cursor in placeN)rrrErr5)rrD old_positionrEs r accept_and_keep_cursorrsC  ! !F))L""J:??+!-r"cF[U5 [R"U5 g)z3Accept autosuggestion and move cursor left in placeN)rr backward_charrs r accept_and_move_cursor_leftr! s5!Ur"rDc~UR(a,URRXR5nXlggr)rrrrE)rDrEs r _update_hintr#s0 ((77P &r"cZ[R"U5 [UR5 g)z4Resume autosuggestions after deleting last characterN)rbackward_delete_charr#rr s r backspace_and_resume_hintr&sE"%%&r"cZ[R"U5 [UR5 g)zResume autosuggestionsN)rreplyr#rr s r resume_hintingr)s u%%&r"cdURnURURS9 [U5 g)zGo up and update hintcountN)rauto_upargr#rrs r up_and_update_hintr0$s*))N+ r"cdURnURURS9 [U5 g)zGo down and update hintr+N)r auto_downr.r#r/s r down_and_update_hintr3,s*))N599- r"c\URnURnU(Gav[UR5nX2R-n/SQnS/nSn[ [ U5R5HnUR[R:Xa [U5n O!URUS[US55n URUSU 5 USn U RU5(dMUS:Xa?[U 5[U5:a'U [U5SUS'URU 5 US- nUSXW'US:Xa O US- nM US(aFUSn US(a[US5S:XaUSn U [U5Sn URU 5 g[ R""U5 g) z$Fill partial autosuggestion by token)NNNr2rrrN)rrEr!rr5rrreadlinetypetokenizeNEWLINErHindexrrrrr) rrrErr5tokens substringsrUtokenr;tokenized_so_farr to_inserts r accept_tokenrA4s AJAJJ''&8T $Xd^%<%<=EzzX---D  58SB-@A   d6El +)"~ **6226c"23c&kA 0V ?F1I%%&67FA!!H 6Q>" !9$R.KayS^q0(n #CKM2I MM) $ OOEr"rdirection_methodcURnU(dg[UR5nXCR-nU"XEURS9 UR XR5nX`lg)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)rEr!rr5rr)rDrrBrErcurrentnew_suggestions r _swap_autosuggestionrFasW""J  v 'Eoo%G5fnnM,,V__EN&r"c[5nURn[U[5(dg[ UR X"R S9$)z%Get next autosuggestion from history.NrDrrB)rrrrerFrrrrrs r swap_autosuggestion_uprJysA ME!!H h ? @ @ ##h r"c[5nURn[U[5(dg[ UR UUR S9$)z)Get previous autosuggestion from history.NrH)rrrrerFrrrIs r swap_autosuggestion_downrLsE ME!!H h ? @ @ ##! r")@rrr9iortypingrrrrrr collections.abcr r rrIprompt_toolkit.bufferr prompt_toolkit.key_bindingr #prompt_toolkit.key_binding.bindingsrrprompt_toolkit.auto_suggestrrprompt_toolkit.documentrprompt_toolkit.historyrprompt_toolkit.shortcutsr prompt_toolkit.layout.processorsrrrIPython.core.getipythonrIPython.utils.tokenutilrfiltersrr!r$rerr r rrrrr!r#r&r)r0r3rAProviderrFrJrLr<r"r r\sj >>/(4DJ,*2 03!88|6I|6~V&<Vr6M6,  -= }*M*.-.} '' ']' '-'!m!! !' 'T ')H$N O' '-''0 -  M r"