fix: missing exported functions and variables

Author: dosco

eslint.config.js

@@ -85,6 +85,11 @@ export default [
                     format: ['camelCase'],
                     leadingUnderscore: 'allow',
                 },
+                {
+                    selector: ['variable'],
+                    modifiers: ['exported'],
+                    format: ['PascalCase','camelCase'],
+                },
                 {
                     selector: 'class',
                     format: ['PascalCase'],
@@ -119,7 +124,7 @@ export default [
                     selector: ['variable', 'function'],
                     modifiers: ['exported'],
                     format: ['PascalCase'],
-                    prefix: ['ax']
+                    prefix: ['Ax']
                 },
                 {
                     selector: ['class', 'interface', 'typeAlias', 'enum', 'typeParameter'],

scripts/generateIndex.ts

@@ -1,6 +1,6 @@
 #!/usr/bin/env tsx
-import * as fs from 'fs'
-import * as path from 'path'
+import * as fs from 'node:fs'
+import * as path from 'node:path'
 
 import * as ts from 'typescript'
 
@@ -32,7 +32,7 @@ function processExportDeclaration(node: ts.ExportDeclaration): ExportInfo[] {
   const exportsMap = new Map<string, ExportInfo>()
 
   if (node.exportClause && ts.isNamedExports(node.exportClause)) {
-    node.exportClause.elements.forEach((element) => {
+    for (const element of node.exportClause.elements) {
       const originalName = element.name.text
       if (hasValidPrefix(originalName)) {
         // Only add if not already present
@@ -45,7 +45,7 @@ function processExportDeclaration(node: ts.ExportDeclaration): ExportInfo[] {
           })
         }
       }
-    })
+    }
   }
 
   return Array.from(exportsMap.values())
@@ -104,6 +104,34 @@ function processDeclaration(
   return exports
 }
 
+/**
+ * Processes a variable statement node to extract exported variables with valid prefixes
+ */
+function processVariableStatement(node: ts.VariableStatement): ExportInfo[] {
+  const exports: ExportInfo[] = []
+  // Check if the variable statement has an export modifier
+  if (
+    !node.modifiers ||
+    !node.modifiers.some((m) => m.kind === ts.SyntaxKind.ExportKeyword)
+  ) {
+    return exports
+  }
+
+  for (const declaration of node.declarationList.declarations) {
+    if (ts.isIdentifier(declaration.name)) {
+      const originalName = declaration.name.text
+      if (hasValidPrefix(originalName)) {
+        exports.push({
+          originalName,
+          prefixedName: originalName,
+          kind: 'value',
+        })
+      }
+    }
+  }
+  return exports
+}
+
 /**
  * Processes a TypeScript source file to extract all exports with valid prefixes
  */
@@ -126,6 +154,8 @@ function processFile(
       exports.push(...processExportDeclaration(node))
     } else if (ts.isExportAssignment(node)) {
       exports.push(...processExportAssignment(node))
+    } else if (ts.isVariableStatement(node)) {
+      exports.push(...processVariableStatement(node))
     } else if (
       ts.isClassDeclaration(node) ||
       ts.isInterfaceDeclaration(node) ||
@@ -169,7 +199,8 @@ function findTsFiles(dir: string): string[] {
     const fullPath = path.join(dir, entry.name)
     if (entry.isDirectory()) {
       return findTsFiles(fullPath)
-    } else if (isTargetTsFile(entry.name)) {
+    }
+    if (isTargetTsFile(entry.name)) {
       return [fullPath]
     }
     return []
@@ -245,8 +276,8 @@ function generateExportStatements(exportMap: Map<string, ExportInfo[]>): {
   const valueExports: string[] = []
   const typeExports: string[] = []
 
-  exportMap.forEach((exports) => {
-    exports.forEach((exp) => {
+  for (const [, exports] of exportMap) {
+    for (const exp of exports) {
       if (exp.kind === 'type') {
         typeExports.push(`export type { ${exp.originalName} };`)
       } else {
@@ -255,8 +286,8 @@ function generateExportStatements(exportMap: Map<string, ExportInfo[]>): {
           : `export { ${exp.originalName} };`
         valueExports.push(exportLine)
       }
-    })
-  })
+    }
+  }
 
   return {
     valueExports: valueExports.sort(),
@@ -276,19 +307,17 @@ function generateIndexContent(exportMap: Map<string, ExportInfo[]>): string {
     .map(([filePath, exports]) => generateImportStatement(filePath, exports))
     .filter(Boolean)
     .sort()
-  content += imports.join('\n') + '\n\n'
+  content = `${content}${imports.join('\n')}\n\n`
 
   // Generate exports
   const { valueExports, typeExports } = generateExportStatements(exportMap)
 
   if (valueExports.length > 0) {
-    content += '// Value exports\n'
-    content += valueExports.join('\n') + '\n\n'
+    content = `${content}// Value exports\n${valueExports.join('\n')}\n\n`
   }
 
   if (typeExports.length > 0) {
-    content += '// Type exports\n'
-    content += typeExports.join('\n') + '\n'
+    content = `${content}// Type exports\n${typeExports.join('\n')}\n`
   }
 
   return content
@@ -303,7 +332,9 @@ function generateIndex(): void {
 
   // Find and process all TypeScript files
   const tsFiles = findTsFiles(currentDir)
-  tsFiles.forEach((file) => processFile(file, currentDir, exportMap))
+  for (const file of tsFiles) {
+    processFile(file, currentDir, exportMap)
+  }
 
   if (exportMap.size === 0) {
     console.log('No ax/Ax exports found')

src/ax/docs/reranker.ts

@@ -1,7 +1,7 @@
 import type { AxAIService } from '../ai/types.js'
 import { AxGen, type AxGenOptions } from '../dsp/generate.js'
 import type { AxProgramForwardOptions } from '../dsp/program.js'
-import { axStringUtil } from '../dsp/strutil.js'
+import { AxStringUtil } from '../dsp/strutil.js'
 
 import type { AxRerankerIn, AxRerankerOut } from './manager.js'
 
@@ -24,7 +24,7 @@ export class AxDefaultResultReranker extends AxGen<
     const { rankedItems } = await super.forward(ai, input, options)
 
     const sortedIndexes: number[] = rankedItems.map((item) => {
-      const { id: index } = axStringUtil.extractIdAndText(item)
+      const { id: index } = AxStringUtil.extractIdAndText(item)
       return index
     })
 

src/ax/dsp/eval.ts

@@ -140,7 +140,7 @@ function novelF1ScoreOptimized(
   return returnRecall ? recall : f1
 }
 
-export const axEvalUtil = {
+export const AxEvalUtil = {
   emScore,
   f1Score,
   novelF1ScoreOptimized,

src/ax/dsp/strutil.ts

@@ -63,7 +63,7 @@ const batchArray = <T>(arr: readonly T[], size: number): T[][] => {
   return chunkedArr
 }
 
-export const axStringUtil = {
+export const AxStringUtil = {
   trimNonAlphaNum,
   splitIntoTwo,
   dedup,

src/ax/index.ts

@@ -9,6 +9,7 @@ import {
 } from './ai/wrap.js';
 import {
   AxAIAnthropic,
+  axAIAnthropicDefaultConfig,
   type AxAIAnthropicArgs
 } from './ai/anthropic/api.js';
 import {
@@ -31,11 +32,17 @@ import {
 } from './ai/anthropic/types.js';
 import {
   AxAIAzureOpenAI,
+  axAIAzureOpenAIBestConfig,
+  axAIAzureOpenAICreativeConfig,
+  axAIAzureOpenAIDefaultConfig,
+  axAIAzureOpenAIFastConfig,
   type AxAIAzureOpenAIArgs,
   type AxAIAzureOpenAIConfig
 } from './ai/azure-openai/api.js';
 import {
   AxAICohere,
+  axAICohereCreativeConfig,
+  axAICohereDefaultConfig,
   type AxAICohereArgs
 } from './ai/cohere/api.js';
 import {
@@ -52,10 +59,14 @@ import {
 } from './ai/cohere/types.js';
 import {
   AxAIDeepSeek,
+  axAIDeepSeekCodeConfig,
+  axAIDeepSeekDefaultConfig,
   type AxAIDeepSeekArgs
 } from './ai/deepseek/api.js';
 import {
   AxAIGoogleGemini,
+  axAIGoogleGeminiDefaultConfig,
+  axAIGoogleGeminiDefaultCreativeConfig,
   type AxAIGoogleGeminiArgs,
   type AxAIGoogleGeminiOptionsTools
 } from './ai/google-gemini/api.js';
@@ -86,6 +97,8 @@ import {
 } from './ai/groq/api.js';
 import {
   AxAIHuggingFace,
+  axAIHuggingFaceCreativeConfig,
+  axAIHuggingFaceDefaultConfig,
   type AxAIHuggingFaceArgs
 } from './ai/huggingface/api.js';
 import {
@@ -96,6 +109,8 @@ import {
 } from './ai/huggingface/types.js';
 import {
   AxAIMistral,
+  axAIMistralBestConfig,
+  axAIMistralDefaultConfig,
   type AxAIMistralArgs
 } from './ai/mistral/api.js';
 import {
@@ -104,12 +119,18 @@ import {
 } from './ai/mistral/types.js';
 import {
   AxAIOllama,
+  axAIOllamaDefaultConfig,
+  axAIOllamaDefaultCreativeConfig,
   type AxAIOllamaAIConfig,
   type AxAIOllamaArgs
 } from './ai/ollama/api.js';
 import {
   AxAIOpenAI,
   AxAIOpenAIBase,
+  axAIOpenAIBestConfig,
+  axAIOpenAICreativeConfig,
+  axAIOpenAIDefaultConfig,
+  axAIOpenAIFastConfig,
   type AxAIOpenAIArgs,
   type AxAIOpenAIBaseArgs
 } from './ai/openai/api.js';
@@ -128,6 +149,10 @@ import {
 } from './ai/openai/types.js';
 import {
   AxAIReka,
+  axAIRekaBestConfig,
+  axAIRekaCreativeConfig,
+  axAIRekaDefaultConfig,
+  axAIRekaFastConfig,
   type AxAIRekaArgs
 } from './ai/reka/api.js';
 import {
@@ -151,6 +176,7 @@ import {
 } from './util/apicall.js';
 import {
   AxAITogether,
+  axAITogetherDefaultConfig,
   type AxAITogetherArgs
 } from './ai/together/api.js';
 import {
@@ -175,6 +201,8 @@ import {
 } from './ai/balance.js';
 import {
   AxBaseAI,
+  axBaseAIDefaultConfig,
+  axBaseAIDefaultCreativeConfig,
   type AxAIFeatures,
   type AxBaseAIArgs
 } from './ai/base.js';
@@ -252,7 +280,9 @@ import {
 } from './funcs/code.js';
 import {
   AxLLMRequestTypeValues,
-  AxSpanKindValues
+  AxSpanKindValues,
+  axSpanAttributes,
+  axSpanEvents
 } from './trace/trace.js';
 import {
   AxMockAIService,
@@ -342,13 +372,25 @@ import {AxChainOfThought} from './prompts/cot.js';
 import {AxDefaultQueryRewriter} from './docs/rewriter.js';
 import {AxDefaultResultReranker} from './docs/reranker.js';
 import {AxEmbeddingAdapter} from './funcs/embed.js';
+import {AxEvalUtil} from './dsp/eval.js';
 import {AxInstanceRegistry} from './dsp/registry.js';
 import {AxMCPClient} from './mcp/client.js';
 import {AxMCPHTTPTransport} from './mcp/httpTransport.js';
 import {AxMCPStdioTransport} from './mcp/stdioTransport.js';
 import {AxMemory} from './mem/memory.js';
 import {AxMultiServiceRouter} from './ai/multiservice.js';
 import {AxRAG} from './prompts/rag.js';
+import {AxStringUtil} from './dsp/strutil.js';
+import {axModelInfoAnthropic} from './ai/anthropic/info.js';
+import {axModelInfoCohere} from './ai/cohere/info.js';
+import {axModelInfoDeepSeek} from './ai/deepseek/info.js';
+import {axModelInfoGoogleGemini} from './ai/google-gemini/info.js';
+import {axModelInfoGroq} from './ai/groq/info.js';
+import {axModelInfoHuggingFace} from './ai/huggingface/info.js';
+import {axModelInfoMistral} from './ai/mistral/info.js';
+import {axModelInfoOpenAI} from './ai/openai/info.js';
+import {axModelInfoReka} from './ai/reka/info.js';
+import {axModelInfoTogether} from './ai/together/info.js';
 import {type AxAIMemory} from './mem/types.js';
 import {type AxMCPTransport} from './mcp/transport.js';
 
@@ -408,6 +450,7 @@ export { AxDefaultQueryRewriter };
 export { AxDefaultResultReranker };
 export { AxDockerSession };
 export { AxEmbeddingAdapter };
+export { AxEvalUtil };
 export { AxFunctionError };
 export { AxFunctionProcessor };
 export { AxGen };
@@ -431,7 +474,48 @@ export { AxSignature };
 export { AxSimpleClassifier };
 export { AxSimpleClassifierClass };
 export { AxSpanKindValues };
+export { AxStringUtil };
 export { AxTestPrompt };
+export { axAIAnthropicDefaultConfig };
+export { axAIAzureOpenAIBestConfig };
+export { axAIAzureOpenAICreativeConfig };
+export { axAIAzureOpenAIDefaultConfig };
+export { axAIAzureOpenAIFastConfig };
+export { axAICohereCreativeConfig };
+export { axAICohereDefaultConfig };
+export { axAIDeepSeekCodeConfig };
+export { axAIDeepSeekDefaultConfig };
+export { axAIGoogleGeminiDefaultConfig };
+export { axAIGoogleGeminiDefaultCreativeConfig };
+export { axAIHuggingFaceCreativeConfig };
+export { axAIHuggingFaceDefaultConfig };
+export { axAIMistralBestConfig };
+export { axAIMistralDefaultConfig };
+export { axAIOllamaDefaultConfig };
+export { axAIOllamaDefaultCreativeConfig };
+export { axAIOpenAIBestConfig };
+export { axAIOpenAICreativeConfig };
+export { axAIOpenAIDefaultConfig };
+export { axAIOpenAIFastConfig };
+export { axAIRekaBestConfig };
+export { axAIRekaCreativeConfig };
+export { axAIRekaDefaultConfig };
+export { axAIRekaFastConfig };
+export { axAITogetherDefaultConfig };
+export { axBaseAIDefaultConfig };
+export { axBaseAIDefaultCreativeConfig };
+export { axModelInfoAnthropic };
+export { axModelInfoCohere };
+export { axModelInfoDeepSeek };
+export { axModelInfoGoogleGemini };
+export { axModelInfoGroq };
+export { axModelInfoHuggingFace };
+export { axModelInfoMistral };
+export { axModelInfoOpenAI };
+export { axModelInfoReka };
+export { axModelInfoTogether };
+export { axSpanAttributes };
+export { axSpanEvents };
 
 // Type exports
 export type { AxAIAnthropicArgs };