@stack-spot/ai-chat-widget 1.25.1-beta.0 → 1.26.0

package/src/views/MessageInput/SelectContent.tsx

@@ -6,8 +6,8 @@ import { useCurrentChatState, useWidget } from '../../context/hooks'
 import { useUploadManager } from '../../utils/upload/context'
 import { useMessageInputDictionary } from './dictionary'
 
-type chatFeatures = 'workspace' | 'knowledgeSource' | 'stack'
-type chatPanel = 'ks' | 'workspace' | 'stack'
+type ChatFeatures = 'workspace' | 'knowledgeSource' | 'stack'
+type ChatPanel = 'ks' | 'workspace' | 'stack'
 
 export const SelectContent = () => {
   const widget = useWidget()
@@ -15,7 +15,7 @@ export const SelectContent = () => {
   const features = useCurrentChatState('features')
   const t = useMessageInputDictionary()
   const uploadManager = useUploadManager()
-  const hasFeatureButtons = features.workspace || features.knowledgeSource || features.stack
+  const hasFeatureButtons = features.workspace || features.knowledgeSource || features.stack || features.upload
 
   const itemConfigs = [
     {
@@ -36,17 +36,25 @@ export const SelectContent = () => {
       icon: <Spaces />,
       panel: 'workspace',
     },
+    {
+      key: 'upload',
+      label: t.upload,
+      onClick: () => uploadManager.open(),
+      className: 'upload-item',
+      icon: <DocumentUpload />,
+    },
   ]
 
   const listItems = useMemo(() =>
-    itemConfigs.filter(chatFeatures => features[chatFeatures.key as chatFeatures])
+    itemConfigs.filter(chatFeatures => features[chatFeatures.key as ChatFeatures])
       .map(chatFeatures => ({
         label: chatFeatures.label,
         icon: chatFeatures.icon,
-        onClick: () => {
-          widget.set('panel', chatFeatures.panel as chatPanel)
+        className: chatFeatures.className,
+        onClick: chatFeatures.panel ? (() => {
+          widget.set('panel', chatFeatures.panel as ChatPanel)
           setVisibleMenu(false)
-        },
+        }) : chatFeatures.onClick,
       })), [features, widget])
 
   if (!hasFeatureButtons) return null
@@ -70,15 +78,7 @@ export const SelectContent = () => {
         id="chatMessageMenu"
         visible={visibleMenu}
         onHide={() => setVisibleMenu(false)}
-        items={[
-          ...listItems,
-          {
-            label: t.upload,
-            onClick: () => uploadManager.open(),
-            className: 'upload-item',
-            icon: <DocumentUpload />,
-          },
-        ]}
+        items={listItems}
       />
     </>
   )

package/src/views/MessageInput/index.tsx

@@ -128,7 +128,7 @@ export const MessageInput = () => {
         <ProgressBar visible={true} animate={isLoading} 
           backgroundColor={isLoading || !focused ? theme.color.light[500] : theme.color.primary[500]} />
         <ContextBar />
-        <UploadBar />
+        {chat.get('features').upload && <UploadBar />}
         <ButtonBar onSend={onSend} isLoading={isLoading} />
       </MessageInputBox>
     </UploadProvider>

package/src/views/Tools.tsx

@@ -1,4 +1,4 @@
-import { agentToolsClient } from '@stack-spot/portal-network'
+import { agentClient } from '@stack-spot/portal-network'
 import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
 import { useEffect, useMemo } from 'react'
 import styled from 'styled-components'
@@ -40,9 +40,8 @@ const ToolsPanel = () => {
     const chat = widget.chatTabs.getAll().find(c => c.id === chatId)
     return chat?.getMessages().find(m => m.id === messageId)?.getValue()
   }, [messageId])
-
-  const [toolKits] = agentToolsClient.tools.useStatefulQuery({ enabled: !!message?.agent?.id })
-  const tools = useMemo(() => message?.tools?.map(id => toolById(id, toolKits) ?? { id }), [messageId, toolKits])
+  const [agent] = agentClient.agent.useStatefulQuery({ agentId: message?.agent?.id ?? '' }, { enabled: !!message?.agent?.id })
+  const tools = useMemo(() => message?.tools?.map(id => toolById(id, agent?.toolkits) ?? { id }), [messageId, agent])
 
   return !!tools?.length && (
     <ToolList>

package/src/views/Workspaces/WorkspacesTab.tsx

@@ -60,7 +60,7 @@ export const WorkspaceResources = ({ workspaceId, allKS, agent, stack }: Omit<Ta
   const handleNavigate = (resource: WorkspaceResource) => {
     startTransition(() => {
       if (resource.resourceType === 'agent')
-        navigate({ component: 'agent', props: { visibility: 'workspace', agent, workspaceId, showSubmitButton }, fullScreen: true })
+        navigate({ component: 'agent', props: { visibility: 'WORKSPACE', agent, workspaceId, showSubmitButton }, fullScreen: true })
 
       if (resource.resourceType === 'ks')
         navigate({ component: 'ks', props: { visibility: 'workspace', allKS, workspaceId, showSubmitButton }, fullScreen: true })