From 7daca39d732ca2ff5d7ee0f5374893f192fc30cc Mon Sep 17 00:00:00 2001 From: Simon Farshid Date: Sun, 14 Jul 2024 21:27:53 -0700 Subject: [PATCH] examples: playground (#502) --- examples/with-playground/.env.example | 1 + examples/with-playground/.eslintrc.json | 3 + examples/with-playground/.gitignore | 36 ++ .../with-playground/app/api/chat/route.ts | 12 + examples/with-playground/app/favicon.ico | Bin 0 -> 25931 bytes examples/with-playground/app/globals.css | 3 + examples/with-playground/app/layout.tsx | 23 + examples/with-playground/app/page.tsx | 54 ++ examples/with-playground/next.config.mjs | 4 + examples/with-playground/package.json | 32 ++ examples/with-playground/postcss.config.mjs | 8 + examples/with-playground/tailwind.config.ts | 18 + examples/with-playground/tsconfig.json | 21 + packages/react-playground/.eslintrc.json | 3 + packages/react-playground/README.md | 3 + packages/react-playground/components.json | 17 + packages/react-playground/package.json | 85 ++++ packages/react-playground/scripts/build.mts | 36 ++ .../src/components/payload-editor-button.tsx | 77 +++ .../ui/assistant-ui/assistant-playground.tsx | 344 +++++++++++++ .../src/components/ui/assistant-ui/image.tsx | 13 + .../ui/assistant-ui/json-editor.tsx | 23 + .../ui/assistant-ui/remove-content-part.tsx | 23 + .../src/components/ui/assistant-ui/text.tsx | 30 ++ .../src/components/ui/assistant-ui/thread.tsx | 320 ++++++++++++ .../components/ui/assistant-ui/tool-ui.tsx | 90 ++++ .../ui/assistant-ui/tooltip-icon-button.tsx | 44 ++ .../src/components/ui/avatar.tsx | 50 ++ .../src/components/ui/button.tsx | 57 +++ .../src/components/ui/dialog.tsx | 122 +++++ .../src/components/ui/dropdown-menu.tsx | 200 ++++++++ .../src/components/ui/input.tsx | 25 + .../src/components/ui/select.tsx | 160 ++++++ .../src/components/ui/slider.tsx | 28 + .../src/components/ui/textarea.tsx | 24 + .../src/components/ui/tooltip.tsx | 30 ++ packages/react-playground/src/index.ts | 3 + .../react-playground/src/lib/converters.ts | 95 ++++ .../src/lib/openai/fromOpenAIMessages.ts | 126 +++++ .../src/lib/openai/toOpenAIMessages.ts | 116 +++++ .../src/lib/openai/tryJsonParse.ts | 7 + .../src/lib/playground-runtime.ts | 477 ++++++++++++++++++ .../src/lib/usePlaygroundRuntime.tsx | 13 + packages/react-playground/src/lib/utils.ts | 6 + packages/react-playground/tailwind.config.ts | 58 +++ packages/react-playground/tsconfig.json | 11 + packages/react/package.json | 2 +- .../src/runtimes/edge/converters/index.ts | 2 + pnpm-lock.yaml | 159 ++++++ 49 files changed, 3093 insertions(+), 1 deletion(-) create mode 100644 examples/with-playground/.env.example create mode 100644 examples/with-playground/.eslintrc.json create mode 100644 examples/with-playground/.gitignore create mode 100644 examples/with-playground/app/api/chat/route.ts create mode 100644 examples/with-playground/app/favicon.ico create mode 100644 examples/with-playground/app/globals.css create mode 100644 examples/with-playground/app/layout.tsx create mode 100644 examples/with-playground/app/page.tsx create mode 100644 examples/with-playground/next.config.mjs create mode 100644 examples/with-playground/package.json create mode 100644 examples/with-playground/postcss.config.mjs create mode 100644 examples/with-playground/tailwind.config.ts create mode 100644 examples/with-playground/tsconfig.json create mode 100644 packages/react-playground/.eslintrc.json create mode 100644 packages/react-playground/README.md create mode 100644 packages/react-playground/components.json create mode 100644 packages/react-playground/package.json create mode 100644 packages/react-playground/scripts/build.mts create mode 100644 packages/react-playground/src/components/payload-editor-button.tsx create mode 100644 packages/react-playground/src/components/ui/assistant-ui/assistant-playground.tsx create mode 100644 packages/react-playground/src/components/ui/assistant-ui/image.tsx create mode 100644 packages/react-playground/src/components/ui/assistant-ui/json-editor.tsx create mode 100644 packages/react-playground/src/components/ui/assistant-ui/remove-content-part.tsx create mode 100644 packages/react-playground/src/components/ui/assistant-ui/text.tsx create mode 100644 packages/react-playground/src/components/ui/assistant-ui/thread.tsx create mode 100644 packages/react-playground/src/components/ui/assistant-ui/tool-ui.tsx create mode 100644 packages/react-playground/src/components/ui/assistant-ui/tooltip-icon-button.tsx create mode 100644 packages/react-playground/src/components/ui/avatar.tsx create mode 100644 packages/react-playground/src/components/ui/button.tsx create mode 100644 packages/react-playground/src/components/ui/dialog.tsx create mode 100644 packages/react-playground/src/components/ui/dropdown-menu.tsx create mode 100644 packages/react-playground/src/components/ui/input.tsx create mode 100644 packages/react-playground/src/components/ui/select.tsx create mode 100644 packages/react-playground/src/components/ui/slider.tsx create mode 100644 packages/react-playground/src/components/ui/textarea.tsx create mode 100644 packages/react-playground/src/components/ui/tooltip.tsx create mode 100644 packages/react-playground/src/index.ts create mode 100644 packages/react-playground/src/lib/converters.ts create mode 100644 packages/react-playground/src/lib/openai/fromOpenAIMessages.ts create mode 100644 packages/react-playground/src/lib/openai/toOpenAIMessages.ts create mode 100644 packages/react-playground/src/lib/openai/tryJsonParse.ts create mode 100644 packages/react-playground/src/lib/playground-runtime.ts create mode 100644 packages/react-playground/src/lib/usePlaygroundRuntime.tsx create mode 100644 packages/react-playground/src/lib/utils.ts create mode 100644 packages/react-playground/tailwind.config.ts create mode 100644 packages/react-playground/tsconfig.json diff --git a/examples/with-playground/.env.example b/examples/with-playground/.env.example new file mode 100644 index 000000000..dd29b38b9 --- /dev/null +++ b/examples/with-playground/.env.example @@ -0,0 +1 @@ +OPENAI_API_KEY=sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx diff --git a/examples/with-playground/.eslintrc.json b/examples/with-playground/.eslintrc.json new file mode 100644 index 000000000..bffb357a7 --- /dev/null +++ b/examples/with-playground/.eslintrc.json @@ -0,0 +1,3 @@ +{ + "extends": "next/core-web-vitals" +} diff --git a/examples/with-playground/.gitignore b/examples/with-playground/.gitignore new file mode 100644 index 000000000..fd3dbb571 --- /dev/null +++ b/examples/with-playground/.gitignore @@ -0,0 +1,36 @@ +# See https://help.github.com/articles/ignoring-files/ for more about ignoring files. + +# dependencies +/node_modules +/.pnp +.pnp.js +.yarn/install-state.gz + +# testing +/coverage + +# next.js +/.next/ +/out/ + +# production +/build + +# misc +.DS_Store +*.pem + +# debug +npm-debug.log* +yarn-debug.log* +yarn-error.log* + +# local env files +.env*.local + +# vercel +.vercel + +# typescript +*.tsbuildinfo +next-env.d.ts diff --git a/examples/with-playground/app/api/chat/route.ts b/examples/with-playground/app/api/chat/route.ts new file mode 100644 index 000000000..1ad9327ef --- /dev/null +++ b/examples/with-playground/app/api/chat/route.ts @@ -0,0 +1,12 @@ +import { createOpenAI } from "@ai-sdk/openai"; +import { createEdgeRuntimeAPI } from "@assistant-ui/react/edge"; + +export const maxDuration = 30; + +export const { POST } = createEdgeRuntimeAPI({ + model: ({ apiKey, modelName }) => { + if (!apiKey) throw new Error("apiKey is required"); + if (!modelName) throw new Error("modelName is required"); + return createOpenAI({ apiKey })(modelName as any); + }, +}); diff --git a/examples/with-playground/app/favicon.ico b/examples/with-playground/app/favicon.ico new file mode 100644 index 0000000000000000000000000000000000000000..718d6fea4835ec2d246af9800eddb7ffb276240c GIT binary patch literal 25931 zcmeHv30#a{`}aL_*G&7qml|y<+KVaDM2m#dVr!KsA!#An?kSQM(q<_dDNCpjEux83 zLb9Z^XxbDl(w>%i@8hT6>)&Gu{h#Oeyszu?xtw#Zb1mO{pgX9699l+Qppw7jXaYf~-84xW z)w4x8?=youko|}Vr~(D$UXIbiXABHh`p1?nn8Po~fxRJv}|0e(BPs|G`(TT%kKVJAdg5*Z|x0leQq0 zkdUBvb#>9F()jo|T~kx@OM8$9wzs~t2l;K=woNssA3l6|sx2r3+kdfVW@e^8e*E}v zA1y5{bRi+3Z`uD3{F7LgFJDdvm;nJilkzDku>BwXH(8ItVCXk*-lSJnR?-2UN%hJ){&rlvg`CDTj z)Bzo!3v7Ou#83zEDEFcKt(f1E0~=rqeEbTnMvWR#{+9pg%7G8y>u1OVRUSoox-ovF z2Ydma(;=YuBY(eI|04{hXzZD6_f(v~H;C~y5=DhAC{MMS>2fm~1H_t2$56pc$NH8( z5bH|<)71dV-_oCHIrzrT`2s-5w_+2CM0$95I6X8p^r!gHp+j_gd;9O<1~CEQQGS8) zS9Qh3#p&JM-G8rHekNmKVewU;pJRcTAog68KYo^dRo}(M>36U4Us zfgYWSiHZL3;lpWT=zNAW>Dh#mB!_@Lg%$ms8N-;aPqMn+C2HqZgz&9~Eu z4|Kp<`$q)Uw1R?y(~S>ePdonHxpV1#eSP1B;Ogo+-Pk}6#0GsZZ5!||ev2MGdh}_m z{DeR7?0-1^zVs&`AV6Vt;r3`I`OI_wgs*w=eO%_#7Kepl{B@xiyCANc(l zzIyd4y|c6PXWq9-|KM8(zIk8LPk(>a)zyFWjhT!$HJ$qX1vo@d25W<fvZQ2zUz5WRc(UnFMKHwe1| zWmlB1qdbiA(C0jmnV<}GfbKtmcu^2*P^O?MBLZKt|As~ge8&AAO~2K@zbXelK|4T<{|y4`raF{=72kC2Kn(L4YyenWgrPiv z@^mr$t{#X5VuIMeL!7Ab6_kG$&#&5p*Z{+?5U|TZ`B!7llpVmp@skYz&n^8QfPJzL z0G6K_OJM9x+Wu2gfN45phANGt{7=C>i34CV{Xqlx(fWpeAoj^N0Biu`w+MVcCUyU* zDZuzO0>4Z6fbu^T_arWW5n!E45vX8N=bxTVeFoep_G#VmNlQzAI_KTIc{6>c+04vr zx@W}zE5JNSU>!THJ{J=cqjz+4{L4A{Ob9$ZJ*S1?Ggg3klFp!+Y1@K+pK1DqI|_gq z5ZDXVpge8-cs!o|;K73#YXZ3AShj50wBvuq3NTOZ`M&qtjj#GOFfgExjg8Gn8>Vq5 z`85n+9|!iLCZF5$HJ$Iu($dm?8~-ofu}tEc+-pyke=3!im#6pk_Wo8IA|fJwD&~~F zc16osQ)EBo58U7XDuMexaPRjU@h8tXe%S{fA0NH3vGJFhuyyO!Uyl2^&EOpX{9As0 zWj+P>{@}jxH)8|r;2HdupP!vie{sJ28b&bo!8`D^x}TE$%zXNb^X1p@0PJ86`dZyj z%ce7*{^oo+6%&~I!8hQy-vQ7E)0t0ybH4l%KltWOo~8cO`T=157JqL(oq_rC%ea&4 z2NcTJe-HgFjNg-gZ$6!Y`SMHrlj}Etf7?r!zQTPPSv}{so2e>Fjs1{gzk~LGeesX%r(Lh6rbhSo_n)@@G-FTQy93;l#E)hgP@d_SGvyCp0~o(Y;Ee8{ zdVUDbHm5`2taPUOY^MAGOw*>=s7=Gst=D+p+2yON!0%Hk` zz5mAhyT4lS*T3LS^WSxUy86q&GnoHxzQ6vm8)VS}_zuqG?+3td68_x;etQAdu@sc6 zQJ&5|4(I?~3d-QOAODHpZ=hlSg(lBZ!JZWCtHHSj`0Wh93-Uk)_S%zsJ~aD>{`A0~ z9{AG(e|q3g5B%wYKRxiL2Y$8(4w6bzchKuloQW#e&S3n+P- z8!ds-%f;TJ1>)v)##>gd{PdS2Oc3VaR`fr=`O8QIO(6(N!A?pr5C#6fc~Ge@N%Vvu zaoAX2&(a6eWy_q&UwOhU)|P3J0Qc%OdhzW=F4D|pt0E4osw;%<%Dn58hAWD^XnZD= z>9~H(3bmLtxpF?a7su6J7M*x1By7YSUbxGi)Ot0P77`}P3{)&5Un{KD?`-e?r21!4vTTnN(4Y6Lin?UkSM z`MXCTC1@4A4~mvz%Rh2&EwY))LeoT=*`tMoqcEXI>TZU9WTP#l?uFv+@Dn~b(>xh2 z;>B?;Tz2SR&KVb>vGiBSB`@U7VIWFSo=LDSb9F{GF^DbmWAfpms8Sx9OX4CnBJca3 zlj9(x!dIjN?OG1X4l*imJNvRCk}F%!?SOfiOq5y^mZW)jFL@a|r-@d#f7 z2gmU8L3IZq0ynIws=}~m^#@&C%J6QFo~Mo4V`>v7MI-_!EBMMtb%_M&kvAaN)@ZVw z+`toz&WG#HkWDjnZE!6nk{e-oFdL^$YnbOCN}JC&{$#$O27@|Tn-skXr)2ml2~O!5 zX+gYoxhoc7qoU?C^3~&!U?kRFtnSEecWuH0B0OvLodgUAi}8p1 zrO6RSXHH}DMc$&|?D004DiOVMHV8kXCP@7NKB zgaZq^^O<7PoKEp72kby@W0Z!Y*Ay{&vfg#C&gG@YVR9g?FEocMUi1gSN$+V+ayF45{a zuDZDTN}mS|;BO%gEf}pjBfN2-gIrU#G5~cucA;dokXW89%>AyXJJI z9X4UlIWA|ZYHgbI z5?oFk@A=Ik7lrEQPDH!H+b`7_Y~aDb_qa=B2^Y&Ow41cU=4WDd40dp5(QS-WMN-=Y z9g;6_-JdNU;|6cPwf$ak*aJIcwL@1n$#l~zi{c{EW?T;DaW*E8DYq?Umtz{nJ&w-M zEMyTDrC&9K$d|kZe2#ws6)L=7K+{ zQw{XnV6UC$6-rW0emqm8wJoeZK)wJIcV?dST}Z;G0Arq{dVDu0&4kd%N!3F1*;*pW zR&qUiFzK=@44#QGw7k1`3t_d8&*kBV->O##t|tonFc2YWrL7_eqg+=+k;!F-`^b8> z#KWCE8%u4k@EprxqiV$VmmtiWxDLgnGu$Vs<8rppV5EajBXL4nyyZM$SWVm!wnCj-B!Wjqj5-5dNXukI2$$|Bu3Lrw}z65Lc=1G z^-#WuQOj$hwNGG?*CM_TO8Bg-1+qc>J7k5c51U8g?ZU5n?HYor;~JIjoWH-G>AoUP ztrWWLbRNqIjW#RT*WqZgPJXU7C)VaW5}MiijYbABmzoru6EmQ*N8cVK7a3|aOB#O& zBl8JY2WKfmj;h#Q!pN%9o@VNLv{OUL?rixHwOZuvX7{IJ{(EdPpuVFoQqIOa7giLVkBOKL@^smUA!tZ1CKRK}#SSM)iQHk)*R~?M!qkCruaS!#oIL1c z?J;U~&FfH#*98^G?i}pA{ z9Jg36t4=%6mhY(quYq*vSxptes9qy|7xSlH?G=S@>u>Ebe;|LVhs~@+06N<4CViBk zUiY$thvX;>Tby6z9Y1edAMQaiH zm^r3v#$Q#2T=X>bsY#D%s!bhs^M9PMAcHbCc0FMHV{u-dwlL;a1eJ63v5U*?Q_8JO zT#50!RD619#j_Uf))0ooADz~*9&lN!bBDRUgE>Vud-i5ck%vT=r^yD*^?Mp@Q^v+V zG#-?gKlr}Eeqifb{|So?HM&g91P8|av8hQoCmQXkd?7wIJwb z_^v8bbg`SAn{I*4bH$u(RZ6*xUhuA~hc=8czK8SHEKTzSxgbwi~9(OqJB&gwb^l4+m`k*Q;_?>Y-APi1{k zAHQ)P)G)f|AyjSgcCFps)Fh6Bca*Xznq36!pV6Az&m{O8$wGFD? zY&O*3*J0;_EqM#jh6^gMQKpXV?#1?>$ml1xvh8nSN>-?H=V;nJIwB07YX$e6vLxH( zqYwQ>qxwR(i4f)DLd)-$P>T-no_c!LsN@)8`e;W@)-Hj0>nJ-}Kla4-ZdPJzI&Mce zv)V_j;(3ERN3_@I$N<^|4Lf`B;8n+bX@bHbcZTopEmDI*Jfl)-pFDvo6svPRoo@(x z);_{lY<;);XzT`dBFpRmGrr}z5u1=pC^S-{ce6iXQlLGcItwJ^mZx{m$&DA_oEZ)B{_bYPq-HA zcH8WGoBG(aBU_j)vEy+_71T34@4dmSg!|M8Vf92Zj6WH7Q7t#OHQqWgFE3ARt+%!T z?oLovLVlnf?2c7pTc)~cc^($_8nyKwsN`RA-23ed3sdj(ys%pjjM+9JrctL;dy8a( z@en&CQmnV(()bu|Y%G1-4a(6x{aLytn$T-;(&{QIJB9vMox11U-1HpD@d(QkaJdEb zG{)+6Dos_L+O3NpWo^=gR?evp|CqEG?L&Ut#D*KLaRFOgOEK(Kq1@!EGcTfo+%A&I z=dLbB+d$u{sh?u)xP{PF8L%;YPPW53+@{>5W=Jt#wQpN;0_HYdw1{ksf_XhO4#2F= zyPx6Lx2<92L-;L5PD`zn6zwIH`Jk($?Qw({erA$^bC;q33hv!d!>%wRhj# zal^hk+WGNg;rJtb-EB(?czvOM=H7dl=vblBwAv>}%1@{}mnpUznfq1cE^sgsL0*4I zJ##!*B?=vI_OEVis5o+_IwMIRrpQyT_Sq~ZU%oY7c5JMIADzpD!Upz9h@iWg_>>~j zOLS;wp^i$-E?4<_cp?RiS%Rd?i;f*mOz=~(&3lo<=@(nR!_Rqiprh@weZlL!t#NCc zO!QTcInq|%#>OVgobj{~ixEUec`E25zJ~*DofsQdzIa@5^nOXj2T;8O`l--(QyU^$t?TGY^7#&FQ+2SS3B#qK*k3`ye?8jUYSajE5iBbJls75CCc(m3dk{t?- zopcER9{Z?TC)mk~gpi^kbbu>b-+a{m#8-y2^p$ka4n60w;Sc2}HMf<8JUvhCL0B&Btk)T`ctE$*qNW8L$`7!r^9T+>=<=2qaq-;ll2{`{Rg zc5a0ZUI$oG&j-qVOuKa=*v4aY#IsoM+1|c4Z)<}lEDvy;5huB@1RJPquU2U*U-;gu z=En2m+qjBzR#DEJDO`WU)hdd{Vj%^0V*KoyZ|5lzV87&g_j~NCjwv0uQVqXOb*QrQ zy|Qn`hxx(58c70$E;L(X0uZZ72M1!6oeg)(cdKO ze0gDaTz+ohR-#d)NbAH4x{I(21yjwvBQfmpLu$)|m{XolbgF!pmsqJ#D}(ylp6uC> z{bqtcI#hT#HW=wl7>p!38sKsJ`r8}lt-q%Keqy%u(xk=yiIJiUw6|5IvkS+#?JTBl z8H5(Q?l#wzazujH!8o>1xtn8#_w+397*_cy8!pQGP%K(Ga3pAjsaTbbXJlQF_+m+-UpUUent@xM zg%jqLUExj~o^vQ3Gl*>wh=_gOr2*|U64_iXb+-111aH}$TjeajM+I20xw(((>fej-@CIz4S1pi$(#}P7`4({6QS2CaQS4NPENDp>sAqD z$bH4KGzXGffkJ7R>V>)>tC)uax{UsN*dbeNC*v}#8Y#OWYwL4t$ePR?VTyIs!wea+ z5Urmc)X|^`MG~*dS6pGSbU+gPJoq*^a=_>$n4|P^w$sMBBy@f*Z^Jg6?n5?oId6f{ z$LW4M|4m502z0t7g<#Bx%X;9<=)smFolV&(V^(7Cv2-sxbxopQ!)*#ZRhTBpx1)Fc zNm1T%bONzv6@#|dz(w02AH8OXe>kQ#1FMCzO}2J_mST)+ExmBr9cva-@?;wnmWMOk z{3_~EX_xadgJGv&H@zK_8{(x84`}+c?oSBX*Ge3VdfTt&F}yCpFP?CpW+BE^cWY0^ zb&uBN!Ja3UzYHK-CTyA5=L zEMW{l3Usky#ly=7px648W31UNV@K)&Ub&zP1c7%)`{);I4b0Q<)B}3;NMG2JH=X$U zfIW4)4n9ZM`-yRj67I)YSLDK)qfUJ_ij}a#aZN~9EXrh8eZY2&=uY%2N0UFF7<~%M zsB8=erOWZ>Ct_#^tHZ|*q`H;A)5;ycw*IcmVxi8_0Xk}aJA^ath+E;xg!x+As(M#0=)3!NJR6H&9+zd#iP(m0PIW8$ z1Y^VX`>jm`W!=WpF*{ioM?C9`yOR>@0q=u7o>BP-eSHqCgMDj!2anwH?s%i2p+Q7D zzszIf5XJpE)IG4;d_(La-xenmF(tgAxK`Y4sQ}BSJEPs6N_U2vI{8=0C_F?@7<(G; zo$~G=8p+076G;`}>{MQ>t>7cm=zGtfbdDXm6||jUU|?X?CaE?(<6bKDYKeHlz}DA8 zXT={X=yp_R;HfJ9h%?eWvQ!dRgz&Su*JfNt!Wu>|XfU&68iRikRrHRW|ZxzRR^`eIGt zIeiDgVS>IeExKVRWW8-=A=yA`}`)ZkWBrZD`hpWIxBGkh&f#ijr449~m`j6{4jiJ*C!oVA8ZC?$1RM#K(_b zL9TW)kN*Y4%^-qPpMP7d4)o?Nk#>aoYHT(*g)qmRUb?**F@pnNiy6Fv9rEiUqD(^O zzyS?nBrX63BTRYduaG(0VVG2yJRe%o&rVrLjbxTaAFTd8s;<<@Qs>u(<193R8>}2_ zuwp{7;H2a*X7_jryzriZXMg?bTuegABb^87@SsKkr2)0Gyiax8KQWstw^v#ix45EVrcEhr>!NMhprl$InQMzjSFH54x5k9qHc`@9uKQzvL4ihcq{^B zPrVR=o_ic%Y>6&rMN)hTZsI7I<3&`#(nl+3y3ys9A~&^=4?PL&nd8)`OfG#n zwAMN$1&>K++c{^|7<4P=2y(B{jJsQ0a#U;HTo4ZmWZYvI{+s;Td{Yzem%0*k#)vjpB zia;J&>}ICate44SFYY3vEelqStQWFihx%^vQ@Do(sOy7yR2@WNv7Y9I^yL=nZr3mb zXKV5t@=?-Sk|b{XMhA7ZGB@2hqsx}4xwCW!in#C zI@}scZlr3-NFJ@NFaJlhyfcw{k^vvtGl`N9xSo**rDW4S}i zM9{fMPWo%4wYDG~BZ18BD+}h|GQKc-g^{++3MY>}W_uq7jGHx{mwE9fZiPCoxN$+7 zrODGGJrOkcPQUB(FD5aoS4g~7#6NR^ma7-!>mHuJfY5kTe6PpNNKC9GGRiu^L31uG z$7v`*JknQHsYB!Tm_W{a32TM099djW%5e+j0Ve_ct}IM>XLF1Ap+YvcrLV=|CKo6S zb+9Nl3_YdKP6%Cxy@6TxZ>;4&nTneadr z_ES90ydCev)LV!dN=#(*f}|ZORFdvkYBni^aLbUk>BajeWIOcmHP#8S)*2U~QKI%S zyrLmtPqb&TphJ;>yAxri#;{uyk`JJqODDw%(Z=2`1uc}br^V%>j!gS)D*q*f_-qf8&D;W1dJgQMlaH5er zN2U<%Smb7==vE}dDI8K7cKz!vs^73o9f>2sgiTzWcwY|BMYHH5%Vn7#kiw&eItCqa zIkR2~Q}>X=Ar8W|^Ms41Fm8o6IB2_j60eOeBB1Br!boW7JnoeX6Gs)?7rW0^5psc- zjS16yb>dFn>KPOF;imD}e!enuIniFzv}n$m2#gCCv4jM#ArwlzZ$7@9&XkFxZ4n!V zj3dyiwW4Ki2QG{@i>yuZXQizw_OkZI^-3otXC{!(lUpJF33gI60ak;Uqitp74|B6I zgg{b=Iz}WkhCGj1M=hu4#Aw173YxIVbISaoc z-nLZC*6Tgivd5V`K%GxhBsp@SUU60-rfc$=wb>zdJzXS&-5(NRRodFk;Kxk!S(O(a0e7oY=E( zAyS;Ow?6Q&XA+cnkCb{28_1N8H#?J!*$MmIwLq^*T_9-z^&UE@A(z9oGYtFy6EZef LrJugUA?W`A8`#=m literal 0 HcmV?d00001 diff --git a/examples/with-playground/app/globals.css b/examples/with-playground/app/globals.css new file mode 100644 index 000000000..b5c61c956 --- /dev/null +++ b/examples/with-playground/app/globals.css @@ -0,0 +1,3 @@ +@tailwind base; +@tailwind components; +@tailwind utilities; diff --git a/examples/with-playground/app/layout.tsx b/examples/with-playground/app/layout.tsx new file mode 100644 index 000000000..7acf2f2a9 --- /dev/null +++ b/examples/with-playground/app/layout.tsx @@ -0,0 +1,23 @@ +import type { Metadata } from "next"; +import { Inter } from "next/font/google"; + +import "./globals.css"; + +const inter = Inter({ subsets: ["latin"] }); + +export const metadata: Metadata = { + title: "Create Next App", + description: "Generated by create next app", +}; + +export default function RootLayout({ + children, +}: Readonly<{ + children: React.ReactNode; +}>) { + return ( + + {children} + + ); +} diff --git a/examples/with-playground/app/page.tsx b/examples/with-playground/app/page.tsx new file mode 100644 index 000000000..602a70d21 --- /dev/null +++ b/examples/with-playground/app/page.tsx @@ -0,0 +1,54 @@ +"use client"; + +import { + AssistantPlayground, + requestOptionsFromOpenAI, +} from "@assistant-ui/react-playground"; +import { usePlaygroundRuntime } from "@assistant-ui/react-playground"; +import { AssistantRuntimeProvider } from "@assistant-ui/react"; + +export default function Home() { + const runtime = usePlaygroundRuntime({ + api: "/api/chat", + initialMessages: [], + }); + + const handleLoadTestData = () => { + runtime.thread.setRequestData( + requestOptionsFromOpenAI({ + model: "gpt-4o", + messages: [ + { + role: "user", + content: "Write a poem about the weather", + }, + { + role: "assistant", + content: "The weather is sunny and warm.", + }, + ], + }), + ); + }; + + return ( + +
+
+

Playground

+ +
+
+
+ +
+
+
+ + ); +} diff --git a/examples/with-playground/next.config.mjs b/examples/with-playground/next.config.mjs new file mode 100644 index 000000000..4678774e6 --- /dev/null +++ b/examples/with-playground/next.config.mjs @@ -0,0 +1,4 @@ +/** @type {import('next').NextConfig} */ +const nextConfig = {}; + +export default nextConfig; diff --git a/examples/with-playground/package.json b/examples/with-playground/package.json new file mode 100644 index 000000000..2c94bc2be --- /dev/null +++ b/examples/with-playground/package.json @@ -0,0 +1,32 @@ +{ + "name": "with-playground", + "version": "0.1.0", + "private": true, + "scripts": { + "dev": "next dev", + "build": "next build", + "start": "next start", + "lint": "next lint" + }, + "dependencies": { + "@ai-sdk/openai": "^0.0.36", + "@assistant-ui/react": "workspace:*", + "@assistant-ui/react-playground": "workspace:*", + "next": "14.2.5", + "react": "^18", + "react-dom": "^18", + "tailwindcss-animate": "^1.0.7" + }, + "devDependencies": { + "@assistant-ui/tsconfig": "workspace:^", + "@types/node": "^20", + "@types/react": "^18", + "@types/react-dom": "^18", + "eslint": "^8", + "eslint-config-next": "14.2.5", + "postcss": "^8", + "tailwindcss": "^3.4.4", + "typescript": "^5" + }, + "packageManager": "pnpm@9.4.0+sha512.f549b8a52c9d2b8536762f99c0722205efc5af913e77835dbccc3b0b0b2ca9e7dc8022b78062c17291c48e88749c70ce88eb5a74f1fa8c4bf5e18bb46c8bd83a" +} diff --git a/examples/with-playground/postcss.config.mjs b/examples/with-playground/postcss.config.mjs new file mode 100644 index 000000000..1a69fd2a4 --- /dev/null +++ b/examples/with-playground/postcss.config.mjs @@ -0,0 +1,8 @@ +/** @type {import('postcss-load-config').Config} */ +const config = { + plugins: { + tailwindcss: {}, + }, +}; + +export default config; diff --git a/examples/with-playground/tailwind.config.ts b/examples/with-playground/tailwind.config.ts new file mode 100644 index 000000000..496a220c0 --- /dev/null +++ b/examples/with-playground/tailwind.config.ts @@ -0,0 +1,18 @@ +import type { Config } from "tailwindcss"; + +const config = { + darkMode: ["class"], + content: [ + "./pages/**/*.{ts,tsx}", + "./components/**/*.{ts,tsx}", + "./app/**/*.{ts,tsx}", + "./src/**/*.{ts,tsx}", + "./node_modules/@assistant-ui/react-playground/{dist,src}/**/*.{js,ts,tsx}", + ], + plugins: [ + require("tailwindcss-animate"), + require("@assistant-ui/react/tailwindcss")({ components: ["base"] }), + ], +} satisfies Config; + +export default config; diff --git a/examples/with-playground/tsconfig.json b/examples/with-playground/tsconfig.json new file mode 100644 index 000000000..b764a9017 --- /dev/null +++ b/examples/with-playground/tsconfig.json @@ -0,0 +1,21 @@ +{ + "extends": "@assistant-ui/tsconfig/base.json", + "compilerOptions": { + "target": "ES6", + "incremental": true, + "plugins": [ + { + "name": "next" + } + ], + "paths": { + "@/*": ["./*"], + + }, + "allowJs": true, + "strictNullChecks": true, + "jsx": "preserve" + }, + "include": ["next-env.d.ts", "**/*.ts", "**/*.tsx", ".next/types/**/*.ts"], + "exclude": ["node_modules"] +} diff --git a/packages/react-playground/.eslintrc.json b/packages/react-playground/.eslintrc.json new file mode 100644 index 000000000..bffb357a7 --- /dev/null +++ b/packages/react-playground/.eslintrc.json @@ -0,0 +1,3 @@ +{ + "extends": "next/core-web-vitals" +} diff --git a/packages/react-playground/README.md b/packages/react-playground/README.md new file mode 100644 index 000000000..9b99bdb9e --- /dev/null +++ b/packages/react-playground/README.md @@ -0,0 +1,3 @@ +# `@assistant-ui/react-playground` + +A playground runtime for `@assistant-ui/react`. diff --git a/packages/react-playground/components.json b/packages/react-playground/components.json new file mode 100644 index 000000000..2d0f0b345 --- /dev/null +++ b/packages/react-playground/components.json @@ -0,0 +1,17 @@ +{ + "$schema": "https://ui.shadcn.com/schema.json", + "style": "default", + "rsc": true, + "tsx": true, + "tailwind": { + "config": "tailwind.config.ts", + "css": "app/globals.css", + "baseColor": "zinc", + "cssVariables": true, + "prefix": "" + }, + "aliases": { + "components": "@/components", + "utils": "@/lib/utils" + } +} \ No newline at end of file diff --git a/packages/react-playground/package.json b/packages/react-playground/package.json new file mode 100644 index 000000000..6e5633e82 --- /dev/null +++ b/packages/react-playground/package.json @@ -0,0 +1,85 @@ +{ + "name": "@assistant-ui/react-playground", + "version": "0.0.0-alpha.0", + "license": "MIT", + "exports": { + ".": { + "import": { + "types": "./dist/index.d.mts", + "default": "./dist/index.mjs" + }, + "require": { + "types": "./dist/index.d.ts", + "default": "./dist/index.js" + } + } + }, + "source": "./src/index.ts", + "main": "./dist/index.js", + "module": "./dist/index.mjs", + "types": "./dist/index.d.ts", + "files": [ + "dist", + "README.md" + ], + "sideEffects": false, + "scripts": { + "build": "tsx scripts/build.mts" + }, + "dependencies": { + "@ai-sdk/provider": "^0.0.12", + "@radix-ui/react-avatar": "^1.1.0", + "@radix-ui/react-dialog": "^1.1.1", + "@radix-ui/react-dropdown-menu": "^2.1.1", + "@radix-ui/react-select": "^2.1.1", + "@radix-ui/react-slider": "^1.2.0", + "@radix-ui/react-slot": "^1.1.0", + "@radix-ui/react-tooltip": "^1.1.2", + "class-variance-authority": "^0.7.0", + "classnames": "^2.5.1", + "clsx": "^2.1.1", + "lucide-react": "^0.407.0", + "openai": "^4.52.7", + "prismjs": "^1.29.0", + "react-simple-code-editor": "^0.14.1", + "react-textarea-autosize": "^8.5.3", + "tailwind-merge": "^2.4.0", + "zustand": "^4.5.4" + }, + "peerDependencies": { + "@assistant-ui/react": "^0.4.4", + "@types/react": "*", + "react": "^18", + "tailwindcss": "^3.4.4" + }, + "peerDependenciesMeta": { + "@types/react": { + "optional": true + }, + "tailwindcss": { + "optional": true + } + }, + "devDependencies": { + "@assistant-ui/tsconfig": "workspace:*", + "@types/prismjs": "^1.26.4", + "eslint": "^8", + "eslint-config-next": "14.2.5", + "tailwindcss": "^3.4.4", + "tailwindcss-animate": "^1.0.7", + "tsup": "^8.1.0", + "tsx": "^4.16.2" + }, + "publishConfig": { + "access": "public", + "provenance": true + }, + "homepage": "https://assistant-ui.com/", + "repository": { + "type": "git", + "url": "git+https://github.com/Yonom/assistant-ui.git" + }, + "bugs": { + "url": "https://github.com/Yonom/assistant-ui/issues" + } +} diff --git a/packages/react-playground/scripts/build.mts b/packages/react-playground/scripts/build.mts new file mode 100644 index 000000000..f62d019e8 --- /dev/null +++ b/packages/react-playground/scripts/build.mts @@ -0,0 +1,36 @@ +import { build } from "tsup"; + +// JS +await build({ + entry: ["src/index.ts"], + format: ["cjs", "esm"], + dts: true, + sourcemap: true, + clean: true, + esbuildOptions: (options) => { + options.banner = { + js: '"use client";', + }; + }, +}); + +// await build({ +// entry: ["src/tailwindcss/index.ts"], +// outDir: "dist/tailwindcss", +// format: ["cjs", "esm"], +// dts: true, +// sourcemap: true, +// }); + +// // css + +// await build({ +// entry: ["src/styles/tailwindcss/markdown.css"], +// outDir: "dist/styles", +// }); + +// mkdirSync("dist/styles/tailwindcss", { recursive: true }); +// copyFileSync( +// "src/styles/tailwindcss/markdown.css", +// "dist/styles/tailwindcss/markdown.css", +// ); diff --git a/packages/react-playground/src/components/payload-editor-button.tsx b/packages/react-playground/src/components/payload-editor-button.tsx new file mode 100644 index 000000000..6be050a97 --- /dev/null +++ b/packages/react-playground/src/components/payload-editor-button.tsx @@ -0,0 +1,77 @@ +import { EdgeRuntimeRequestOptions } from "@assistant-ui/react"; +import { FC, useState } from "react"; +import { + Dialog, + DialogContent, + DialogFooter, + DialogHeader, + DialogTitle, + DialogTrigger, +} from "./ui/dialog"; +import { Button } from "./ui/button"; +import { + requestOptionsFromOpenAI, + requestOptionsToOpenAI, +} from "../lib/converters"; +import { useGetPlaygroundRuntime } from "../lib/usePlaygroundRuntime"; +import { toCoreMessages, toLanguageModelTools } from "@assistant-ui/react"; +import { JSONEditor } from "./ui/assistant-ui/json-editor"; + +export const PayloadEditorButton: FC = () => { + const getPlaygroundRuntime = useGetPlaygroundRuntime(); + + const [isOpen, setIsOpen] = useState(false); + const [value, setValue] = useState(""); + + const handleOpen = (open: boolean) => { + setIsOpen(open); + if (!open) return; + + const runtime = getPlaygroundRuntime(); + const config = runtime.useModelConfig.getState(); + const options: EdgeRuntimeRequestOptions = { + modelName: config.config?.modelName ?? "gpt-4o", + messages: toCoreMessages(runtime.messages), + tools: toLanguageModelTools(config.tools ?? {}), + ...config.callSettings, + }; + const openAi = requestOptionsToOpenAI(options); + setValue(JSON.stringify(openAi, null, 2)); + }; + + const handleSave = () => { + let jsonValue; + try { + jsonValue = JSON.parse(value); + } catch (e) { + alert("Unable to parse input as JSON"); + return; + } + + const options = requestOptionsFromOpenAI(jsonValue); + getPlaygroundRuntime().setRequestData(options); + setIsOpen(false); + }; + + return ( + + + + + e.preventDefault()} + > + + Edit Payload + +
+ +
+ + + + + + ); +}; diff --git a/packages/react-playground/src/components/ui/assistant-ui/assistant-playground.tsx b/packages/react-playground/src/components/ui/assistant-ui/assistant-playground.tsx new file mode 100644 index 000000000..4ff49321f --- /dev/null +++ b/packages/react-playground/src/components/ui/assistant-ui/assistant-playground.tsx @@ -0,0 +1,344 @@ +"use client"; + +import { ChangeEvent, FC, PropsWithChildren, useState } from "react"; +import { Tool, useAssistantContext } from "@assistant-ui/react"; +import { PayloadEditorButton } from "../../payload-editor-button"; +import { Thread } from "./thread"; +import { Button } from "../button"; + +import { + Dialog, + DialogContent, + DialogFooter, + DialogHeader, + DialogTitle, + DialogTrigger, +} from "../dialog"; +import { Input } from "../input"; +import { + Select, + SelectContent, + SelectItem, + SelectTrigger, + SelectValue, +} from "../select"; +import { Slider } from "../slider"; +import { usePlaygroundRuntime } from "../../../lib/usePlaygroundRuntime"; +import { CircleXIcon, FunctionSquareIcon } from "lucide-react"; +import { TooltipIconButton } from "./tooltip-icon-button"; +import { JSONEditor } from "./json-editor"; + +type ToolDialogProps = PropsWithChildren<{ + name: string; + tool: Tool; + setTool: (name: string, tool: Tool) => void; +}>; + +type ToolItemProps = { + name: string; + tool: Tool; + setTool: (name: string, tool: Tool) => void; + deleteTool: () => void; +}; + +const ToolDialog: FC = ({ name, tool, setTool, children }) => { + const [isOpen, setIsOpen] = useState(false); + const [value, setValue] = useState(""); + + const handleOpen = (open: boolean) => { + setIsOpen(open); + if (!open) return; + + setValue(JSON.stringify({ name, ...tool }, null, 2)); + }; + + const handleSave = () => { + let jsonValue; + try { + jsonValue = JSON.parse(value); + } catch (e) { + alert("Unable to parse input as JSON"); + return; + } + + const { name, ...toolWithoutName } = jsonValue; + if (typeof name !== "string" || name.trim().length === 0) { + alert("Name is required"); + return; + } + try { + setTool(name.trim(), toolWithoutName); + setIsOpen(false); + } catch (e) { + alert((e as Error).message); + } + }; + + return ( + + {children} + e.preventDefault()} + > + + {name === "" ? "Add" : "Edit"} Tool + +
+ +
+ + + + + + ); +}; + +const ToolItem: FC = ({ name, tool, setTool, deleteTool }) => { + return ( + + + + ); +}; + +const DEFAULT_GET_WEATHER_TOOL: Tool = { + description: "Determine weather in my location", + parameters: { + type: "object", + properties: { + location: { + type: "string", + description: "The city and state e.g. San Francisco, CA", + }, + unit: { + type: "string", + enum: ["c", "f"], + }, + }, + required: ["location"], + }, +}; + +const ToolManager: FC = () => { + const { useModelConfig } = usePlaygroundRuntime(); + const tools = useModelConfig((c) => c.tools ?? {}); + + return ( +
+ {Object.entries(tools).map(([name, tool], idx) => ( + { + if (useModelConfig.getState().tools?.[newName] && newName !== name) + throw new Error( + "A tool with the same name already exists. Tool names must be unique.", + ); + + useModelConfig.setState({ + tools: Object.fromEntries( + Object.entries(tools).map(([k, v]) => + k === name ? [newName, newTool] : [k, v], + ), + ), + }); + }} + deleteTool={() => { + const { [name]: _, ...otherTools } = tools; + useModelConfig.setState({ + tools: otherTools, + }); + }} + /> + ))} + { + if (useModelConfig.getState().tools?.[name]) + throw new Error( + "A tool with the same name already exists. Tool names must be unique.", + ); + + useModelConfig.setState({ + tools: { + ...tools, + [name]: newTool, + }, + }); + }} + > + + +
+ ); +}; + +const APIKeyInput: FC = () => { + const { useModelConfig } = usePlaygroundRuntime(); + const value = useModelConfig((c) => c.config?.apiKey ?? ""); + const setValue = (e: ChangeEvent) => { + useModelConfig.setState({ + config: { ...useModelConfig.getState().config, apiKey: e.target.value }, + }); + }; + + return ( +
+ + +
+ ); +}; + +const ModelSelector: FC = () => { + const { useModelConfig } = usePlaygroundRuntime(); + const value = useModelConfig((c) => c.config?.modelName ?? ""); + const setValue = (value: string) => { + useModelConfig.setState({ + config: { + ...useModelConfig.getState().config, + modelName: value, + }, + }); + }; + + return ( +
+ + +
+ ); +}; + +const TemperatureSlider: FC = () => { + const { useModelConfig } = usePlaygroundRuntime(); + const values = useModelConfig((c) => [c.callSettings?.temperature ?? 1]); + const setValues = ([value]: number[]) => { + useModelConfig.setState({ + callSettings: { + ...useModelConfig.getState().callSettings, + temperature: value, + }, + }); + }; + + const setValue = (e: ChangeEvent) => { + setValues([parseFloat(e.target.value)]); + }; + + return ( +
+
+ + +
+ +
+ ); +}; + +const MaxTokensSlider: FC = () => { + const { useModelConfig } = usePlaygroundRuntime(); + const values = useModelConfig((c) => [c.callSettings?.maxTokens ?? 256]); + const setValues = ([value]: number[]) => { + useModelConfig.setState({ + callSettings: { + ...useModelConfig.getState().callSettings, + maxTokens: value, + }, + }); + }; + + const setValue = (e: ChangeEvent) => { + setValues([parseInt(e.target.value)]); + }; + + return ( +
+
+ + +
+ +
+ ); +}; + +const Sidebar: FC = () => { + const { useAssistantActions } = useAssistantContext(); + const handleReset = () => { + useAssistantActions.getState().switchToThread(null); + }; + + return ( +
+ + + + + + + + +
+

Tools

+ +
+
+ ); +}; + +export const AssistantPlayground: FC = () => { + return ( +
+
+ +
+ +
+ ); +}; diff --git a/packages/react-playground/src/components/ui/assistant-ui/image.tsx b/packages/react-playground/src/components/ui/assistant-ui/image.tsx new file mode 100644 index 000000000..80ecb3fe6 --- /dev/null +++ b/packages/react-playground/src/components/ui/assistant-ui/image.tsx @@ -0,0 +1,13 @@ +import { ImageContentPartComponent } from "@assistant-ui/react"; +import { RemoveContentPartButton } from "./remove-content-part"; + +export const Image: ImageContentPartComponent = ({ part }) => { + return ( +
+ {/* eslint-disable-next-line @next/next/no-img-element */} + Image + + +
+ ); +}; diff --git a/packages/react-playground/src/components/ui/assistant-ui/json-editor.tsx b/packages/react-playground/src/components/ui/assistant-ui/json-editor.tsx new file mode 100644 index 000000000..e9e7558db --- /dev/null +++ b/packages/react-playground/src/components/ui/assistant-ui/json-editor.tsx @@ -0,0 +1,23 @@ +import Editor from "react-simple-code-editor"; +import { highlight, languages } from "prismjs"; +import "prismjs/components/prism-json"; +import "prismjs/themes/prism.css"; +import { FC } from "react"; +import { cn } from "../../../lib/utils"; + +type JSONEditorProps = { + placeholder?: string; + value: string; + onValueChange: (value: string) => void; + className?: string; +}; + +export const JSONEditor: FC = ({ className, ...props }) => { + return ( + highlight(code, languages["json"]!, "json")} + className={cn("w-full font-mono [&>textarea]:outline-none", className)} + /> + ); +}; diff --git a/packages/react-playground/src/components/ui/assistant-ui/remove-content-part.tsx b/packages/react-playground/src/components/ui/assistant-ui/remove-content-part.tsx new file mode 100644 index 000000000..2a4e26f31 --- /dev/null +++ b/packages/react-playground/src/components/ui/assistant-ui/remove-content-part.tsx @@ -0,0 +1,23 @@ +import { useContentPartContext, useMessageContext } from "@assistant-ui/react"; +import { TooltipIconButton } from "./tooltip-icon-button"; +import { CircleXIcon } from "lucide-react"; +import { FC } from "react"; +import { useGetPlaygroundRuntime } from "../../../lib/usePlaygroundRuntime"; + +export const RemoveContentPartButton: FC = () => { + const getPlaygroundRuntime = useGetPlaygroundRuntime(); + const { useMessage } = useMessageContext(); + const { useContentPart } = useContentPartContext(); + const handleRemove = () => { + getPlaygroundRuntime().deleteContentPart( + useMessage.getState().message.id, + useContentPart.getState().part, + ); + }; + + return ( + + + + ); +}; diff --git a/packages/react-playground/src/components/ui/assistant-ui/text.tsx b/packages/react-playground/src/components/ui/assistant-ui/text.tsx new file mode 100644 index 000000000..30e88ac1e --- /dev/null +++ b/packages/react-playground/src/components/ui/assistant-ui/text.tsx @@ -0,0 +1,30 @@ +import { + TextContentPartComponent, + useMessageContext, +} from "@assistant-ui/react"; +import TextareaAutosize from "react-textarea-autosize"; +import { useGetPlaygroundRuntime } from "../../../lib/usePlaygroundRuntime"; + +export const Text: TextContentPartComponent = ({ part }) => { + const getPlaygroundRuntime = useGetPlaygroundRuntime(); + const { useMessage } = useMessageContext(); + const handleChange = (e: React.ChangeEvent) => { + try { + getPlaygroundRuntime().setMessageText({ + messageId: useMessage.getState().message.id, + contentPart: part, + text: e.target.value, + }); + } catch (ex) { + console.error(ex); + } + }; + + return ( + + ); +}; diff --git a/packages/react-playground/src/components/ui/assistant-ui/thread.tsx b/packages/react-playground/src/components/ui/assistant-ui/thread.tsx new file mode 100644 index 000000000..ebdbfa20c --- /dev/null +++ b/packages/react-playground/src/components/ui/assistant-ui/thread.tsx @@ -0,0 +1,320 @@ +"use client"; + +import { + ActionBarPrimitive, + ComposerPrimitive, + MessagePrimitive, + ThreadPrimitive, + useMessageContext, + useThreadContext, +} from "@assistant-ui/react"; +import { useState, type FC, type KeyboardEvent, type MouseEvent } from "react"; +import { + CheckIcon, + CopyIcon, + PlusIcon, + SendHorizontalIcon, + SquareFunction, + TrashIcon, +} from "lucide-react"; +import { Select, SelectContent, SelectItem, SelectValue } from "../select"; +import { Avatar, AvatarFallback } from "../avatar"; +import { SelectTrigger } from "@radix-ui/react-select"; +import { Button } from "../button"; +import { TooltipIconButton } from "./tooltip-icon-button"; +import { + DropdownMenu, + DropdownMenuContent, + DropdownMenuItem, + DropdownMenuLabel, + DropdownMenuTrigger, +} from "../dropdown-menu"; +import { + useGetPlaygroundRuntime, + usePlaygroundRuntime, +} from "../../../lib/usePlaygroundRuntime"; +import { ToolUI } from "./tool-ui"; +import { Text } from "./text"; +import { + Dialog, + DialogContent, + DialogFooter, + DialogHeader, + DialogTitle, + DialogTrigger, +} from "../dialog"; +import { Input } from "../input"; +import { Image } from "./image"; + +export const Thread: FC = () => { + return ( + + + + + + +
+ +
+ + + ); +}; + +const ThreadWelcome: FC = () => { + return ( + +
+ + C + +

How can I help you today?

+
+ + ); +}; + +type RoleSelectProps = { + role: string; + setRole: (role: "system" | "assistant" | "user") => void; + children: React.ReactNode; +}; + +const RoleSelect: FC = ({ role, setRole, children }) => { + return ( + + ); +}; + +const Composer: FC = () => { + const [role, setRole] = useState<"user" | "assistant" | "system">("user"); + const { useThread, useThreadActions, useComposer, useThreadMessages } = + useThreadContext(); + const isRunning = useThread((t) => t.isRunning); + const hasValue = useComposer((c) => c.value.length > 0); + + const performAdd = () => { + const composer = useComposer.getState(); + const value = composer.value; + if (!value) return; + + composer.setValue(""); + + useThreadActions.getState().append({ + parentId: useThreadMessages.getState().at(-1)?.id ?? null, + role, + content: [{ type: "text", text: value }], + }); + + setRole("user"); + }; + + const performSubmit = () => { + performAdd(); + useThreadActions.getState().startRun(null); + }; + + const handleAdd = (e: MouseEvent) => { + e.preventDefault(); + performAdd(); + }; + + const handleSubmit = (e: MouseEvent) => { + e.preventDefault(); + performSubmit(); + }; + + const handleKeyDown = (e: KeyboardEvent) => { + // cmd+enter + if (e.metaKey && e.key === "Enter") { + e.preventDefault(); + performSubmit(); + } else if (e.key === "Enter") { + e.preventDefault(); + performAdd(); + } + }; + + return ( + + +
+ + + +
+ + +
+ + ); +}; + +const AddToolCallButton = () => { + const runtime = usePlaygroundRuntime(); + const { useMessage } = useMessageContext(); + const toolNames = runtime.useModelConfig((c) => Object.keys(c.tools ?? {})); + return ( + + + + + + {toolNames.length === 0 && ( + No tools available + )} + {toolNames.map((toolName, idx) => ( + { + runtime.addTool({ + messageId: useMessage.getState().message.id, + toolName, + }); + }} + > + + {toolName} + + ))} + + + ); +}; + +const AddImageButton = () => { + const getPlaygroundRuntime = useGetPlaygroundRuntime(); + const { useMessage } = useMessageContext(); + + const [isOpen, setIsOpen] = useState(false); + const [imageUrl, setImageUrl] = useState(""); + const handleAddImage = () => { + getPlaygroundRuntime().addImage({ + image: new URL(imageUrl).href, + messageId: useMessage.getState().message.id, + }); + setIsOpen(false); + }; + + return ( + + + + + e.preventDefault()} + > + + Add Image + + setImageUrl(e.target.value)} + /> + + + + + + ); +}; + +const Message: FC = () => { + const getPlaygroundRuntime = useGetPlaygroundRuntime(); + const { useMessage } = useMessageContext(); + const role = useMessage((m) => m.message.role); + + const handleDelete = () => { + getPlaygroundRuntime().deleteMessage(useMessage.getState().message.id); + }; + + const setRole = (role: "system" | "assistant" | "user") => { + getPlaygroundRuntime().setRole({ + messageId: useMessage.getState().message.id, + role, + }); + }; + + return ( + +
+ + + +
+ + + + + + + + + + + + + +
+
+ +
+
+ {role === "user" && } + {role === "assistant" && } +
+ + ); +}; diff --git a/packages/react-playground/src/components/ui/assistant-ui/tool-ui.tsx b/packages/react-playground/src/components/ui/assistant-ui/tool-ui.tsx new file mode 100644 index 000000000..ca481a69c --- /dev/null +++ b/packages/react-playground/src/components/ui/assistant-ui/tool-ui.tsx @@ -0,0 +1,90 @@ +import { + ToolCallContentPart, + ToolCallContentPartComponent, + useContentPartContext, +} from "@assistant-ui/react"; +import { FC, useEffect, useState } from "react"; +import { CornerDownRightIcon } from "lucide-react"; +import { JSONEditor } from "./json-editor"; +import { tryJsonParse } from "../../../lib/openai/tryJsonParse"; +import { RemoveContentPartButton } from "./remove-content-part"; + +export const ToolUI: ToolCallContentPartComponent = ({ part }) => { + return ( +
+
+

+ Used tool: + {part.toolName} +

+ +
+ +
+
+ + +
+
+ ); +}; + +const useContentPartTool = () => { + const { useContentPart } = useContentPartContext(); + const part = useContentPart((c) => c.part); + return part as ToolCallContentPart; +}; + +const ToolArgumentsEditor: FC = () => { + const part = useContentPartTool(); + + const [value, setValue] = useState(""); + + useEffect(() => { + setValue(part.argsText ?? JSON.stringify(part.args, null, 2)); + }, [part]); + + const handleValueChange = (value: string) => { + setValue(value); + part.argsText = value; + part.args = tryJsonParse(value); + }; + + return ( + + ); +}; + +const ToolResultEditor: FC = () => { + const part = useContentPartTool(); + + const [value, setValue] = useState(""); + + useEffect(() => { + setValue( + !part.result + ? "" + : typeof part.result === "string" + ? part.result + : JSON.stringify(part.result, null, 2), + ); + }, [part]); + + const handleValueChange = (value: string) => { + setValue(value); + part.result = value; + }; + + return ( + + ); +}; diff --git a/packages/react-playground/src/components/ui/assistant-ui/tooltip-icon-button.tsx b/packages/react-playground/src/components/ui/assistant-ui/tooltip-icon-button.tsx new file mode 100644 index 000000000..ed9fe8b0e --- /dev/null +++ b/packages/react-playground/src/components/ui/assistant-ui/tooltip-icon-button.tsx @@ -0,0 +1,44 @@ +"use client"; + +import { forwardRef } from "react"; + +import { + Tooltip, + TooltipContent, + TooltipProvider, + TooltipTrigger, +} from "../tooltip"; +import { Button, ButtonProps } from "../button"; +import { cn } from "../../../lib/utils"; + +export type TooltipIconButtonProps = ButtonProps & { + tooltip: string; + side?: "top" | "bottom" | "left" | "right"; +}; + +export const TooltipIconButton = forwardRef< + HTMLButtonElement, + TooltipIconButtonProps +>(({ children, tooltip, className, side = "bottom", ...rest }, ref) => { + return ( + + + + + + {tooltip} + + + ); +}); + +TooltipIconButton.displayName = "TooltipIconButton"; diff --git a/packages/react-playground/src/components/ui/avatar.tsx b/packages/react-playground/src/components/ui/avatar.tsx new file mode 100644 index 000000000..412792238 --- /dev/null +++ b/packages/react-playground/src/components/ui/avatar.tsx @@ -0,0 +1,50 @@ +"use client"; + +import * as React from "react"; +import * as AvatarPrimitive from "@radix-ui/react-avatar"; + +import { cn } from "../../lib/utils"; + +const Avatar = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +Avatar.displayName = AvatarPrimitive.Root.displayName; + +const AvatarImage = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +AvatarImage.displayName = AvatarPrimitive.Image.displayName; + +const AvatarFallback = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +AvatarFallback.displayName = AvatarPrimitive.Fallback.displayName; + +export { Avatar, AvatarImage, AvatarFallback }; diff --git a/packages/react-playground/src/components/ui/button.tsx b/packages/react-playground/src/components/ui/button.tsx new file mode 100644 index 000000000..abdc331a9 --- /dev/null +++ b/packages/react-playground/src/components/ui/button.tsx @@ -0,0 +1,57 @@ +import * as React from "react"; +import { Slot } from "@radix-ui/react-slot"; +import { cva, type VariantProps } from "class-variance-authority"; + +import { cn } from "../../lib/utils"; + +const buttonVariants = cva( + "inline-flex items-center justify-center whitespace-nowrap rounded-md text-sm font-medium ring-offset-background transition-colors focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-aui-ring focus-visible:ring-offset-2 disabled:pointer-events-none disabled:opacity-50", + { + variants: { + variant: { + default: + "bg-aui-primary text-aui-primary-foreground hover:bg-aui-primary/90", + destructive: + "bg-aui-destructive text-aui-destructive-foreground hover:bg-aui-destructive/90", + outline: + "border border-aui-input bg-aui-background hover:bg-aui-accent hover:text-aui-accent-foreground", + secondary: + "bg-aui-secondary text-aui-secondary-foreground hover:bg-aui-secondary/80", + ghost: "hover:bg-aui-accent hover:text-aui-accent-foreground", + link: "text-aui-primary underline-offset-4 hover:underline", + }, + size: { + default: "h-10 px-4 py-2", + sm: "h-9 rounded-md px-3", + lg: "h-11 rounded-md px-8", + icon: "h-10 w-10", + }, + }, + defaultVariants: { + variant: "default", + size: "default", + }, + }, +); + +export interface ButtonProps + extends React.ButtonHTMLAttributes, + VariantProps { + asChild?: boolean; +} + +const Button = React.forwardRef( + ({ className, variant, size, asChild = false, ...props }, ref) => { + const Comp = asChild ? Slot : "button"; + return ( + + ); + }, +); +Button.displayName = "Button"; + +export { Button, buttonVariants }; diff --git a/packages/react-playground/src/components/ui/dialog.tsx b/packages/react-playground/src/components/ui/dialog.tsx new file mode 100644 index 000000000..572049bb5 --- /dev/null +++ b/packages/react-playground/src/components/ui/dialog.tsx @@ -0,0 +1,122 @@ +"use client"; + +import * as React from "react"; +import * as DialogPrimitive from "@radix-ui/react-dialog"; +import { X } from "lucide-react"; + +import { cn } from "../../lib/utils"; + +const Dialog = DialogPrimitive.Root; + +const DialogTrigger = DialogPrimitive.Trigger; + +const DialogPortal = DialogPrimitive.Portal; + +const DialogClose = DialogPrimitive.Close; + +const DialogOverlay = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +DialogOverlay.displayName = DialogPrimitive.Overlay.displayName; + +const DialogContent = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, children, ...props }, ref) => ( + + + + {children} + + + Close + + + +)); +DialogContent.displayName = DialogPrimitive.Content.displayName; + +const DialogHeader = ({ + className, + ...props +}: React.HTMLAttributes) => ( +
+); +DialogHeader.displayName = "DialogHeader"; + +const DialogFooter = ({ + className, + ...props +}: React.HTMLAttributes) => ( +
+); +DialogFooter.displayName = "DialogFooter"; + +const DialogTitle = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +DialogTitle.displayName = DialogPrimitive.Title.displayName; + +const DialogDescription = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +DialogDescription.displayName = DialogPrimitive.Description.displayName; + +export { + Dialog, + DialogPortal, + DialogOverlay, + DialogClose, + DialogTrigger, + DialogContent, + DialogHeader, + DialogFooter, + DialogTitle, + DialogDescription, +}; diff --git a/packages/react-playground/src/components/ui/dropdown-menu.tsx b/packages/react-playground/src/components/ui/dropdown-menu.tsx new file mode 100644 index 000000000..4383551b5 --- /dev/null +++ b/packages/react-playground/src/components/ui/dropdown-menu.tsx @@ -0,0 +1,200 @@ +"use client"; + +import * as React from "react"; +import * as DropdownMenuPrimitive from "@radix-ui/react-dropdown-menu"; +import { Check, ChevronRight, Circle } from "lucide-react"; + +import { cn } from "../../lib/utils"; + +const DropdownMenu = DropdownMenuPrimitive.Root; + +const DropdownMenuTrigger = DropdownMenuPrimitive.Trigger; + +const DropdownMenuGroup = DropdownMenuPrimitive.Group; + +const DropdownMenuPortal = DropdownMenuPrimitive.Portal; + +const DropdownMenuSub = DropdownMenuPrimitive.Sub; + +const DropdownMenuRadioGroup = DropdownMenuPrimitive.RadioGroup; + +const DropdownMenuSubTrigger = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef & { + inset?: boolean; + } +>(({ className, inset, children, ...props }, ref) => ( + + {children} + + +)); +DropdownMenuSubTrigger.displayName = + DropdownMenuPrimitive.SubTrigger.displayName; + +const DropdownMenuSubContent = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +DropdownMenuSubContent.displayName = + DropdownMenuPrimitive.SubContent.displayName; + +const DropdownMenuContent = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, sideOffset = 4, ...props }, ref) => ( + + + +)); +DropdownMenuContent.displayName = DropdownMenuPrimitive.Content.displayName; + +const DropdownMenuItem = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef & { + inset?: boolean; + } +>(({ className, inset, ...props }, ref) => ( + +)); +DropdownMenuItem.displayName = DropdownMenuPrimitive.Item.displayName; + +const DropdownMenuCheckboxItem = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, children, checked, ...props }, ref) => ( + + + + + + + {children} + +)); +DropdownMenuCheckboxItem.displayName = + DropdownMenuPrimitive.CheckboxItem.displayName; + +const DropdownMenuRadioItem = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, children, ...props }, ref) => ( + + + + + + + {children} + +)); +DropdownMenuRadioItem.displayName = DropdownMenuPrimitive.RadioItem.displayName; + +const DropdownMenuLabel = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef & { + inset?: boolean; + } +>(({ className, inset, ...props }, ref) => ( + +)); +DropdownMenuLabel.displayName = DropdownMenuPrimitive.Label.displayName; + +const DropdownMenuSeparator = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +DropdownMenuSeparator.displayName = DropdownMenuPrimitive.Separator.displayName; + +const DropdownMenuShortcut = ({ + className, + ...props +}: React.HTMLAttributes) => { + return ( + + ); +}; +DropdownMenuShortcut.displayName = "DropdownMenuShortcut"; + +export { + DropdownMenu, + DropdownMenuTrigger, + DropdownMenuContent, + DropdownMenuItem, + DropdownMenuCheckboxItem, + DropdownMenuRadioItem, + DropdownMenuLabel, + DropdownMenuSeparator, + DropdownMenuShortcut, + DropdownMenuGroup, + DropdownMenuPortal, + DropdownMenuSub, + DropdownMenuSubContent, + DropdownMenuSubTrigger, + DropdownMenuRadioGroup, +}; diff --git a/packages/react-playground/src/components/ui/input.tsx b/packages/react-playground/src/components/ui/input.tsx new file mode 100644 index 000000000..98f83cc42 --- /dev/null +++ b/packages/react-playground/src/components/ui/input.tsx @@ -0,0 +1,25 @@ +import * as React from "react"; + +import { cn } from "../../lib/utils"; + +export interface InputProps + extends React.InputHTMLAttributes {} + +const Input = React.forwardRef( + ({ className, type, ...props }, ref) => { + return ( + + ); + }, +); +Input.displayName = "Input"; + +export { Input }; diff --git a/packages/react-playground/src/components/ui/select.tsx b/packages/react-playground/src/components/ui/select.tsx new file mode 100644 index 000000000..46868646f --- /dev/null +++ b/packages/react-playground/src/components/ui/select.tsx @@ -0,0 +1,160 @@ +"use client"; + +import * as React from "react"; +import * as SelectPrimitive from "@radix-ui/react-select"; +import { Check, ChevronDown, ChevronUp } from "lucide-react"; + +import { cn } from "../../lib/utils"; + +const Select = SelectPrimitive.Root; + +const SelectGroup = SelectPrimitive.Group; + +const SelectValue = SelectPrimitive.Value; + +const SelectTrigger = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, children, ...props }, ref) => ( + span]:line-clamp-1", + className, + )} + {...props} + > + {children} + + + + +)); +SelectTrigger.displayName = SelectPrimitive.Trigger.displayName; + +const SelectScrollUpButton = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + + + +)); +SelectScrollUpButton.displayName = SelectPrimitive.ScrollUpButton.displayName; + +const SelectScrollDownButton = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + + + +)); +SelectScrollDownButton.displayName = + SelectPrimitive.ScrollDownButton.displayName; + +const SelectContent = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, children, position = "popper", ...props }, ref) => ( + + + + + {children} + + + + +)); +SelectContent.displayName = SelectPrimitive.Content.displayName; + +const SelectLabel = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +SelectLabel.displayName = SelectPrimitive.Label.displayName; + +const SelectItem = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, children, ...props }, ref) => ( + + + + + + + + {children} + +)); +SelectItem.displayName = SelectPrimitive.Item.displayName; + +const SelectSeparator = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)); +SelectSeparator.displayName = SelectPrimitive.Separator.displayName; + +export { + Select, + SelectGroup, + SelectValue, + SelectTrigger, + SelectContent, + SelectLabel, + SelectItem, + SelectSeparator, + SelectScrollUpButton, + SelectScrollDownButton, +}; diff --git a/packages/react-playground/src/components/ui/slider.tsx b/packages/react-playground/src/components/ui/slider.tsx new file mode 100644 index 000000000..bae088000 --- /dev/null +++ b/packages/react-playground/src/components/ui/slider.tsx @@ -0,0 +1,28 @@ +"use client"; + +import * as React from "react"; +import * as SliderPrimitive from "@radix-ui/react-slider"; + +import { cn } from "../../lib/utils"; + +const Slider = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + + + + + + +)); +Slider.displayName = SliderPrimitive.Root.displayName; + +export { Slider }; diff --git a/packages/react-playground/src/components/ui/textarea.tsx b/packages/react-playground/src/components/ui/textarea.tsx new file mode 100644 index 000000000..5c414b68f --- /dev/null +++ b/packages/react-playground/src/components/ui/textarea.tsx @@ -0,0 +1,24 @@ +import * as React from "react"; + +import { cn } from "../../lib/utils"; + +export interface TextareaProps + extends React.TextareaHTMLAttributes {} + +const Textarea = React.forwardRef( + ({ className, ...props }, ref) => { + return ( +