eaypdZddlmZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZmZmZmZmZmZmZddlZd d lmZd d lmZd d lmZm Z d d l!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(ddl)m*Z*erddl+m,Z,ddl-m.Z.ddl/m0Z0dZ1dZ2dZ3dZ4dZ5dZ6dZ7ide1de1de1d e1d!e1d"e1d#e2d$e2d%e2d&e2d'e2d(e2d)e2d*e3d+e4d,e4d-did.d/d0d1d2d3d4d5d6d7d8d9d:e5d;e5de6d?d@dAdBdCdDdEdFdGe7dHe7dIdJdKZ8dLZ9GdMdNeZ:GdOdPe;Z<GdQdRe Z=dS)Sz} The feed module provides a high-level API for interacting with a list of `ChatMessage` objects through the backend methods. ) annotationsN)Enum)partial) isasyncgenisasyncgenfunction isawaitableiscoroutinefunction isgenerator)BytesIO) TYPE_CHECKINGAnyCallableClassVarDictListLiteral)Margin)CDN_DIST)Column ListPanel)Card)VSpacer)SVG)Button) ChatMessage)Document)Model)Commu🧑u🤖u⚙️u❌zuhttps://upload.wikimedia.org/wikipedia/commons/thumb/0/04/ChatGPT_logo.svg/1024px-ChatGPT_logo.svg.png?20230318122128z=https://upload.wikimedia.org/wikipedia/commons/a/a4/GPT-4.pngzvhttps://upload.wikimedia.org/wikipedia/commons/thumb/e/eb/WolframCorporateLogo.svg/1920px-WolframCorporateLogo.svg.pngclientcustomeremployeehumanpersonuseragentai assistantbotchatbotmachinerobotsystem exceptionerroradultbabyu👶boyu👦childu🧒girlu👧manu👨womanu👩chatgptgpt3gpt4dalleopenai huggingfaceu🤗 calculatoru🧮 langchainu🦜 translatoru🌐wolframz wolfram alphau🦙u🐪)llamallama2a c"eZdZdZdZdZdZdZdS) CallbackStateidlerunning generatingstoppingstoppedN)__name__ __module__ __qualname__IDLERUNNING GENERATINGSTOPPINGSTOPPED/lib/python3.11/site-packages/panel/chat/feed.pyrErEds' DGJHGGGrTrEceZdZdS) StopCallbackN)rKrLrMrSrTrUrWrWlsDrTrWcheZdZUdZejdddZejddZej d gd d Z ej d dZ ej idZejdddZejddZej idZejdZeddZejgdZej ddZejdddZejdZejd dd!Zejdd"Zejdd#Z ej!e"dd$%Z#ej e$e%d&'Z&ejdd(Z'e(d)gZ)d*e*d+<fd,Z+ d{d|d7Z,d}d>Z-d~dfd? Z.ej/d@dAdBZ0ej/dCddDdEZ1d~ddHZ2 dddOZ3 d~ddQZ4ddSZ5ddUZ6ddZZ7d[Z8dd\Z9d]Z: dddaZ; d{ddcZdddhZ?ddiZ@ dddsZA dddyZBd~dzZCxZDS)ChatFeedu A widget to display a list of `ChatMessage` objects and interact with them. This widget provides methods to: - Send (append) messages to the chat log. - Stream tokens to the latest `ChatMessage` in the chat log. - Execute callbacks when a user sends a message. - Undo a number of sent `ChatMessage` objects. - Clear the chat log of all `ChatMessage` objects. Reference: https://panel.holoviz.org/reference/chat/ChatFeed.html :Example: >>> async def say_welcome(contents, user, instance): >>> yield "Welcome!" >>> yield "Glad you're here!" >>> chat_feed = ChatFeed(callback=say_welcome, header="Welcome Feed") >>> chat_feed.send("Hello World!", user="New User", avatar="😊") )rNz Max pixel distance from the latest object in the Column to activate automatic scrolling upon update. Setting to 0 disables auto-scrolling.)defaultboundsdocFz Callback to execute when a user sends a message or when `respond` is called. The signature must include the previous message value `contents`, the previous `user` name, and the component `instance`.) allow_refsr]summary)raiser_verboseignorea/ How to handle exceptions raised by the callback. If "raise", the exception will be raised. If "summary", a summary will be sent to the chat feed. If "verbose", the full traceback will be sent to the chat feed. If "ignore", the exception will be ignored. )r[objectsr] AssistantzO The default user name to use for the message provided by the callback.)r[r]zY Params to pass to Card, like `header`, `header_background`, `header_color`, etc.Tz? Whether the Card should be expandable and collapsible.)r[readonlyr]z% Whether the feed is disabled.z Params to pass to each ChatMessage, like `reaction_icons`, `timestamp_format`, `show_avatar`, `show_user`, and `show_timestamp`. Params passed that are not ChatFeed params will be forwarded into `message_params`.zl The header of the chat feed; commonly used for the title. Can be a string, pane, or widget.)r]z Allows to create additional space around the component. May be specified as a two-tuple of the form (vertical, horizontal) or a four-tuple (top, right, bottom, left).z; The list of child objects that make up the layout.z] If placeholder is the default LoadingSpinner the text to display next to it.rz Min duration in seconds of buffering before displaying the placeholder. If 0, the placeholder will be disabled.a) A callable or list of callables that accept the value and return a Panel object to render the value. If a list is provided, will attempt to use the first renderer that does not raise an exception. If None, will attempt to infer the renderer from the value.dz Min pixel distance from the latest object in the Column to display the scroll button. Setting to 0 disables the scroll button.zj Whether to show an activity dot on the ChatMessage while streaming the callback response.zu Whether to scroll to the latest object on init. If not enabled the view will be on the first object.z~ The placeholder wrapped in a ChatMessage object; primarily to prevent recursion error in _update_placeholder.)class_r^r]z+ The current state of the callback.)rcr]z1 The previous disabled state of the feed.zcss/chat_feed.csszClassVar[List[str]] _stylesheetscd|_|dr't|dts |dg|d<|d|dd|d<|di}t|D]6}|t jvr&|tjvr||||<7||d<tj |i|t|jj |jj |jj|jj|jj|jj}t%||j|jdg|jd||_|}|j|jjz|jjd gz}||jj|jj|j|jj|jjj d d d g|jj!zd g|jj"|jj#dgddi| |j}|d d|||$|jddtK|jtMfi||_'tQd |_)|j)*|j+dS)N rendererswidth sizing_mode stretch_widthmessage_params)designrnrm max_width min_widthvisiblez chat-feed-log)auto_scroll_limitscroll_button_threshold css_classes stylesheetsrx)NrgFz chat-feedzchat-feed-headerzchat-feed-titlepadding0px) marginalignheaderheight hide_header collapsiblerwheader_css_classes max_height min_heighttitle_css_classesstylesrxrcT)rc bidirectional)rt),_callback_futureget isinstancelistkeysrYparamrpopsuper__init__dictrqrnrmrrrsrtrrurvrj _chat_logcopyrxrx card_paramsupdater{r|r}r~in_rwrrlinkrr_cardr_callback_triggeron_click_prepare_response) selfrcparamsrp param_key linked_paramsrcard_stylesheetscard_overrides __class__s rUrzChatFeed.__init__sG $ ::k " " 8:f[6I4+P+P 8#)+#6"7F;  ::g   &6::m+D+D+L$3F= ! $4b99fkkmm,, B BI..9 @Q3Q3Q,2JJy,A,Ay)#1 ',V,,,:$ .*"j*j*J&     "4$($@())      $((**   J " % % ' ' ( J " % % ' ' + +M2 > > ?  :$*";:$ ),,..155jAA$  (>(A(A(C(CC 23z,z,01u%(    )..00=$///>*** $.)4 HHH N II     "(!6!6!6 ''(>?????rTNr]rroot Model | Noneparentcomm Comm | Nonereturnrcv|j||||}|p|jd}||f|j|<|S)Nid)r _get_modelref_models)rr]rrrmodelrs rUrzChatFeed._get_model'sE %%c4>>}u!$'"FO S rTevents$Dict[str, param.parameterized.Event]msgDict[str, Any]rNonecdSNrS)rrrrrr]rs rU _update_modelzChatFeed._update_model0s  rTc~|j|t|dSr)r_cleanupr)rrrs rUrzChatFeed._cleanup6s7 D!!! rTr)watchc|j}|dd|jjjdi|dS)NrxrS)rrrrrr)rrs rU_update_card_paramszChatFeed._update_card_params:sN&++--  t,,, ..+.....rTplaceholder_text)ron_initc|j'|jj|jdSt t ddg}t |jdd|id|_dS)N)objectzrotating-placeholder)rnrw F)r&show_timestampavatarreaction_iconsshow_copy_icon) _placeholderrrrrPLACEHOLDER_SVGr)rloading_avatars rU_update_placeholderzChatFeed._update_placeholder@s   (   # * *$2G * H H H F CC eUUt~UATUU  *%)N6 "  .'-N8 $ : ;&)$*r/&:&:N7 #////rTr c|jtjk}|jduo|j}|dS|s|rt d|j}d}t|tr+| d|}| d}|Dt|tr|j }|j }|j }|||||St|tr|||St|tsd|i}||||}|||S)z| Replace the placeholder message with the response or update the message's value with the response. NCallback was stopped.r&rr&rr)_callback_staterErQr cancelledrW callback_userrrrrr&rrrrr)rrr is_stopping is_stoppedr&r new_messages rU_upsert_messagezChatFeed._upsert_messagesi*m.DD *$6\4;P;Z;Z;\;\ = F  8J 8677 7! eT " " )99VT**DYYx((F  %-- %z  NN5tFN ; ; ;N { + +   % %e , , ,L%&& &u%E))%d6)JJ  !!+...rTrc|j}t|dr|j}n2t|dr|j}nt|dr|j}n|}||j|fS)zH Extracts the contents from the message's panel object. rrcr) _object_panelhasattrrrcrr&)rrrcontentss rU_gather_callback_argszChatFeed._gather_callback_argssq% 5( # # |HH UI & & }HH UG $ $ {HHHt++rTresponsecKd} t|r?tj|_|23d{V}|||}|j|_*6nt |r9tj|_|D]$}|||}|j|_%nBt|r||d{V|}n|||}|rd|_n#|rd|_wxYw|S)zh Serializes the response by iterating over it and updating the message's value. NF)rrErPrrshow_activity_dotr r)rrresponse_messagetokens rU_serialize_responsezChatFeed._serialize_responsesn   ;(## T'4'?$#+PPPPPPP%'+';';ECS'T'T$9=9O$66$,8X&& T'4'?$%PPE'+';';ECS'T'T$9=9O$66PX&& T#'#7#7hHX#Y#Y  #'#7#7BR#S#S  ;5: 2  ;5: 2::::s"C(AB0C(( C3task asyncio.Task num_entriesrcK|jdkrdStj}|s|t |jkrtj|z }||jkr||jdStj dd{V|s|t |jkdSdSdSdS)z Schedules the placeholder to be added to the chat log if the callback takes longer than the placeholder threshold. rNg?) rasyncioget_event_looptimedonelenrrrsleep)rrrstartdurations rU_schedule_placeholderzChatFeed._schedule_placeholders  % * * F&((--//))++ %+T^1D1D"D"D-//4466>H$444 D-...-$$ $ $ $ $ $ $ $ ))++ %+T^1D1D"D"D"D"D % %"D"D % %rTcJK||}t|jr|j|d{V}nKt|jr |j|}n,|dt |jg|Rd{V}||d{VdSr)rr callbackrrun_in_executorrr)rrloop callback_argsrs rU_handle_callbackzChatFeed._handle_callbacks227;; t} - - *T]M:::::::HH  . . $t}m4HH!11gdm> = = $ '4'<$ = = = = = = = = = = = = = = =nR(Gg{33 6((** * * * * * * * * *3dn--K)++D%%d&;&;GT&J&JKKF$*D !.**6;??         9 9 9#0#8D    K???K&)33 UAUUU" (I55 E (<(>(>EEEUUUUUU(H44((** * * * * * * * * * ((** * * * * * * * * *$((** * * * * * * * *srD.A D.A!!D.$A!%%D.'BD.-H.G) H G)A)G$=HG$H$G))HH$cKtj|5|dtj|_|j|_ddddS#1swxYwYdS)zF Events to always execute after the callback is done. N) rrrrrErNrrrrs rUr zChatFeed._cleanup_response%s 4 4T : : / /  % %d + + +#0#5D  .DM / / / / / / / / / / / / / / / / / /s3A""A&)A&ChatMessage | dict | Anyrboolct|tr||td|}n1t|tsd|i}||||}|||r||S)a Sends a value and creates a new message in the chat log. If `respond` is `True`, additionally executes the callback, if provided. Arguments --------- value : ChatMessage | dict | Any The message contents to send. user : str | None The user to send as; overrides the message message's user if provided. avatar : str | bytes | BytesIO | None The avatar to use; overrides the message message's avatar if provided. respond : bool Whether to execute the callback. Returns ------- The message that was created. NzfCannot set user or avatar when explicitly sending a ChatMessage. Set them directly on the ChatMessage.rr)rrrrrrr)rrr&rrrs rUr z ChatFeed.send0s6 e[ ) ) K6#5 KGGeT** *!5)))%d6)JJG G   LLNNNrTstrc0|j(|jrtdt|tr||t d|r^t|t tfr(|||r||_ |r||_ n| ||||St|tr|}n1t|tsd|i}| |||}| ||S)a Streams a token and updates the provided message, if provided. Otherwise creates a new message in the chat log, so be sure the returned message is passed back into the method, e.g. `message = chat.stream(token, message=message)`. This method is primarily for outputs that are not generators-- notably LangChain. For most cases, use the send method instead. Arguments --------- value : str | dict | ChatMessage The new token value to stream. user : str | None The user to stream as; overrides the message's user if provided. avatar : str | bytes | BytesIO | None The avatar to use; overrides the message's avatar if provided. message : ChatMessage | None The message to update. Returns ------- The message that was updated. NrzhCannot set user or avatar when explicitly streaming a ChatMessage. Set them directly on the ChatMessage.rr)rrrWrrrrrstreamr&rrrr)rrr&rrs rUrzChatFeed.stream[s<>  ,1F1P1P1R1R ,677 7 e[ ) ) t/?6CUG  %#t-- @u%%%(#'GL,%+GNu4???N e[ ) ) KGGeT** *!5)))%d6)JJG !!'***rTcD|jjddS)zP Executes the callback with the latest message in the chat log. clicksN)rrtriggerrs rUrzChatFeed.responds$ $,,X66666rTc|jd}nB|jtjkrtj|_d}n|j}|r!|j|_|d|S)z Cancels the current callback task if possible. Returns ------- Whether the task was successfully stopped or done. NFT) rrrErPrQcancelrrr)rrs rUstopz ChatFeed.stopsy  (II  !]%= = =$1#9D II-4466I  , .DM  % %d + + +rTcount List[Any]cd|dkrgS|jj}|| d}|d| |dd<|S)a, Removes the last `count` of messages from the chat log and returns them. Parameters ---------- count : int The number of messages to remove, starting from the last message. Returns ------- The messages that were removed. rN)rrc)rr"messagesundone_entriess rUundoz ChatFeed.undosJ A::I>)!5&''*7UF7#QQQrTcP|jj}|j|S)z Clears the chat log and returns the messages that were cleared. Returns ------- The messages that were cleared. )rrcclear)rcleared_entriess rUr)zChatFeed.clears(.0 rTr)r%List[ChatMessage] role_names!Dict[str, str | List[str]] | None default_rolecustom_serializerrList[Dict[str, Any]]cN| dg|jgd}i}|D]9\}}t|tr|g}|D]}|||<:g} |D]} | j} | |vr|st d| jd|d|| |}|rF|| j} t| ts t dt| dnt| } | || d | S) zA Exports the chat log for use with transformers. Nr&)r&r)zUser z not found in role_names; got .zCThe provided custom_serializer must return a string; it returned a z type)rolecontent) ritemsrrlowerr&rrrtyper) rr%r,r.r/ names_roler3namesnameserialized_messagesrlowercase_namer4s rU_serialize_for_transformersz$ChatFeed._serialize_for_transformerss  "01J  %++-- 0 0KD%%%%  0 0+/ 4::<<(( 0! K KG$\//11NZ// / +GL++%+++ >>.,??D  '++GN;;!'3//$>)-g>>> g,,  & &'I'I J J J J""rT transformers filter_byCallable | NoneformatLiteral['transformers']c |jj}| ||}|dkr|j|fd|i|St d|d)a Exports the chat log. Arguments --------- format : str The format to export the chat log as; currently only supports "transformers". filter_by : callable A function to filter the chat log by. The function must accept and return a list of ChatMessage objects. custom_serializer : callable A custom function to format the ChatMessage's object. The function must accept one positional argument, the ChatMessage object, and return a string. If not provided, uses the serialize method on ChatMessage. **serialize_kwargs Additional keyword arguments to use for the specified format. - format="transformers" role_names : dict(str, str | list(str)) | None A dictionary mapping the role to the ChatMessage's user name. Defaults to `{"user": ["user"], "assistant": [self.callback_user]}` if not set. The keys and values are case insensitive as the strings will all be lowercased. The values can be a string or a list of strings, e.g. `{"user": "user", "assistant": ["executor", "langchain"]}`. default_role : str The default role to use if the user name is not found in role_names. If this is set to None, raises a ValueError if the user name is not found. Returns ------- The chat log serialized in the specified format. Nr>r/zFormat z is not supported.)rrcrr=NotImplementedError)rr?rAr/serialize_kwargsr%s rU serializezChatFeed.serializesP>)..00   y**H ^ # #343,=AQ ""HF"H"H"HIIIrTcg}|Tt|trt||s/t|r5t|ts ||r||||j|zS)a Iterates over the ChatInterface and any potential children in the applying the selector. Arguments --------- selector: type or callable or None The selector allows selecting a subset of Viewables by declaring a type or callable function to filter by. Returns ------- viewables: list(Viewable) )rr7callablerrselect)rselectorselecteds rUrIzChatFeed.select8s   $ ' ' ,6tX,F,F  h   (28T(B(B GOxPT~~  OOD ! ! !$*++H5555rT)NNN) r]rrrrrrrrr)rrrrrrrrr]rrrrrr)rrrr)rrrr)NN)rrr&rrrrr)rr rrrr)rrrr )rr rr)rrrrrr)rr)NNT) rrr&rrrrrrr) rrr&rrrrrrr)rr)r)r"rrr#)rr#)Nr)N) r%r+r,r-r.rr/rrr0)Nr>N)r?r@rArBr/r@)ErKrLrM__doc__rIntegerrurrObjectSelectorr StringrrrBooleanrrrp Parameterr}rr{rrcrNumberrHookListrlrvr view_latest ClassSelectorrrrrErrrrj__annotations__rrrrdependsrrrrrrrrrrr r rrr!r'r)r=rFrI __classcell__)rs@rUrYrYps,& c)J$&&& u~5)***H .-#L#L#LS    !EL;RSSSM%*R.\]]]K %-CBCCCKu}U1()))H UZ1QRRRN U_"-...FVA$7888F ej*>???G$u|B5)EL9K3444$I,emC O'((( & d9,--- %-39:::K'5&keRHIIIL+e*44 3F3FM.///O"EM%64555M-5)G)G)G(HLHHHHE@E@E@E@E@P379= U]=---//.-/ U]%T4@@@  A@ "!!!!!0 /3 <9=)))))V , , , ,    4%%%%( 1 1 1++++++++Z/// /3 )))))\ /3&* 99999v777 0(    9=#.&* 0#0#0#0#0#h&**8-1 0J0J0J0J0Jd66666666rTrY)>rL __future__rrrenumr functoolsrinspectrrrr r ior typingr r rrrrrr_paramr io.resourcesrlayoutrr layout.cardr layout.spacerr pane.imagerwidgets.buttonrrrbokeh.documentr bokeh.modelr pyviz_commsr USER_LOGOASSISTANT_LOGO SYSTEM_LOGO ERROR_LOGO GPT_3_LOGO GPT_4_LOGO WOLFRAM_LOGODEFAULT_AVATARSrrErrWrYrSrTrUrqs #""""" ######&&&&&&&&############ !''''''!!!!!!         E L H + i+ + +  Y +  i + I+ ^+ .++ >+~+~+ ^+" k#+$%+& Z'+* V+++, F-+. 6/+0 V1+2 F3+4 65+6 V7+:z;+< J=+> J?+@ ZA+B jC+D6E+F&G+HI+J&K+L|M+N\O++RU+++ZD     9   \ 6\ 6\ 6\ 6\ 6y\ 6\ 6\ 6\ 6\ 6rT