Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -197,27 +197,37 @@ export default function PublishPage() {
setStatus('saving');
mutate(
{
status: 'ACTIVE',
isPublic: true,
isActive: true,
status: isPublished ? 'REGISTERED' : 'ACTIVE',
isPublic: isPublished ? false : true,
isActive: isPublished ? false : true,
},
{
onSuccess: () => {
toast('Model published successfully', {
id: PUBLISH_SUCCESS_TOAST_ID,
});
toast(
isPublished
? 'Model unpublished successfully'
: 'Model published successfully',
{
id: PUBLISH_SUCCESS_TOAST_ID,
}
);
setStatus('saved');
refetch();
router.push(
`/dashboard/${params.entityType}/${params.entitySlug}/aimodels?tab=active`
`/dashboard/${params.entityType}/${params.entitySlug}/aimodels?tab=${isPublished ? 'draft' : 'published'}`
);
},
onError: (error: any) => {
const errorMessage =
typeof error?.message === 'string' && error.message.trim()
? error.message.trim()
: 'Unable to publish model right now. Please try again.';
toast(`Error: ${errorMessage}`, { id: PUBLISH_ERROR_TOAST_ID });
: isPublished
? 'Unable to unpublish model right now. Please try again.'
: 'Unable to publish model right now. Please try again.';
toast(
isPublished ? `Error: ${errorMessage}` : `Error: ${errorMessage}`,
{ id: PUBLISH_ERROR_TOAST_ID }
);
setStatus('unsaved');
},
}
Expand Down Expand Up @@ -531,7 +541,7 @@ export default function PublishPage() {
disabled={isPublishDisabled}
loading={updateLoading}
>
Publish
{isPublished ? 'Unpublish' : 'Publish'}
</Button>
</>
)}
Expand Down
52 changes: 30 additions & 22 deletions app/[locale]/dashboard/[entityType]/[entitySlug]/aimodels/page.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -70,20 +70,20 @@ export default function AIModelsPage({

let navigationOptions = [
{
label: 'Registered',
url: `registered`,
selected: navigationTab === 'registered',
label: 'Draft',
url: `draft`,
selected: navigationTab === 'draft',
},
{
label: 'Active',
url: `active`,
selected: navigationTab === 'active',
},
{
label: 'Approved',
url: `approved`,
selected: navigationTab === 'approved',
label: 'Published',
url: `published`,
selected: navigationTab === 'published',
},
// {
// label: 'Approved',
// url: `approved`,
// selected: navigationTab === 'approved',
// },
];

const AllAIModels: { data: any; isLoading: boolean; refetch: any } = useQuery(
Expand All @@ -97,11 +97,11 @@ export default function AIModelsPage({
{
filters: {
status:
navigationTab === 'active'
navigationTab === 'published'
? 'ACTIVE'
: navigationTab === 'approved'
? 'APPROVED'
: 'REGISTERED',
: navigationTab === 'draft'
? 'REGISTERED'
: 'APPROVED',
},
order: { updatedAt: 'DESC' },
}
Expand All @@ -110,7 +110,7 @@ export default function AIModelsPage({

useEffect(() => {
if (navigationTab === null || navigationTab === undefined)
setNavigationTab('registered');
setNavigationTab('draft');
AllAIModels.refetch();
}, [AllAIModels, navigationTab, setNavigationTab]);

Expand All @@ -130,11 +130,15 @@ export default function AIModelsPage({
),
{
onSuccess: () => {
toast(`Deleted AI Model successfully`,{id: AIMODELS_ACTION_TOAST_ID});
toast(`Deleted AI Model successfully`, {
id: AIMODELS_ACTION_TOAST_ID,
});
AllAIModels.refetch();
},
onError: (err: any) => {
toast('Error: ' + err.message.split(':')[0],{id: AIMODELS_ACTION_TOAST_ID});
toast('Error: ' + err.message.split(':')[0], {
id: AIMODELS_ACTION_TOAST_ID,
});
},
}
);
Expand Down Expand Up @@ -163,13 +167,17 @@ export default function AIModelsPage({
{
onSuccess: (data: any) => {
const newModelId = data.createAiModel.data.id;
toast(`Created AI Model successfully`,{id: AIMODELS_ACTION_TOAST_ID});
toast(`Created AI Model successfully`, {
id: AIMODELS_ACTION_TOAST_ID,
});
router.push(
`/dashboard/${entityType}/${entitySlug}/aimodels/edit/${newModelId}/details`
);
},
onError: (err: any) => {
toast('Error: ' + err.message.split(':')[0],{id: AIMODELS_ACTION_TOAST_ID});
toast('Error: ' + err.message.split(':')[0], {
id: AIMODELS_ACTION_TOAST_ID,
});
},
}
);
Expand All @@ -186,7 +194,7 @@ export default function AIModelsPage({
<LinkButton
kind="tertiary"
size="medium"
href={`/dashboard/${entityType}/${entitySlug}/aimodels/edit/${row.original.id}/details?tab=${navigationTab ?? 'registered'}`}
href={`/dashboard/${entityType}/${entitySlug}/aimodels/edit/${row.original.id}/details?tab=${navigationTab ?? 'draft'}`}
>
<span className="line-clamp-1 max-w-[280px]">
{row.original.displayName}
Expand Down Expand Up @@ -260,7 +268,7 @@ export default function AIModelsPage({
<Navigation
setNavigationTab={setNavigationTab}
options={navigationOptions}
currentValue={navigationTab ?? 'registered'}
currentValue={navigationTab ?? 'draft'}
/>

{AllAIModels.data?.aiModels.length > 0 ? (
Expand Down
Loading