@probelabs/probe 0.6.0-rc130 → 0.6.0-rc131

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.
Files changed (6) hide show
  1. package/build/agent/ProbeAgent.js +4 -3
  2. package/build/agent/index.js +3 -3
  3. package/cjs/agent/ProbeAgent.cjs +3 -3
  4. package/cjs/index.cjs +3 -3
  5. package/package.json +1 -1
  6. package/src/agent/ProbeAgent.js +4 -3
package/build/agent/ProbeAgent.js CHANGED
@@ -302,7 +302,8 @@ export class ProbeAgent {
302
302
  }
303
303
 
304
304
  // Get API keys from environment variables
305
- const anthropicApiKey = process.env.ANTHROPIC_API_KEY;
305
+ // Support both ANTHROPIC_API_KEY and ANTHROPIC_AUTH_TOKEN (used by Z.AI)
306
+ const anthropicApiKey = process.env.ANTHROPIC_API_KEY || process.env.ANTHROPIC_AUTH_TOKEN;
306
307
  const openaiApiKey = process.env.OPENAI_API_KEY;
307
308
  // Support both GOOGLE_GENERATIVE_AI_API_KEY (official) and GOOGLE_API_KEY (legacy)
308
309
  const googleApiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY || process.env.GOOGLE_API_KEY;
@@ -314,7 +315,7 @@ export class ProbeAgent {
314
315
 
315
316
  // Get custom API URLs if provided
316
317
  const llmBaseUrl = process.env.LLM_BASE_URL;
317
- const anthropicApiUrl = process.env.ANTHROPIC_API_URL || llmBaseUrl;
318
+ const anthropicApiUrl = process.env.ANTHROPIC_API_URL || process.env.ANTHROPIC_BASE_URL || llmBaseUrl;
318
319
  const openaiApiUrl = process.env.OPENAI_API_URL || llmBaseUrl;
319
320
  const googleApiUrl = process.env.GOOGLE_API_URL || llmBaseUrl;
320
321
  const awsBedrockBaseUrl = process.env.AWS_BEDROCK_BASE_URL || llmBaseUrl;
@@ -364,7 +365,7 @@ export class ProbeAgent {
364
365
  } else if ((awsAccessKeyId && awsSecretAccessKey && awsRegion) || awsApiKey) {
365
366
  this.initializeBedrockModel(awsAccessKeyId, awsSecretAccessKey, awsRegion, awsSessionToken, awsApiKey, awsBedrockBaseUrl, modelName);
366
367
  } else {
367
- throw new Error('No API key provided. Please set ANTHROPIC_API_KEY, OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.');
368
+ throw new Error('No API key provided. Please set ANTHROPIC_API_KEY (or ANTHROPIC_AUTH_TOKEN), OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.');
368
369
  }
369
370
  }
370
371
 
