pi-nvidia-nim 1.1.2 → 1.1.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/index.ts +4 -3
- package/package.json +1 -1
package/index.ts
CHANGED
|
@@ -48,7 +48,8 @@ import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
|
|
|
48
48
|
// =============================================================================
|
|
49
49
|
|
|
50
50
|
const NVIDIA_NIM_BASE_URL = "https://integrate.api.nvidia.com/v1";
|
|
51
|
-
const
|
|
51
|
+
const NVIDIA_NIM_API_KEY_ENV_NAME = "NVIDIA_NIM_API_KEY"; // Bare env var name for process.env
|
|
52
|
+
const NVIDIA_NIM_API_KEY_ENV = "$NVIDIA_NIM_API_KEY"; // $ prefix for pi's apiKey resolution
|
|
52
53
|
const PROVIDER_NAME = "nvidia-nim";
|
|
53
54
|
|
|
54
55
|
// =============================================================================
|
|
@@ -638,10 +639,10 @@ export default function (pi: ExtensionAPI) {
|
|
|
638
639
|
|
|
639
640
|
// On session start, discover additional models from the API
|
|
640
641
|
pi.on("session_start", async (_event, ctx) => {
|
|
641
|
-
const apiKey = process.env[
|
|
642
|
+
const apiKey = process.env[NVIDIA_NIM_API_KEY_ENV_NAME];
|
|
642
643
|
if (!apiKey) {
|
|
643
644
|
ctx.ui.notify(
|
|
644
|
-
`NVIDIA NIM: Set ${
|
|
645
|
+
`NVIDIA NIM: Set ${NVIDIA_NIM_API_KEY_ENV_NAME} env var to enable models. Get a key at https://build.nvidia.com`,
|
|
645
646
|
"warning",
|
|
646
647
|
);
|
|
647
648
|
return;
|