@flisk/analyze-tracking 0.5.0 → 0.5.2

package/README.md

@@ -26,7 +26,7 @@ npx @flisk/analyze-tracking /path/to/project [options]
 ### Key Options:
 - `-g, --generateDescription`: Generate descriptions of fields (default: `false`)
 - `-p, --provider <provider>`: Specify a provider (options: `openai`, `gemini`)
-- `-m, --model <model>`: Specify a model (options: `gpt-4o-mini`, `gemini-2.0-flash-lite-001`)
+- `-m, --model <model>`: Specify a model (ex: `gpt-4.1-nano`, `gpt-4o-mini`, `gemini-2.0-flash-lite-001`)
 - `-o, --output <output_file>`: Name of the output file (default: `tracking-schema.yaml`)
 - `-c, --customFunction <function_name>`: Specify a custom tracking function
 
@@ -73,7 +73,7 @@ events:
 
 Use this to understand where your events live in the code and how they’re being tracked.
 
-[GPT-4o mini](https://platform.openai.com/docs/models/gpt-4o-mini) is used for generating descriptions of events, properties, and implementations.
+Your LLM of choice is used for generating descriptions of events, properties, and implementations.
 
 See [schema.json](schema.json) for a JSON Schema of the output.
 

package/bin/cli.js

@@ -6,10 +6,7 @@ const commandLineUsage = require('command-line-usage');
 const { run } = require('../src/index');
 const { helpContent } = require('./help');
 
-const SUPPORTED_MODELS = {
-  openai: ['gpt-4o-mini'],
-  gemini: ['gemini-2.0-flash-lite-001'],
-};
+const SUPPORTED_MODEL_PROVIDERS = ['openai', 'gemini'];
 
 // Parse command-line arguments
 const optionDefinitions = [
@@ -100,31 +97,19 @@ if (!targetDir) {
 }
 
 if (generateDescription) {
-  if (!Object.keys(SUPPORTED_MODELS).includes(provider)) {
+  if (!SUPPORTED_MODEL_PROVIDERS.includes(provider)) {
     console.error('Please provide a valid provider. Options: openai, gemini');
     process.exit(1);
   }
 
-  if (provider === 'openai') {
-    if (!SUPPORTED_MODELS.openai.includes(model)) {
-      console.error(`Please provide a valid model for OpenAI. Options: ${SUPPORTED_MODELS.openai.join(', ')}`);
-      process.exit(1);
-    }
-    if (!process.env.OPENAI_API_KEY) {
-      console.error('Please set the `OPENAI_API_KEY` environment variable to use OpenAI for `generateDescription`.');
-      process.exit(1);
-    }
+  if (provider === 'openai' && !process.env.OPENAI_API_KEY) {
+    console.error('Please set the `OPENAI_API_KEY` environment variable to use OpenAI for `generateDescription`.');
+    process.exit(1);
   }
-  
-  if (provider === 'gemini') {
-    if (!SUPPORTED_MODELS.gemini.includes(model)) {
-      console.error(`Please provide a valid model for Gemini. Options: ${SUPPORTED_MODELS.gemini.join(', ')}`);
-      process.exit(1);
-    }
-    if (!process.env.GOOGLE_APPLICATION_CREDENTIALS) {
-      console.error('Please set the `GOOGLE_APPLICATION_CREDENTIALS` environment variable to use Gemini for `generateDescription`.');
-      process.exit(1);
-    }
+
+  if (provider === 'gemini' && !process.env.GOOGLE_APPLICATION_CREDENTIALS) {
+    console.error('Please set the `GOOGLE_APPLICATION_CREDENTIALS` environment variable to use Gemini for `generateDescription`.');
+    process.exit(1);
   }
 }
 

package/bin/help.js

@@ -57,10 +57,10 @@ const helpContent = [
       {
         name: 'model',
         alias: 'm',
-        description: 'Specify a model (options: {italic gpt-4o-mini}, {italic gemini-2.0-flash-lite-001})',
+        description: 'Specify a model (ex: {italic gpt-4.1-nano, gpt-4o-mini}, {italic gemini-2.0-flash-lite-001})',
         type: String,
-        defaultValue: 'gpt-4o-mini',
-        typeLabel: '{underline gpt-4o-mini}'
+        defaultValue: 'gpt-4.1-nano',
+        typeLabel: '{underline gpt-4.1-nano}'
       },
       {
         name: 'output',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@flisk/analyze-tracking",
-  "version": "0.5.0",
+  "version": "0.5.2",
   "description": "Analyzes tracking code in a project and generates data schemas",
   "main": "src/index.js",
   "bin": {
@@ -20,9 +20,9 @@
   },
   "homepage": "https://github.com/fliskdata/analyze-tracking#readme",
   "dependencies": {
-    "@langchain/core": "^0.3.49",
-    "@langchain/google-vertexai": "^0.2.5",
-    "@langchain/openai": "^0.5.7",
+    "@langchain/core": "^0.3.56",
+    "@langchain/google-vertexai": "^0.2.9",
+    "@langchain/openai": "^0.5.10",
     "@ruby/prism": "^1.4.0",
     "@typescript-eslint/parser": "^8.1.0",
     "acorn": "^8.12.1",
@@ -34,7 +34,7 @@
     "isomorphic-git": "^1.27.1",
     "js-yaml": "^4.1.0",
     "typescript": "^5.5.4",
-    "zod": "^3.24.3"
+    "zod": "^3.24.4"
   },
   "devDependencies": {
     "jest": "^29.7.0"

package/src/index.js

@@ -12,13 +12,13 @@ async function run(targetDir, outputPath, customFunction, customSourceDetails, g
     let llm;
     if (provider === 'openai') {
       llm = new ChatOpenAI({
-        modelName: model,
+        model: model,
         temperature: 0,
       });
     }
     if (provider === 'gemini') {
       llm = new ChatVertexAI({
-        modelName: model,
+        model: model,
         temperature: 0,
       });
     }