@stack-spot/ai-chat-widget 1.0.0-dev.1769003015775 → 1.0.0-dev.1769120820021

package/src/views/MessageInput/ModelSwitcher/index.tsx

@@ -15,8 +15,9 @@ export const ModelSwitcher = () => {
   const chat = useCurrentChat()
   const [filter, setFilter] = useState('')
   const [visibleMenu, setVisibleMenu] = useState(false)
-  const [agentData, isLoadingAgentData] = agentToolsClient.agent.useStatefulQuery({ agentId: agentCurrentChat?.id || '' }, 
-    { enabled: !!agentCurrentChat?.id })
+  const [agentData, isLoadingAgentData] = agentToolsClient.agentV2.useStatefulQuery({ 
+    agentCoreId: agentCurrentChat?.id || '', versionNumber: agentCurrentChat?.agent_version_number }, 
+  { enabled: !!agentCurrentChat?.id })
   const [models, isLoadingModels] = 
     genAiInferenceClient.listModels.useStatefulQuery({ pageSize: 999, active: true })
 

package/src/views/MessageInput/ModelSwitcher/utils.tsx

@@ -1,6 +1,6 @@
 import { CitricIconOutline, CitricIconSocial } from '@stack-spot/citric-icons'
 import { IconBox } from '@stack-spot/citric-react'
-import { AgentModel } from '@stack-spot/portal-network/api/agent-tools'
+import { AgentCoreWithSingleVersionResponse2 } from '@stack-spot/portal-network/api/agent-tools'
 import { LlmModelsResponse, PaginatedResponseLlmModelsResponse } from '@stack-spot/portal-network/api/genAiInference'
 import { theme } from '@stack-spot/portal-theme'
 import { Dispatch, ReactElement } from 'react'
@@ -27,19 +27,20 @@ export const providerIcon: Record<string, CitricIconSocial | CitricIconOutline>
 export function getListModelsData(
   chat: ChatState, 
   setVisibleMenu: Dispatch<React.SetStateAction<boolean>>, 
-  agent?: AgentModel,
+  agent?: AgentCoreWithSingleVersionResponse2,
   models?: PaginatedResponseLlmModelsResponse) {
 
   const chatSelectedModelId = chat.get('selected_model_id')
   
-  const listModelsToShow = agent?.visibility_level !== 'built_in' && !!agent?.model_id ? 
-    models?.items.filter((model) => agent?.available_llm_models?.find((modelAvailable) => modelAvailable.model_id === model.id)) : 
+  const listModelsToShow = agent?.visibility_level !== 'built_in' && !!agent?.version?.model_id ? 
+    models?.items.filter((model) => agent?.version?.available_llm_models?.find((modelAvailable) => modelAvailable.model_id === model.id)) : 
     models?.items
 
-  const modelAvailableDefault = agent?.available_llm_models?.find((model) => (model.is_default || model.model_id === agent.model_id))
+  const modelAvailableDefault = agent?.version?.available_llm_models?.find((model) => 
+    (model.is_default || model.model_id === agent.version?.model_id))
   const modelListData = parseModelList(chat, setVisibleMenu, listModelsToShow, modelAvailableDefault?.model_id)
   
-  if (agent?.visibility_level === 'built_in' || !agent?.model_id) {
+  if (agent?.visibility_level === 'built_in' || !agent?.version?.model_id) {
     
     const modelDefaultProviderType = models?.items.find((modelAccount) => 
       chatSelectedModelId ? modelAccount.id === chatSelectedModelId : 
@@ -50,9 +51,9 @@ export function getListModelsData(
     return { modelName: modelDefaultActive?.label, modelProviderType: modelDefaultProviderType, listItemsData: modelListData }
   }
 
-  const modelAccount = models?.items.find((modelAccount) => modelAccount.id === (chatSelectedModelId || agent.model_id))
+  const modelAccount = models?.items.find((modelAccount) => modelAccount.id === (chatSelectedModelId || agent.version?.model_id))
   const modelProviderType = modelAccount?.model_configuration.provider.provider_type
-  const modelSelectedName = modelAccount?.display_name || modelAvailableDefault?.model_name || agent?.model_name
+  const modelSelectedName = modelAccount?.display_name || modelAvailableDefault?.model_name || agent?.version?.model_name
 
   const listItemsData = 
     modelListData && modelListData?.length > 0 ? modelListData : 

package/src/views/Resources.tsx

@@ -46,15 +46,19 @@ const ResourcesPanel = () => {
     const chat = widget.chatTabs.getAll().find(c => c.id === chatId)
     return chat?.getMessages().find(m => m.id === messageId)?.getValue()
   }, [messageId])
-  const [agent] = agentToolsClient.agent.useStatefulQuery({ agentId: message?.agent?.id || '' },
-    { enabled: !!message?.agent?.id })
+  
+  const [agent] = agentToolsClient.agentV2.useStatefulQuery({ 
+    agentCoreId: message?.agent?.id || '', versionNumber: message?.agent?.agent_version_number },
+  { enabled: !!message?.agent?.id })
   const toolkits = useMemo(() => [
-    ...agent?.toolkits?.builtin_toolkits ?? [],
-    ...agent?.toolkits?.custom_toolkits ?? [],
-    ...agent?.toolkits?.mcp_toolkits ?? [],
+    ...agent?.version?.toolkits?.builtin_toolkits ?? [],
+    ...agent?.version?.toolkits?.custom_toolkits ?? [],
+    ...agent?.version?.toolkits?.mcp_toolkits ?? [],
   ], [agent])
   const tools = useMemo(() => message?.tools?.map(id => toolById(id, toolkits)), [messageId, toolkits])
-  const [agentsTools] = agentToolsClient.agentsByIds.useStatefulQuery({ searchAgentsRequest:{ ids: message?.tools || [] } })
+  const agentsSearch = message?.tools?.map((id) => ({ agent_core_id: id })) ?? []
+  const [agentsTools] = agentToolsClient.agentsByIdsV2.useStatefulQuery(
+    { searchAgentsRequestV2: { agents:  agentsSearch } })
   const hasAgentTool = useMemo(() => message?.tools?.some(id => agentsTools?.find((agent) => agent.id === id)), [messageId])
 
   const header = (image?: string, label?: string) => (

package/src/views/Tools.tsx

@@ -42,11 +42,12 @@ const ToolsPanel = () => {
   }, [messageId])
 
   const [toolKits] = agentToolsClient.tools.useStatefulQuery({}, { enabled: !!message?.agent?.id })
-  const [agent] = agentToolsClient.agent.useStatefulQuery({ agentId: message?.agent?.id || '' },
-    { enabled: !!message?.agent?.id })
+  const [agent] = agentToolsClient.agentV2.useStatefulQuery({ agentCoreId: message?.agent?.id || '', 
+    versionNumber: message?.agent?.agent_version_number },
+  { enabled: !!message?.agent?.id })
   const tools = useMemo(() => message?.tools?.map(id => toolById(id, toolKits)), [messageId, toolKits])
-  const customTools = useMemo(() => message?.tools?.map(id => toolById(id, agent?.toolkits?.custom_toolkits)),
-    [messageId, agent?.toolkits?.custom_toolkits])
+  const customTools = useMemo(() => message?.tools?.map(id => toolById(id, agent?.version?.toolkits?.custom_toolkits)),
+    [messageId, agent?.version?.toolkits?.custom_toolkits])
   return !!(tools?.length || customTools?.length) && (
     <ToolList>
       {[...(tools || []), ...(customTools || [])].map(