fix/nvidia-nim

packages/components/credentials/NvdiaNIMApi.credential.ts

@@ -1,4 +1,4 @@
-import { INodeParams, INodeCredential } from '../src/Interface'
+import { INodeCredential, INodeParams } from '../src/Interface'
 
 class NvidiaNIMApi implements INodeCredential {
     label: string
@@ -8,12 +8,12 @@ class NvidiaNIMApi implements INodeCredential {
     inputs: INodeParams[]
 
     constructor() {
-        this.label = 'Nvdia NIM API Key'
+        this.label = 'NVIDIA NGC API Key'
         this.name = 'nvidiaNIMApi'
         this.version = 1.0
         this.inputs = [
             {
-                label: 'Nvidia NIM API Key',
+                label: 'NVIDIA NGC API Key',
                 name: 'nvidiaNIMApiKey',
                 type: 'password'
             }

packages/server/package.json

@@ -97,7 +97,7 @@
         "multer": "^1.4.5-lts.1",
         "multer-s3": "^3.0.1",
         "mysql2": "^3.11.3",
-        "nim-container-manager": "^1.0.5",
+        "flowise-nim-container-manager": "^1.0.8",
         "openai": "^4.82.0",
         "pg": "^8.11.1",
         "posthog-node": "^3.5.0",

packages/server/src/controllers/nvidia-nim/index.ts

@@ -1,7 +1,7 @@
 import axios from 'axios'
-import { Request, Response, NextFunction } from 'express'
+import { NextFunction, Request, Response } from 'express'
 
-const { NimContainerManager } = require('nim-container-manager')
+const { NimContainerManager } = require('flowise-nim-container-manager')
 
 const getToken = async (req: Request, res: Response, next: NextFunction) => {
     try {

packages/ui/src/ui-component/dialog/NvidiaNIMDialog.jsx

@@ -1,34 +1,38 @@
-import { useState, useEffect } from 'react'
-import { createPortal } from 'react-dom'
-import axios from 'axios'
-import PropTypes from 'prop-types'
 import {
-    Dialog,
-    DialogTitle,
-    DialogContent,
-    DialogActions,
     Button,
     CircularProgress,
-    Stepper,
+    Dialog,
+    DialogActions,
+    DialogContent,
+    DialogTitle,
+    FormControl,
+    InputLabel,
+    MenuItem,
+    Select,
     Step,
     StepLabel,
-    Select,
-    MenuItem,
-    FormControl,
-    InputLabel
+    Stepper
 } from '@mui/material'
+import axios from 'axios'
+import PropTypes from 'prop-types'
+import { useEffect, useState } from 'react'
+import { createPortal } from 'react-dom'
 
 const NvidiaNIMDialog = ({ open, onClose, onComplete }) => {
     const portalElement = document.getElementById('portal')
 
     const modelOptions = {
-        'nv-mistralai/mistral-nemo-12b-instruct:latest': {
-            label: 'Mistral Nemo 12B Instruct',
-            licenseUrl: 'https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nv-mistralai/containers/mistral-nemo-12b-instruct'
-        },
-        'meta/llama-3.1-8b-instruct-rtx:latest': {
+        'nvcr.io/nim/meta/llama-3.1-8b-instruct:1.8.0-RTX': {
             label: 'Llama 3.1 8B Instruct',
             licenseUrl: 'https://catalog.ngc.nvidia.com/orgs/nim/teams/meta/containers/llama-3.1-8b-instruct'
+        },
+        'nvcr.io/nim/deepseek-ai/deepseek-r1-distill-llama-8b:1.8.0-RTX': {
+            label: 'DeepSeek R1 Distill Llama 8B',
+            licenseUrl: 'https://catalog.ngc.nvidia.com/orgs/nim/teams/deepseek-ai/containers/deepseek-r1-distill-llama-8b'
+        },
+        'nvcr.io/nim/nv-mistralai/mistral-nemo-12b-instruct:1.8.0-rtx': {
+            label: 'Mistral Nemo 12B Instruct',
+            licenseUrl: 'https://catalog.ngc.nvidia.com/orgs/nim/teams/nv-mistralai/containers/mistral-nemo-12b-instruct'
         }
     }