Add support for function messages (#297)

yieldprotocol · Jun 27, 2023 · f4e9104 · f4e9104
1 parent 465779c
commit f4e9104
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/src/components/MessageItem.tsx b/src/components/MessageItem.tsx
@@ -53,6 +53,8 @@ export const MessageItem = ({ message }: { message: Message }) => {
           <MessageTranslator message={payload} />
         ) : actor === 'system' ? (
           <SystemMessage message={payload} />
+        ) : actor === 'function' ? (
+          <SystemMessage message={payload} />
         ) : (
           <UserMessage
             {...{

diff --git a/src/components/MessageList.tsx b/src/components/MessageList.tsx
@@ -40,7 +40,7 @@ export const MessageList = () => {
   return (
     <div className="h-full">
       {messages.map((message, i) => {
-        if (!showDebugMessages && message.actor == 'system') {
+        if (!showDebugMessages && (message.actor === 'system' || message.actor === 'function')) {
           return <React.Fragment key={message.messageId} />;
         }
         return (

diff --git a/src/components/experimental_/MessageItem_.tsx b/src/components/experimental_/MessageItem_.tsx
@@ -36,6 +36,7 @@ export const MessageItem = ({ message }: { message: Message }) => {
     <div className={`flex w-full flex-col`}>
       {actor === 'bot' && <MessageTranslator message={message} />}
       {actor === 'system' && <SystemMessage message={payload} />}
+      {actor === 'function' && <SystemMessage message={payload} />}
       {isUser && (
         <UserMessage
           {...{

diff --git a/src/components/experimental_/MessageList_.tsx b/src/components/experimental_/MessageList_.tsx
@@ -30,7 +30,7 @@ export const MessageList = () => {
   return (
     <div className="h-full pt-8">
       {messages.map((message, i) => {
-        if (!showDebugMessages && message.actor == 'system') {
+        if (!showDebugMessages && (message.actor === 'system' || message.actor === 'function')) {
           return <React.Fragment key={message.messageId} />;
         }