@stack-spot/ai-chat-widget 1.18.0-beta.8 → 1.18.0

package/src/views/Agents/AgentsTab.tsx

@@ -2,32 +2,21 @@ import { Button, IconBox, Text } from '@citric/core'
 import { Agent, Search } from '@citric/icons'
 import { Placeholder } from '@stack-spot/portal-components/Placeholder'
 import { MiniLogo } from '@stack-spot/portal-components/svg'
-import { agentClient, workspaceAiClient } from '@stack-spot/portal-network'
+import { agentClient } from '@stack-spot/portal-network'
 import { AgentResponse, VisibilityLevel } from '@stack-spot/portal-network/api/agent'
-import { WorkspaceResponse } from '@stack-spot/portal-network/api/workspace-ai'
-import { useCallback, useMemo, useState } from 'react'
+import { useMemo, useState } from 'react'
 import { ButtonFavorite } from '../../components/ButtonFavorite'
-import { NavigationItem } from '../../components/ComponentNavigator'
 import { DescribedRadioGroup } from '../../components/form/DescribedRadioGroup'
 import { IconInput } from '../../components/IconInput'
-import { WorkspaceTabNavigator } from '../../components/WorkspaceTabNavigator'
 import { useCurrentChat } from '../../context/hooks'
 import { useRightPanel } from '../../right-panel/hooks'
-import { ChatProperties } from '../../state/ChatState'
 import { isAgentDefault } from '../../utils/agent'
 import { AgentDescription } from './AgentDescription'
 import { useAgentsDictionary } from './dictionary'
 import { AgentLabel } from './styled'
 import { useAgentFavorites } from './useAgentFavorites'
 
