Skip to content

Commit 37efd79

Browse files
HarshHarsh
authored andcommitted
ai trust centre and model inventory changes
1 parent 8ffb51d commit 37efd79

File tree

4 files changed

+16
-3
lines changed

4 files changed

+16
-3
lines changed

Clients/src/presentation/pages/AITrustCenter/index.tsx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ const AITrustCenter: React.FC = () => {
136136
value="settings"
137137
disableRipple
138138
/>
139-
<div data-joyride-id="preview-mode-button">
139+
<div data-joyride-id="preview-mode-button" style={{ marginLeft: "auto" }}>
140140
<CustomizableButton
141141
variant="contained"
142142
text="Preview mode"

Clients/src/presentation/pages/AITrustCenter/styles.ts

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,4 @@ export const aiTrustCenterPreviewButtonStyle = {
3838
backgroundColor: "#13715B",
3939
border: "1px solid #13715B",
4040
gap: 2,
41-
marginLeft: "auto",
42-
marginRight: "0",
4341
};

Clients/src/presentation/pages/ModelInventory/ModelRisksTable.tsx

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,11 @@ const SelectorVertical = (props: any) => <ChevronsUpDown size={16} {...props} />
2323
import RiskChip from "../../components/RiskLevel/RiskChip";
2424
import { IModelRisk } from "../../../domain/interfaces/i.modelRisk";
2525
import { User } from "../../../domain/types/User";
26+
import { IModelInventory } from "../../../domain/interfaces/i.modelInventory";
2627

2728
const titleOfTableColumns = [
2829
"risk name",
30+
"model name",
2931
"category",
3032
"risk level",
3133
"status",
@@ -41,6 +43,7 @@ interface ModelRisksTableProps {
4143
onDelete: (riskId: number) => void;
4244
deletingId?: number | null;
4345
users?: User[];
46+
models?: IModelInventory[];
4447
}
4548

4649
const ModelRisksTable: React.FC<ModelRisksTableProps> = ({
@@ -50,6 +53,7 @@ const ModelRisksTable: React.FC<ModelRisksTableProps> = ({
5053
onDelete,
5154
deletingId,
5255
users = [],
56+
models = [],
5357
}) => {
5458
const theme = useTheme();
5559
const [page, setPage] = useState(0);
@@ -104,6 +108,12 @@ const ModelRisksTable: React.FC<ModelRisksTableProps> = ({
104108
return owner?.name || "Unknown";
105109
};
106110

111+
const getModelName = (modelId: number | null | undefined) => {
112+
if (!modelId) return "N/A";
113+
const model = models.find((m) => m.id == modelId);
114+
return model?.model || "Unknown";
115+
};
116+
107117
const tableHeader = useMemo(
108118
() => (
109119
<TableHead
@@ -161,6 +171,9 @@ const ModelRisksTable: React.FC<ModelRisksTableProps> = ({
161171
{row.risk_name}
162172
</Typography>
163173
</TableCell>
174+
<TableCell sx={getCellStyle(row)}>
175+
{getModelName(row.model_id)}
176+
</TableCell>
164177
<TableCell sx={getCellStyle(row)}>
165178
{row.risk_category}
166179
</TableCell>
@@ -227,6 +240,7 @@ const ModelRisksTable: React.FC<ModelRisksTableProps> = ({
227240
rowsPerPage,
228241
cellStyle,
229242
formattedUsers,
243+
models,
230244
onEdit,
231245
onDelete,
232246
deletingId,

Clients/src/presentation/pages/ModelInventory/index.tsx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -916,6 +916,7 @@ const ModelInventory: React.FC = () => {
916916
onDelete={handleDeleteModelRisk}
917917
deletingId={deletingModelRiskId}
918918
users={users}
919+
models={modelInventoryData}
919920
/>
920921
</>
921922
)}

0 commit comments

Comments
 (0)