package/build/agent/index.js CHANGED
@@ -48273,7 +48273,7 @@ var init_ProbeAgent = __esm({
48273
48273
  this.initializeMockModel();
48274
48274
  return;
48275
48275
  }
48276
- const anthropicApiKey = process.env.ANTHROPIC_API_KEY;
48276
+ const anthropicApiKey = process.env.ANTHROPIC_API_KEY || process.env.ANTHROPIC_AUTH_TOKEN;
48277
48277
  const openaiApiKey = process.env.OPENAI_API_KEY;
48278
48278
  const googleApiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY || process.env.GOOGLE_API_KEY;
48279
48279
  const awsAccessKeyId = process.env.AWS_ACCESS_KEY_ID;
@@ -48282,7 +48282,7 @@ var init_ProbeAgent = __esm({
48282
48282
  const awsSessionToken = process.env.AWS_SESSION_TOKEN;
48283
48283
  const awsApiKey = process.env.AWS_BEDROCK_API_KEY;
48284
48284
  const llmBaseUrl = process.env.LLM_BASE_URL;
48285
- const anthropicApiUrl = process.env.ANTHROPIC_API_URL || llmBaseUrl;
48285
+ const anthropicApiUrl = process.env.ANTHROPIC_API_URL || process.env.ANTHROPIC_BASE_URL || llmBaseUrl;
48286
48286
  const openaiApiUrl = process.env.OPENAI_API_URL || llmBaseUrl;
48287
48287
  const googleApiUrl = process.env.GOOGLE_API_URL || llmBaseUrl;
48288
48288
  const awsBedrockBaseUrl = process.env.AWS_BEDROCK_BASE_URL || llmBaseUrl;
@@ -48323,7 +48323,7 @@ var init_ProbeAgent = __esm({
48323
48323
  } else if (awsAccessKeyId && awsSecretAccessKey && awsRegion || awsApiKey) {
48324
48324
  this.initializeBedrockModel(awsAccessKeyId, awsSecretAccessKey, awsRegion, awsSessionToken, awsApiKey, awsBedrockBaseUrl, modelName);
48325
48325
  } else {
48326
- throw new Error("No API key provided. Please set ANTHROPIC_API_KEY, OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.");
48326
+ throw new Error("No API key provided. Please set ANTHROPIC_API_KEY (or ANTHROPIC_AUTH_TOKEN), OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.");
48327
48327
  }
48328
48328
  }
48329
48329
  /**
package/cjs/agent/ProbeAgent.cjs CHANGED
@@ -73554,7 +73554,7 @@ var init_ProbeAgent = __esm({
73554
73554
  this.initializeMockModel();
73555
73555
  return;
73556
73556
  }
73557
- const anthropicApiKey = process.env.ANTHROPIC_API_KEY;
73557
+ const anthropicApiKey = process.env.ANTHROPIC_API_KEY || process.env.ANTHROPIC_AUTH_TOKEN;
73558
73558
  const openaiApiKey = process.env.OPENAI_API_KEY;
73559
73559
  const googleApiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY || process.env.GOOGLE_API_KEY;
73560
73560
  const awsAccessKeyId = process.env.AWS_ACCESS_KEY_ID;
@@ -73563,7 +73563,7 @@ var init_ProbeAgent = __esm({
73563
73563
  const awsSessionToken = process.env.AWS_SESSION_TOKEN;
73564
73564
  const awsApiKey = process.env.AWS_BEDROCK_API_KEY;
73565
73565
  const llmBaseUrl = process.env.LLM_BASE_URL;
73566
- const anthropicApiUrl = process.env.ANTHROPIC_API_URL || llmBaseUrl;
73566
+ const anthropicApiUrl = process.env.ANTHROPIC_API_URL || process.env.ANTHROPIC_BASE_URL || llmBaseUrl;
73567
73567
  const openaiApiUrl = process.env.OPENAI_API_URL || llmBaseUrl;
73568
73568
  const googleApiUrl = process.env.GOOGLE_API_URL || llmBaseUrl;
73569
73569
  const awsBedrockBaseUrl = process.env.AWS_BEDROCK_BASE_URL || llmBaseUrl;
@@ -73604,7 +73604,7 @@ var init_ProbeAgent = __esm({
73604
73604
  } else if (awsAccessKeyId && awsSecretAccessKey && awsRegion || awsApiKey) {
73605
73605
  this.initializeBedrockModel(awsAccessKeyId, awsSecretAccessKey, awsRegion, awsSessionToken, awsApiKey, awsBedrockBaseUrl, modelName);
73606
73606
  } else {
73607
- throw new Error("No API key provided. Please set ANTHROPIC_API_KEY, OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.");
73607
+ throw new Error("No API key provided. Please set ANTHROPIC_API_KEY (or ANTHROPIC_AUTH_TOKEN), OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.");
73608
73608
  }
73609
73609
  }
73610
73610
  /**
package/cjs/index.cjs CHANGED
@@ -73734,7 +73734,7 @@ var init_ProbeAgent = __esm({
73734
73734
  this.initializeMockModel();
73735
73735
  return;
73736
73736
  }
73737
- const anthropicApiKey = process.env.ANTHROPIC_API_KEY;
73737
+ const anthropicApiKey = process.env.ANTHROPIC_API_KEY || process.env.ANTHROPIC_AUTH_TOKEN;
73738
73738
  const openaiApiKey = process.env.OPENAI_API_KEY;
73739
73739
  const googleApiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY || process.env.GOOGLE_API_KEY;
73740
73740
  const awsAccessKeyId = process.env.AWS_ACCESS_KEY_ID;
@@ -73743,7 +73743,7 @@ var init_ProbeAgent = __esm({
73743
73743
  const awsSessionToken = process.env.AWS_SESSION_TOKEN;
73744
73744
  const awsApiKey = process.env.AWS_BEDROCK_API_KEY;
73745
73745
  const llmBaseUrl = process.env.LLM_BASE_URL;
73746
- const anthropicApiUrl = process.env.ANTHROPIC_API_URL || llmBaseUrl;
73746
+ const anthropicApiUrl = process.env.ANTHROPIC_API_URL || process.env.ANTHROPIC_BASE_URL || llmBaseUrl;
73747
73747
  const openaiApiUrl = process.env.OPENAI_API_URL || llmBaseUrl;
73748
73748
  const googleApiUrl = process.env.GOOGLE_API_URL || llmBaseUrl;
73749
73749
  const awsBedrockBaseUrl = process.env.AWS_BEDROCK_BASE_URL || llmBaseUrl;
@@ -73784,7 +73784,7 @@ var init_ProbeAgent = __esm({
73784
73784
  } else if (awsAccessKeyId && awsSecretAccessKey && awsRegion || awsApiKey) {
73785
73785
  this.initializeBedrockModel(awsAccessKeyId, awsSecretAccessKey, awsRegion, awsSessionToken, awsApiKey, awsBedrockBaseUrl, modelName);
73786
73786
  } else {
73787
- throw new Error("No API key provided. Please set ANTHROPIC_API_KEY, OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.");
73787
+ throw new Error("No API key provided. Please set ANTHROPIC_API_KEY (or ANTHROPIC_AUTH_TOKEN), OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.");
73788
73788
  }
73789
73789
  }
73790
73790
  /**
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@probelabs/probe",
3
- "version": "0.6.0-rc130",
3
+ "version": "0.6.0-rc131",
4
4
  "description": "Node.js wrapper for the probe code search tool",
5
5
  "main": "src/index.js",
6
6
  "module": "src/index.js",
package/src/agent/ProbeAgent.js CHANGED
@@ -302,7 +302,8 @@ export class ProbeAgent {
302
302
  }
303
303
 
304
304
  // Get API keys from environment variables
305
- const anthropicApiKey = process.env.ANTHROPIC_API_KEY;
305
+ // Support both ANTHROPIC_API_KEY and ANTHROPIC_AUTH_TOKEN (used by Z.AI)
306
+ const anthropicApiKey = process.env.ANTHROPIC_API_KEY || process.env.ANTHROPIC_AUTH_TOKEN;
306
307
  const openaiApiKey = process.env.OPENAI_API_KEY;
307
308
  // Support both GOOGLE_GENERATIVE_AI_API_KEY (official) and GOOGLE_API_KEY (legacy)
308
309
  const googleApiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY || process.env.GOOGLE_API_KEY;
@@ -314,7 +315,7 @@ export class ProbeAgent {
314
315
 
315
316
  // Get custom API URLs if provided
316
317
  const llmBaseUrl = process.env.LLM_BASE_URL;
317
- const anthropicApiUrl = process.env.ANTHROPIC_API_URL || llmBaseUrl;
318
+ const anthropicApiUrl = process.env.ANTHROPIC_API_URL || process.env.ANTHROPIC_BASE_URL || llmBaseUrl;
318
319
  const openaiApiUrl = process.env.OPENAI_API_URL || llmBaseUrl;
319
320
  const googleApiUrl = process.env.GOOGLE_API_URL || llmBaseUrl;
320
321
  const awsBedrockBaseUrl = process.env.AWS_BEDROCK_BASE_URL || llmBaseUrl;
@@ -364,7 +365,7 @@ export class ProbeAgent {
364
365
  } else if ((awsAccessKeyId && awsSecretAccessKey && awsRegion) || awsApiKey) {
365
366
  this.initializeBedrockModel(awsAccessKeyId, awsSecretAccessKey, awsRegion, awsSessionToken, awsApiKey, awsBedrockBaseUrl, modelName);
366
367
  } else {
367
- throw new Error('No API key provided. Please set ANTHROPIC_API_KEY, OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.');
368
+ throw new Error('No API key provided. Please set ANTHROPIC_API_KEY (or ANTHROPIC_AUTH_TOKEN), OPENAI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY (or GOOGLE_API_KEY), AWS credentials (AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_REGION), or AWS_BEDROCK_API_KEY environment variables.');
368
369
  }
369
370
  }
370
371