-export interface AgentTabProps {
-  visibility: VisibilityLevel | 'BUILT-IN',
-  workspaceId?: string,
-  agent: React.MutableRefObject<ChatProperties['agent']>,
-  showSubmitButton?: boolean,
-}
-
-export const AgentsTab = ({ visibility, workspaceId, agent, showSubmitButton = true }: AgentTabProps) => {
+export const AgentsTab = ({ visibility }: { visibility: VisibilityLevel | 'BUILT-IN' }) => {
   const t = useAgentsDictionary()
   const { close } = useRightPanel()
   const chat = useCurrentChat()
@@ -37,14 +26,10 @@ export const AgentsTab = ({ visibility, workspaceId, agent, showSubmitButton = t
 
   const agentsBuiltIn = agentClient.publicAgents.useQuery({})
   const agentDefault = agentsBuiltIn.find((agent) => isAgentDefault(agent.slug))
-  const agents = workspaceId
-    ? workspaceAiClient.getAgentFromWorkspaceAi.useQuery({ workspaceId })
-    : visibility === 'BUILT-IN' ? agentsBuiltIn : agentClient.agents.useQuery({ visibility })
-
+  const agents = visibility === 'BUILT-IN' ? agentsBuiltIn : agentClient.agents.useQuery({ visibility })
+  
   const [value, setValue] = useState<AgentResponse | undefined>(
-    agent.current
-      ? agents.find(a => a.id === agent.current?.id)
-      : chat.get('agent') ? agents.find(a => a.id === chat.get('agent')?.id) : agentDefault,
+    chat.get('agent') ? agents.find(a => a.id === chat.get('agent')?.id) : agentDefault,
   )
 
   const filtered = useMemo(
@@ -68,13 +53,6 @@ export const AgentsTab = ({ visibility, workspaceId, agent, showSubmitButton = t
     close()
   }
 
-  const onChange = useCallback((newValue: AgentResponse) => {
-    const isBuiltIn = visibility === 'BUILT-IN' || agentsBuiltIn.some((agent) => agent.id === newValue.id)
-
-    setValue(newValue)
-    agent.current = { ...newValue, builtIn: isBuiltIn, label: newValue.name }
-  }, [])
-
   return (
     <>
       <div className="content">
@@ -87,7 +65,7 @@ export const AgentsTab = ({ visibility, workspaceId, agent, showSubmitButton = t
           )}
           keygen={a => a.id}
           value={value}
-          onChange={onChange}
+          onChange={setValue}
           renderLabel={({ name, avatar, id }) => (
             <AgentLabel>
               {id ? (avatar ? <img src={avatar} /> : <IconBox size="xs"><Agent /></IconBox>) : <MiniLogo />}
@@ -96,7 +74,7 @@ export const AgentsTab = ({ visibility, workspaceId, agent, showSubmitButton = t
           )}
           renderDescription={a => <AgentDescription
             agentId={a.id}
-            visibility={visibility}
+            visibility={visibility}            
             description={a.description}
             llm={a.llm_config?.model_slug}
             numberOfKnowledgeSources={a.knowledge_sources_config?.knowledge_sources?.length ?? 0}
@@ -111,18 +89,7 @@ export const AgentsTab = ({ visibility, workspaceId, agent, showSubmitButton = t
           <Placeholder title={t.noSearchResults} description={t.noSearchResultsDescription} className="no-data-placeholder" />}
         {!agents.length && <Placeholder title={t.noData} description={t.noDataDescription} />}
       </div>
-      {!!filtered.length && showSubmitButton && <Button onClick={submit} disabled={!value}>{t.apply}</Button>}
+      {!!filtered.length && <Button onClick={submit} disabled={!value}>{t.apply}</Button>}
     </>
   )
 }
-
-export function AgentsTabWorkspace({ agent, visibility, showSubmitButton }: AgentTabProps) {
-  const workspaceTabComponents = useMemo(() => ({ agent: AgentsTab }), [agent])
-
-  const buildNavigateParams = (workspace: WorkspaceResponse): NavigationItem<typeof workspaceTabComponents> => ({
-    component: 'agent',
-    props: { visibility, workspaceId: workspace.id, agent, showSubmitButton },
-  })
-
-  return <WorkspaceTabNavigator components={workspaceTabComponents} getNavigateParam={buildNavigateParams} />
-}

package/src/views/Agents/dictionary.ts

@@ -17,7 +17,6 @@ const dictionary = {
     description: 'Description',
     favorites: 'Favorites',
     tools: 'Tools',
-    spaces: 'Spaces',
   },
   pt: {
     title: 'Agentes',
@@ -35,8 +34,6 @@ const dictionary = {
     description: 'Descrição',
     favorites: 'Favoritos',
     tools: 'Ferramentas',
-    spaces: 'Spaces',
-
   },
 } satisfies Dictionary
 

package/src/views/Chat/ChatMessage.tsx

@@ -61,7 +61,8 @@ const RenderInputsEntry = ({ isLast, entry, value, setValue, labels, setLabels }
 
   const renderInputs = () => {
     if (entry.type === 'input-text') {
-      return <Input name={entry.name}  
+      return <Input name={entry.name}
+        value={value[0] ?? ''}
         sx={{ maxWidth: '500px' }}
         autoFocus
         {...entry.validations} onChange={(data) => setValue([data.target.value])} required={entry.required} />
@@ -113,7 +114,7 @@ const RenderInputsEntry = ({ isLast, entry, value, setValue, labels, setLabels }
 
     if (entry.type === 'input-checkbox') {
       return <Flex className="checkbox-group">
-        {entry.options?.map((option) => (<Flex w={6} key={option.label} alignItems="center" className="checkbox-item">
+        {entry.options?.map((option) => (<Flex as="label" w={6} key={option.label} alignItems="center" className="checkbox-item">
           <Checkbox name={entry.name} key={option.label} onChange={(data) => {
             if (data.target.checked) {
               setValue([...value, option.label])
@@ -159,8 +160,8 @@ const RenderInputsEntry = ({ isLast, entry, value, setValue, labels, setLabels }
 export const ChatMessage = ({ message, username, isLast }: Props) => {
   const t = useTranslate(dictionary)
   const [liked, setLiked] = useState<boolean | undefined>()
-  const [value, setValue] = useState<string[]>([])
-  const [labels, setLabels] = useState<string[]>([])
+  const [value, setValue] = useState<string[]>(message.getValue()?.initialValue ?? [])
+  const [labels, setLabels] = useState<string[]>(message.getValue()?.initialValue ?? [])
   const entry = useChatEntry(message)
   const dateFormatter = useDateFormatter()
   const userInfo = entry.agentType === 'user' ? <Avatar size="xs">{username}</Avatar> : <AgentInfo agent={entry.agent} />
@@ -267,7 +268,7 @@ export const ChatMessage = ({ message, username, isLast }: Props) => {
   }
 
   return (entry.content || entry.error || !!entry.steps?.length) && (
-    <li className={entry.agentType} ref={ref}>
+    <li key={entry.messageId} className={entry.agentType} ref={ref}>
       <div className="chat-message" ref={chatRef} onKeyDown={handleKeyDown} tabIndex={0}>
         <div className={`user-info ${entry.agentType}`}>{userInfo}</div>
         {(entry.content || entry.steps) && <div className={listToClass(['message-content', entry.card && 'card', entry.type])}>

package/src/views/ChatHistory/utils.ts

@@ -1,4 +1,4 @@
-import { agentClient, aiClient, workspaceAiClient, workspaceClient } from '@stack-spot/portal-network'
+import { agentClient, aiClient, workspaceClient } from '@stack-spot/portal-network'
 import { ChatProperties } from '../../state/ChatState'
 import { LabeledWithImage } from '../../state/types'
 
@@ -44,19 +44,9 @@ export async function findWorkspace(id: string | null): Promise<ChatProperties['
  */
 export async function getAllAgents(): Promise<LabeledWithImage[]> {
   try {
-    const [agentsAccount, agentsPersonal, agentsShared, publicAgents, workspaceAgents] = await Promise.all([
-      agentClient.agents.query({ visibility: 'ACCOUNT' }),
-      agentClient.agents.query({ visibility: 'PERSONAL' }),
-      agentClient.agents.query({ visibility: 'SHARED' }),
-      agentClient.publicAgents.query({}),
-      
-      workspaceAiClient.workspacesContentsByType.query({ contentType: 'agent' }),
-    ])
-    const workspaceAgentsAsArray = workspaceAgents.flatMap((workspaceAgent: any) =>workspaceAgent?.agents)
+    const [agents, publicAgents] = await Promise.all([agentClient.agents.query({}), agentClient.publicAgents.query({})])
     const builtInAgents = publicAgents.map((a) => ({ id: a.id, label: a.name, image: a.avatar, slug: a.slug, builtIn: true }))
-    return [...agentsAccount, ...agentsPersonal, ...agentsShared, ...workspaceAgentsAsArray]
-      .map(a => ({ id: a.id, label: a.name, image: a.avatar, slug: a.slug, builtIn: false }))
-      .concat(builtInAgents)
+    return agents.map(a => ({ id: a.id, label: a.name, image: a.avatar, slug: a.slug, builtIn: false })).concat(builtInAgents)
   } catch (error) {
     // eslint-disable-next-line no-console
     console.error(error)

package/src/views/KnowledgeSources.tsx

@@ -1,29 +1,24 @@
 import { Button } from '@citric/core'
 import { Search } from '@citric/icons'
 import { Placeholder } from '@stack-spot/portal-components/Placeholder'
-import { aiClient, dataIntegrationClient, workspaceAiClient } from '@stack-spot/portal-network'
+import { aiClient, dataIntegrationClient } from '@stack-spot/portal-network'
 import { KnowledgeSourceItemResponse, VisibilityLevelEnum } from '@stack-spot/portal-network/api/ai'
-import { WorkspaceResponse } from '@stack-spot/portal-network/api/workspace-ai'
 import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
 import { difference, uniqBy } from 'lodash'
 import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react'
 import { ButtonFavorite } from '../components/ButtonFavorite'
-import { NavigationItem } from '../components/ComponentNavigator'
 import { DescribedCheckboxGroup } from '../components/form/DescribedCheckboxGroup'
 import { IconInput } from '../components/IconInput'
 import { RightPanelTabs } from '../components/RightPanelTabs'
-import { WorkspaceTabNavigator } from '../components/WorkspaceTabNavigator'
 import { useCurrentChat, useWidget, useWidgetState } from '../context/hooks'
 import { useRightPanel } from '../right-panel/hooks'
 import { ChatProperties } from '../state/ChatState'
 import { checkIsTrial } from '../utils/check-is-trial'
 
-export interface TabProps {
+interface TabProps {
   visibility: VisibilityLevelEnum,
+  onSubmit: () => void,
   allKS: React.MutableRefObject<ChatProperties['knowledgeSources']>,
-  workspaceId?: string,
-  showSubmitButton?: boolean,
-  onSubmit?: () => void,
 }
 
 export const KnowledgeSources = () => {
@@ -60,7 +55,7 @@ const KnowledgeSourcesPanel = () => {
   useEffect(() => {
     allKS.current = chat.get('knowledgeSources') ?? []
   }, [chat])
-
+  
   const tabs = isTrial ? [
     { title: t.favorites, content: <KnowledgeSourcesTab key="favorite" visibility="favorite" allKS={allKS} onSubmit={onSubmit} /> },
     { title: t.personal, content: <KnowledgeSourcesTab key="personal" visibility="personal" allKS={allKS} onSubmit={onSubmit} /> },
@@ -68,23 +63,19 @@ const KnowledgeSourcesPanel = () => {
     { title: t.favorites, content: <KnowledgeSourcesTab key="favorite" visibility="favorite" allKS={allKS} onSubmit={onSubmit} /> },
     { title: t.personal, content: <KnowledgeSourcesTab key="personal" visibility="personal" allKS={allKS} onSubmit={onSubmit} /> },
     { title: t.shared, content: <KnowledgeSourcesTab key="shared" visibility="shared" allKS={allKS} onSubmit={onSubmit} /> },
-    { title: t.spaces, content: <KnowledgeSourcesTabWorkspace key="workspace" visibility="workspace" allKS={allKS} onSubmit={onSubmit} /> },
-    { title: t.account, content: <KnowledgeSourcesTab key="account" visibility="account" allKS={allKS} onSubmit={onSubmit} /> },    
+    { title: t.account, content: <KnowledgeSourcesTab key="account" visibility="account" allKS={allKS} onSubmit={onSubmit} /> },
   ]
-
+  
   return <RightPanelTabs key={chat.id} tabs={tabs} />
 }
 
-export const KnowledgeSourcesTab = ({ visibility, allKS, onSubmit, workspaceId, showSubmitButton = true }: TabProps) => {
+const KnowledgeSourcesTab = ({ visibility, allKS, onSubmit }: TabProps) => {
   const t = useTranslate(dictionary)
   const [filter, setFilter] = useState('')
-
-  const knowledgeSources = workspaceId
-    ? workspaceAiClient.getKSFromWorkspaceAi.useQuery({ workspaceId })
-    : aiClient.knowledgeSources.useQuery({
-      visibility, order: 'a-to-z', types: ['snippet', 'api', 'event', 'custom'],
-    })
-
+  
+  const knowledgeSources = aiClient.knowledgeSources.useQuery({ 
+    visibility, order: 'a-to-z', types: ['snippet', 'api', 'event', 'custom'], 
+  })
   const listFavorites = dataIntegrationClient.knowledgeSources.useQuery({ visibility: 'favorite' })
   const [addFavorite, pendingAddFav] = dataIntegrationClient.addFavoriteKnowledgeSource.useMutation()
   const [removeFavorite, pendingRemoveFav] = dataIntegrationClient.removeFavoriteKnowledgeSource.useMutation()
@@ -140,7 +131,7 @@ export const KnowledgeSourcesTab = ({ visibility, allKS, onSubmit, workspaceId,
       reject(error)
     } 
   })
-
+  
   const [value, setValue] = useState<KnowledgeSourceItemResponse[]>((() => {
     const currentlySelected = allKS.current?.map(ks => ks.id)
     return knowledgeSources.filter(ks => currentlySelected?.includes(ks.slug))
@@ -148,7 +139,7 @@ export const KnowledgeSourcesTab = ({ visibility, allKS, onSubmit, workspaceId,
   const filtered = useMemo(
     () => filter 
       // Recreate the list so that the favorites list is taken into account
-      ? knowledgeSources.filter(ks => value.includes(ks) || ks.name.toLocaleLowerCase().includes(filter.toLocaleLowerCase()))
+      ? knowledgeSources.filter(ks => value.includes(ks) || ks.name.toLocaleLowerCase().includes(filter.toLocaleLowerCase())) 
       : [...knowledgeSources]
     , [knowledgeSources, filter, value, listFavorites],
   )
@@ -190,23 +181,11 @@ export const KnowledgeSourcesTab = ({ visibility, allKS, onSubmit, workspaceId,
         )}
         {!filtered.length && <Placeholder title={t.noData} description={t.noDataDescription} className="no-data-placeholder" />}
       </div>
-      {!!filtered.length && showSubmitButton && <Button onClick={onSubmit}>{t.apply}</Button>}
+      {!!filtered.length && <Button onClick={onSubmit}>{t.apply}</Button>}
     </>
   )
 }
 
-export function KnowledgeSourcesTabWorkspace({ allKS, onSubmit }: TabProps) {
-  const workspaceTabComponents = useMemo(() => ({ ks: KnowledgeSourcesTab }), [allKS, onSubmit])
-
-  const buildNavigateParams = (workspace: WorkspaceResponse): NavigationItem<typeof workspaceTabComponents> => ({
-    component: 'ks',
-    props: { visibility: 'workspace', workspaceId: workspace.id, allKS, onSubmit },
-  })
-  
-  return <WorkspaceTabNavigator components={workspaceTabComponents} getNavigateParam={buildNavigateParams} />
-}
-
-
 const dictionary = {
   en: {
     title: 'Knowledge Sources',
@@ -220,7 +199,6 @@ const dictionary = {
     noSearchResultsDescription: 'Please, try another search term.',
     noData: 'There are no knowledge sources in this category yet.',
     noDataDescription: 'Use the tabs above to try other categories or use the AI portal to create new knowledge sources.',
-    spaces: 'Spaces',
   },
   pt: {
     title: 'Knowledge Sources',
@@ -234,6 +212,5 @@ const dictionary = {
     noSearchResultsDescription: 'Por favor, tente outra busca.',
     noData: 'Ainda não há knowledge sources nesta categoria.',
     noDataDescription: 'Use as abas acima para tentar outras categorias ou use o Portal AI para criar novos knowledge sources.',
-    spaces: 'Spaces',
   },
 } satisfies Dictionary

package/src/views/MessageInput/AgentSelector.tsx

@@ -1,6 +1,6 @@
 import { Flex, IconBox, Image } from '@citric/core'
 import { Agent } from '@citric/icons'
-import { agentClient, workspaceAiClient } from '@stack-spot/portal-network'
+import { agentClient } from '@stack-spot/portal-network'
 import { AgentResponse } from '@stack-spot/portal-network/api/agent'
 import { uniqBy } from 'lodash'
 import { useCallback } from 'react'
@@ -9,22 +9,17 @@ import { useCurrentChat, useCurrentChatState } from '../../context/hooks'
 import { agentRegex } from '../../regex'
 import { useAgentFavorites } from '../Agents/useAgentFavorites'
 
-const AgentItem = ({ avatar, name, spaceName }: AgentResponse & { spaceName?: string }) => {
+const AgentItem = ({ avatar, name }: AgentResponse) => {
   const avatarComponent = avatar ? <Image width="32" height="32" radius="full" src={avatar} /> : <IconBox size="md"><Agent /></IconBox>
 
   return <Flex flexWrap="nowrap" alignItems="center" sx={{ gap: '8px' }}>
     {avatarComponent}
-    <div>
-      <p className="selector-description">{spaceName}</p>
-      <p className="selector-title">{name}</p>
-    </div>
-
+    <p className="selector-title">{name}</p>
   </Flex>
 }
 
 export const AgentSelector = ({ inputRef, isTrial }: { isTrial: boolean,
-  inputRef: React.RefObject<HTMLTextAreaElement | HTMLInputElement>,
-}) => {
+  inputRef: React.RefObject<HTMLTextAreaElement | HTMLInputElement>, }) => {
   const chat = useCurrentChat()
   const isAgentEnabled = useCurrentChatState('features').agent
   
@@ -38,10 +33,8 @@ export const AgentSelector = ({ inputRef, isTrial }: { isTrial: boolean,
     const isBuiltIn = agent.visibility_level === 'builtIn' || builtInsAgents.some((builtInAgent) => builtInAgent.id === agent.id)
     chat.set(
       'agent',
-      {
-        id: agent.id, label: agent.name, image: agent.avatar, slug: agent.slug,
-        builtIn: isBuiltIn, visibility_level: agent.visibility_level
-      },
+      { id: agent.id, label: agent.name, image: agent.avatar, slug: agent.slug, 
+        builtIn: isBuiltIn, visibility_level: agent.visibility_level },
     )
 
     if (!inputRef.current) return
@@ -53,10 +46,6 @@ export const AgentSelector = ({ inputRef, isTrial }: { isTrial: boolean,
     const personalAgents = agentClient.agents.useQuery({ visibility: 'PERSONAL' })
     const publicAgents = agentClient.publicAgents.useQuery({})
     const builtInsAgents = [...publicAgents.map((agent) => ({ ...agent, visibility_level: 'builtIn' }))]
-    const workspaceAgents = workspaceAiClient.workspacesContentsByType.useQuery({ contentType: 'agent' })
-    
-    const workspaceAgentsWithWorkspaceName = workspaceAgents.flatMap(({ agents, space_name }: any) =>
-      agents.map((agent: any) => ({ ...agent, spaceName: space_name })))
     let accountAgents: AgentResponse[] = []
     let sharedAgents: AgentResponse[] = []
     if (!isTrial) {
@@ -64,7 +53,7 @@ export const AgentSelector = ({ inputRef, isTrial }: { isTrial: boolean,
       sharedAgents = agentClient.agents.useQuery({ visibility: 'SHARED' }) || []
     }
 
-    return uniqBy([...personalAgents, ...workspaceAgentsWithWorkspaceName, ...accountAgents, ...sharedAgents, ...builtInsAgents], 'id')
+    return uniqBy([...personalAgents, ...accountAgents, ...sharedAgents, ...builtInsAgents], 'id')
   }
 
   return <Selector
@@ -77,7 +66,7 @@ export const AgentSelector = ({ inputRef, isTrial }: { isTrial: boolean,
       regex: agentRegex,
       urlBuilder: (agent) => `/agents/${agent?.id}`,
       searchProp: 'name',
-      sections:  isTrial ? ['favorite', 'personal', 'builtIn'] : ['favorite', 'personal', 'workspace', 'account', 'shared', 'builtIn'],
+      sections:  isTrial ? ['favorite', 'personal', 'builtIn'] : ['favorite', 'personal', 'account', 'shared', 'builtIn'],
       renderComponentItem: AgentItem,
       isEnabled: isAgentEnabled,
       onSelect: onSelectItem,

package/src/views/MessageInput/ButtonGroup.tsx

@@ -1,4 +1,4 @@
-import { ChevronRight, Circle, Code, KnowledgeSource, Send, Stack, Times } from '@citric/icons'
+import { ChevronRight, Code, KnowledgeSource, Send, Stack, Times, Workspace } from '@citric/icons'
 import { IconButton } from '@citric/ui'
 import { listToClass } from '@stack-spot/portal-theme'
 import { useEffect, useRef } from 'react'
@@ -62,8 +62,8 @@ export const ButtonGroup = ({ onSend, onCancel, expanded, setExpanded, isLoading
           style={{ width: expanded ? featureButtonsWidth.current : 0 }}
         >
           {features.workspace && (
-            <IconButton aria-label={t.space} title={t.space} onClick={() => widget.set('panel', 'workspace')}>
-              <Circle />
+            <IconButton aria-label={t.workspace} title={t.workspace} onClick={() => widget.set('panel', 'workspace')}>
+              <Workspace />
             </IconButton>
           )}
           {features.knowledgeSource && (

package/src/views/MessageInput/QuickCommandSelector.tsx

@@ -1,5 +1,5 @@
 import { QuickCommand } from '@citric/icons'
-import { aiClient, workspaceAiClient } from '@stack-spot/portal-network'
+import { aiClient } from '@stack-spot/portal-network'
 import { QuickCommandListResponse } from '@stack-spot/portal-network/api/ai'
 import { useCallback } from 'react'
 import { Selector } from '../../components/Selector'
@@ -74,19 +74,8 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
     inputRef.current.focus()
   }, [chat, inputRef])
 
-  const getQuickCommands = () => {
-    const quickCommands = aiClient.quickCommands.useQuery({ order: 'a-to-z' })
-    const quickCommandsFiltered = quickCommands.filter((qc) => qc.visibility_level.toLowerCase() !== 'workspace')
-    const workspaceQuickCommands = workspaceAiClient.workspacesContentsByType.useQuery({ contentType: 'quick_command' })
-    const workspaceQuickCommandsWithWorkspaceName = workspaceQuickCommands.flatMap(({ qcs, space_name }: any) =>
-      qcs.map((qc: any) => ({ ...qc, spaceName: space_name })))
-
-    return [...quickCommandsFiltered, ...workspaceQuickCommandsWithWorkspaceName]
-  }
-
-  const QuickCommandItem = ({ slug, description, spaceName }: QuickCommandListResponse & { spaceName?: string }) => <>
-    <p className="selector-description">{spaceName}</p>
-    <p className="selector-title">/{slug?.toUpperCase()}</p>
+  const QuickCommandItem = ({ slug, description }: QuickCommandListResponse) => <>
+    <p className="selector-title">/{slug.toUpperCase()}</p>
     <p className="selector-description">{description}</p>
   </>
 
@@ -106,7 +95,7 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
       isEnabled: isQuickCommandEnabled,
       onSelect: onSelectItem,
       renderComponentItem: QuickCommandItem,
-      useData: getQuickCommands,
+      useData: () => aiClient.quickCommands.useQuery({ order: 'a-to-z' }),
     }}
   />
 }

package/src/views/MessageInput/dictionary.ts

@@ -5,7 +5,6 @@ const dictionary = {
     stack: 'Select stack',
     code: 'Open code editor',
     workspace: 'Select workspace',
-    space: 'Select space',
     knowledgeSource: 'Select knowledge sources',
     agent: 'Select agent',
     collapse: 'Hide buttons',
@@ -25,7 +24,6 @@ const dictionary = {
     stack: 'Selecionar stack',
     code: 'Abrir editor de código',
     workspace: 'Selecionar workspace',
-    space: 'Select Space',
     knowledgeSource: 'Selecionar knowledge sources',
     agent: 'Selecionar agente',
     collapse: 'Esconder botões',

package/src/views/Stacks.tsx

@@ -1,20 +1,16 @@
 import { Button } from '@citric/core'
 import { Search } from '@citric/icons'
 import { Placeholder } from '@stack-spot/portal-components/Placeholder'
-import { aiClient, workspaceAiClient } from '@stack-spot/portal-network'
+import { aiClient } from '@stack-spot/portal-network'
 import { GetAiStackResponse, VisibilityLevelEnum } from '@stack-spot/portal-network/api/ai'
-import { WorkspaceResponse } from '@stack-spot/portal-network/api/workspace-ai'
 import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
-import { useCallback, useEffect, useMemo, useRef, useState } from 'react'
+import { useEffect, useMemo, useState } from 'react'
 import { ButtonFavorite } from '../components/ButtonFavorite'
-import { NavigationItem } from '../components/ComponentNavigator'
 import { DescribedRadioGroup } from '../components/form/DescribedRadioGroup'
 import { IconInput } from '../components/IconInput'
 import { RightPanelTabs } from '../components/RightPanelTabs'
-import { WorkspaceTabNavigator } from '../components/WorkspaceTabNavigator'
 import { useCurrentChat, useWidget, useWidgetState } from '../context/hooks'
 import { useRightPanel } from '../right-panel/hooks'
-import { ChatProperties } from '../state/ChatState'
 import { checkIsTrial } from '../utils/check-is-trial'
 
 /**
@@ -39,35 +35,23 @@ export const Stacks = () => {
 const StacksPanel =  () => {
   const t = useTranslate(dictionary)
   const chat = useCurrentChat()
+  
   const isTrial = checkIsTrial()
-  const stack = useRef(chat.get('stack'))
-
-  useEffect(() => {
-    stack.current = chat.get('stack')
-  }, [chat])
 
   const tabs = useMemo(() => isTrial ? [
-    { title: t.favorites, content: <StacksTab key="favorites" visibility="favorite" stack={stack} /> },
-    { title: t.personal, content: <StacksTab key="personal" visibility="personal" stack={stack} /> },
+    { title: t.favorites, content: <StacksTab key="favorites" visibility="favorite" /> },
+    { title: t.personal, content: <StacksTab key="personal" visibility="personal" /> },
   ]: [
-    { title: t.favorites, content: <StacksTab key="favorites" visibility="favorite" stack={stack} /> },
-    { title: t.personal, content: <StacksTab key="personal" visibility="personal" stack={stack} /> },
-    { title: t.shared, content: <StacksTab key="shared" visibility="shared" stack={stack} /> },
-    { title: t.spaces, content: <StacksTabWorkspace key="workspace" visibility="workspace" stack={stack} /> },
-    { title: t.account, content: <StacksTab key="account" visibility="account" stack={stack} /> },
+    { title: t.favorites, content: <StacksTab key="favorites" visibility="favorite" /> },
+    { title: t.personal, content: <StacksTab key="personal" visibility="personal" /> },
+    { title: t.shared, content: <StacksTab key="shared" visibility="shared" /> },
+    { title: t.account, content: <StacksTab key="account" visibility="account" /> },
   ], [t, isTrial])
   
   return <RightPanelTabs key={chat.id} tabs={tabs} />
 }
 
-export interface StacksTabProps {
-  visibility: VisibilityLevelEnum,
-  workspaceId?: string,
-  stack: React.MutableRefObject<ChatProperties['stack']>,
-  showSubmitButton?: boolean,
-}
-
-export const StacksTab = ({ visibility, workspaceId, stack, showSubmitButton = true }: StacksTabProps) => {
+const StacksTab = ({ visibility }: { visibility: VisibilityLevelEnum }) => {
   const t = useTranslate(dictionary)
   const { close } = useRightPanel()
   const chat = useCurrentChat()
@@ -127,19 +111,13 @@ export const StacksTab = ({ visibility, workspaceId, stack, showSubmitButton = t
     } 
   })
   
- 
-
-  const stacks = workspaceId
-    ? workspaceAiClient.getStackFromWorkspaceAi.useQuery({ workspaceId })
-    //@ts-ignore
-    : aiClient.aiStacks.useQuery({ visibility, order: 'a-to-z' })
-
-  const currentStackId = stack.current ? stack.current.id : chat.get('stack')?.id
-  const [value, setValue] = useState<GetAiStackResponse | undefined>(stacks.find(s => s.id === currentStackId))
+  //@ts-ignore
+  const stacks = aiClient.aiStacks.useQuery({ visibility, order: 'a-to-z' })
+  const [value, setValue] = useState<GetAiStackResponse | undefined>(stacks.find(s => s.id === chat.get('stack')?.id))
   const filtered = useMemo(() => filter ? 
     // Recreate the list so that the favorites list is taken into account
-    stacks.filter(s => s === value || s.name.toLocaleLowerCase().includes(filter.toLocaleLowerCase())) : [...stacks],
-  [stacks, listFavorites, filter, value], 
+    stacks.filter(s => s === value || s.name.toLocaleLowerCase().includes(filter.toLocaleLowerCase())) : [...stacks], 
+  [stacks, listFavorites, filter, value],
   )
 
   function submit() {
@@ -147,11 +125,6 @@ export const StacksTab = ({ visibility, workspaceId, stack, showSubmitButton = t
     close()
   }
 
-  const onChange = useCallback((newValue: GetAiStackResponse) => {
-    setValue(newValue)
-    stack.current = { ...newValue, label: newValue.name }
-  }, [])
-
   return (
     <>
       <div className="content">
@@ -163,7 +136,7 @@ export const StacksTab = ({ visibility, workspaceId, stack, showSubmitButton = t
           }
           keygen={s => s.id}
           value={value}
-          onChange={onChange}
+          onChange={setValue}
           renderLabel={s => s.name}
           renderDescription={s => s.use_case}
           optionClassName={s => (s === value && filter && !s.name.toLocaleLowerCase().includes(filter.toLocaleLowerCase()))
@@ -176,22 +149,11 @@ export const StacksTab = ({ visibility, workspaceId, stack, showSubmitButton = t
           <Placeholder title={t.noSearchResults} description={t.noSearchResultsDescription} className="no-data-placeholder"/>}
         {!stacks.length && <Placeholder title={t.noData} description={t.noDataDescription} />}
       </div>
-      {!!filtered.length && showSubmitButton && <Button onClick={submit} disabled={!value}>{t.apply}</Button>}
+      {!!filtered.length && <Button onClick={submit} disabled={!value}>{t.apply}</Button>}
     </>
   )
 }
 
-function StacksTabWorkspace({ stack, visibility, showSubmitButton }: StacksTabProps) {
-  const workspaceTabComponents = useMemo(() => ({ stack: StacksTab }), [stack])
-
-  const buildNavigateParams = (workspace: WorkspaceResponse): NavigationItem<typeof workspaceTabComponents> => ({
-    component: 'stack',
-    props: { visibility, workspaceId: workspace.id, stack, showSubmitButton },
-  }) 
-
-  return <WorkspaceTabNavigator components={workspaceTabComponents} getNavigateParam={buildNavigateParams} />
-}
-
 const dictionary = {
   en: {
     title: 'Stacks AI',
@@ -205,7 +167,6 @@ const dictionary = {
     noData: 'There are no stacks in this category yet.',
     noDataDescription: 'Use the tabs above to try other categories or use the AI portal to create new stacks.',
     favorites: 'Favorites',
-    spaces: 'Spaces',
   },
   pt: {
     title: 'Stacks AI',
@@ -219,6 +180,5 @@ const dictionary = {
     noData: 'Ainda não há stacks nesta categoria.',
     noDataDescription: 'Use as abas acima para tentar outras categorias ou use o Portal AI para criar novas stacks.',
     favorites: 'Favoritos',
-    spaces: 'Spaces',
   },
 } satisfies Dictionary