@juspay/yama 1.1.0 → 1.2.0

package/dist/utils/ProviderLimits.d.ts

@@ -0,0 +1,58 @@
+/**
+ * Provider Token Limits Utility
+ * Centralized management of AI provider token limits and validation
+ */
+/**
+ * AI Provider types supported by the system
+ */
+export type AIProvider = 'vertex' | 'google-ai' | 'gemini' | 'openai' | 'gpt-4' | 'anthropic' | 'claude' | 'azure' | 'bedrock' | 'auto';
+/**
+ * Provider token limits configuration
+ * These limits are conservative values to avoid API errors
+ */
+export declare const PROVIDER_TOKEN_LIMITS: Record<AIProvider, number>;
+/**
+ * Conservative limits used by CodeReviewer for safety
+ * These are slightly lower than the actual limits to provide buffer
+ */
+export declare const CONSERVATIVE_PROVIDER_LIMITS: Record<AIProvider, number>;
+/**
+ * Get the token limit for a specific provider
+ * @param provider - The AI provider name
+ * @param conservative - Whether to use conservative limits (default: false)
+ * @returns The token limit for the provider
+ */
+export declare function getProviderTokenLimit(provider: string, conservative?: boolean): number;
+/**
+ * Validate and adjust token limit for a provider
+ * @param provider - The AI provider name
+ * @param configuredTokens - The configured token limit
+ * @param conservative - Whether to use conservative limits (default: false)
+ * @returns The validated and potentially adjusted token limit
+ */
+export declare function validateProviderTokenLimit(provider: string, configuredTokens: number | undefined, conservative?: boolean): number;
+/**
+ * Get all supported providers
+ * @returns Array of supported provider names
+ */
+export declare function getSupportedProviders(): AIProvider[];
+/**
+ * Check if a provider is supported
+ * @param provider - The provider name to check
+ * @returns True if the provider is supported
+ */
+export declare function isProviderSupported(provider: string): boolean;
+/**
+ * Get provider information including limits and support status
+ * @param provider - The provider name
+ * @param conservative - Whether to use conservative limits
+ * @returns Provider information object
+ */
+export declare function getProviderInfo(provider: string, conservative?: boolean): {
+    provider: string;
+    isSupported: boolean;
+    tokenLimit: number;
+    conservativeLimit: number;
+    standardLimit: number;
+};
+//# sourceMappingURL=ProviderLimits.d.ts.map

package/dist/utils/ProviderLimits.js

@@ -0,0 +1,143 @@
+/**
+ * Provider Token Limits Utility
+ * Centralized management of AI provider token limits and validation
+ */
+import { logger } from "./Logger.js";
+/**
+ * Provider token limits configuration
+ * These limits are conservative values to avoid API errors
+ */
+export const PROVIDER_TOKEN_LIMITS = {
+    // Google/Vertex AI providers
+    'vertex': 65536, // Vertex AI limit is 65537 exclusive = 65536 max
+    'google-ai': 65536, // Google AI Studio limit
+    'gemini': 65536, // Gemini model limit
+    // OpenAI providers
+    'openai': 128000, // OpenAI GPT-4 and newer models
+    'gpt-4': 128000, // GPT-4 specific limit
+    // Anthropic providers
+    'anthropic': 200000, // Claude models limit
+    'claude': 200000, // Claude specific limit
+    // Microsoft Azure
+    'azure': 128000, // Azure OpenAI limit
+    // AWS Bedrock
+    'bedrock': 100000, // AWS Bedrock limit
+    // Auto-selection mode (conservative default)
+    'auto': 60000, // Conservative default for auto-selection
+};
+/**
+ * Conservative limits used by CodeReviewer for safety
+ * These are slightly lower than the actual limits to provide buffer
+ */
+export const CONSERVATIVE_PROVIDER_LIMITS = {
+    'vertex': 65536,
+    'google-ai': 65536,
+    'gemini': 65536,
+    'openai': 120000, // Slightly lower for safety
+    'gpt-4': 120000,
+    'anthropic': 190000, // Slightly lower for safety
+    'claude': 190000,
+    'azure': 120000,
+    'bedrock': 95000, // Significantly lower for safety
+    'auto': 60000,
+};
+/**
+ * Get the token limit for a specific provider
+ * @param provider - The AI provider name
+ * @param conservative - Whether to use conservative limits (default: false)
+ * @returns The token limit for the provider
+ */
+export function getProviderTokenLimit(provider, conservative = false) {
+    // Handle null, undefined, or empty string
+    if (!provider || typeof provider !== 'string') {
+        return conservative ? CONSERVATIVE_PROVIDER_LIMITS.auto : PROVIDER_TOKEN_LIMITS.auto;
+    }
+    const normalizedProvider = provider.toLowerCase();
+    const limits = conservative ? CONSERVATIVE_PROVIDER_LIMITS : PROVIDER_TOKEN_LIMITS;
+    // Handle empty string after normalization
+    if (normalizedProvider === '') {
+        return conservative ? CONSERVATIVE_PROVIDER_LIMITS.auto : PROVIDER_TOKEN_LIMITS.auto;
+    }
+    // Direct match
+    if (normalizedProvider in limits) {
+        return limits[normalizedProvider];
+    }
+    // Partial match - check if provider contains any known provider name
+    for (const [key, limit] of Object.entries(limits)) {
+        if (normalizedProvider.includes(key) || key.includes(normalizedProvider)) {
+            return limit;
+        }
+    }
+    // Default fallback
+    return conservative ? CONSERVATIVE_PROVIDER_LIMITS.auto : PROVIDER_TOKEN_LIMITS.auto;
+}
+/**
+ * Validate and adjust token limit for a provider
+ * @param provider - The AI provider name
+ * @param configuredTokens - The configured token limit
+ * @param conservative - Whether to use conservative limits (default: false)
+ * @returns The validated and potentially adjusted token limit
+ */
+export function validateProviderTokenLimit(provider, configuredTokens, conservative = false) {
+    const providerLimit = getProviderTokenLimit(provider, conservative);
+    if (!configuredTokens || configuredTokens <= 0) {
+        logger.debug(`No configured tokens for ${provider}, using provider default: ${providerLimit}`);
+        return providerLimit;
+    }
+    if (configuredTokens > providerLimit) {
+        logger.warn(`Configured maxTokens (${configuredTokens}) exceeds ${provider} limit (${providerLimit}). Adjusting to ${providerLimit}.`);
+        return providerLimit;
+    }
+    logger.debug(`Token limit validation passed: ${configuredTokens} <= ${providerLimit} for provider ${provider}`);
+    return configuredTokens;
+}
+/**
+ * Get all supported providers
+ * @returns Array of supported provider names
+ */
+export function getSupportedProviders() {
+    return Object.keys(PROVIDER_TOKEN_LIMITS);
+}
+/**
+ * Check if a provider is supported
+ * @param provider - The provider name to check
+ * @returns True if the provider is supported
+ */
+export function isProviderSupported(provider) {
+    // Handle null, undefined, or empty string
+    if (!provider || typeof provider !== 'string') {
+        return false;
+    }
+    const normalizedProvider = provider.toLowerCase();
+    // Handle empty string after normalization
+    if (normalizedProvider === '') {
+        return false;
+    }
+    // Check direct match
+    if (normalizedProvider in PROVIDER_TOKEN_LIMITS) {
+        return true;
+    }
+    // Check partial match
+    for (const key of Object.keys(PROVIDER_TOKEN_LIMITS)) {
+        if (normalizedProvider.includes(key) || key.includes(normalizedProvider)) {
+            return true;
+        }
+    }
+    return false;
+}
+/**
+ * Get provider information including limits and support status
+ * @param provider - The provider name
+ * @param conservative - Whether to use conservative limits
+ * @returns Provider information object
+ */
+export function getProviderInfo(provider, conservative = false) {
+    return {
+        provider,
+        isSupported: isProviderSupported(provider),
+        tokenLimit: getProviderTokenLimit(provider, conservative),
+        conservativeLimit: getProviderTokenLimit(provider, true),
+        standardLimit: getProviderTokenLimit(provider, false),
+    };
+}
+//# sourceMappingURL=ProviderLimits.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@juspay/yama",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "Enterprise-grade Pull Request automation toolkit with AI-powered code review and description enhancement",
   "keywords": [
     "pr",
@@ -52,7 +52,8 @@
     "build": "tsc && tsc-alias",
     "dev": "ts-node-dev --respawn --transpile-only src/cli/index.ts",
     "test": "jest",
-    "lint": "eslint src/**/*.ts",
+    "lint": "eslint .",
+    "lint:fix": "eslint . --fix",
     "type-check": "tsc --noEmit",
     "format": "prettier --write .",
     "format:check": "prettier --check .",
@@ -63,7 +64,6 @@
     "changeset": "changeset",
     "changeset:version": "changeset version && git add --all",
     "release": "npm run build && npm run test && changeset publish",
-    "release:check": "npm run build && publint && size-limit",
     "release:dry": "npm publish --dry-run",
     "release:github": "npm publish --registry https://npm.pkg.github.com",
     "version:check": "npm version --no-git-tag-version",
@@ -89,9 +89,10 @@
     "@types/jest": "^29.0.0",
     "@types/lodash": "^4.14.0",
     "@types/node": "^20.0.0",
-    "@typescript-eslint/eslint-plugin": "^6.0.0",
-    "@typescript-eslint/parser": "^6.0.0",
-    "eslint": "^8.0.0",
+    "@eslint/js": "^9.0.0",
+    "@typescript-eslint/eslint-plugin": "^8.0.0",
+    "@typescript-eslint/parser": "^8.0.0",
+    "eslint": "^9.0.0",
     "jest": "^29.0.0",
     "rimraf": "^5.0.0",
     "ts-jest": "^29.0.0",

package/yama.config.example.yaml

@@ -4,21 +4,21 @@
 # AI Provider Configuration
 providers:
   ai:
-    provider: "auto"  # Options: auto, google-ai, openai, anthropic, azure, bedrock
-    model: "best"     # Model name or "best" for auto-selection
-    temperature: 0.3  # Lower = more focused (0.0-1.0)
-    maxTokens: 2000000  # Maximum tokens for response
-    timeout: "15m"    # Timeout for AI operations
+    provider: "auto" # Options: auto, google-ai, openai, anthropic, azure, bedrock
+    model: "best" # Model name or "best" for auto-selection
+    temperature: 0.3 # Lower = more focused (0.0-1.0)
+    maxTokens: 60000 # Maximum tokens for response (provider-aware limits will be applied automatically)
+    timeout: "15m" # Timeout for AI operations
     enableAnalytics: true
     enableEvaluation: false
 
   # Git Platform Configuration
   git:
-    platform: "bitbucket"  # Options: bitbucket, github, gitlab, azure-devops
+    platform: "bitbucket" # Options: bitbucket, github, gitlab, azure-devops
     credentials:
-      username: "${BITBUCKET_USERNAME}"  # Environment variable
-      token: "${BITBUCKET_TOKEN}"        # Environment variable
-      baseUrl: "${BITBUCKET_BASE_URL}"   # Your Bitbucket server URL
+      username: "${BITBUCKET_USERNAME}" # Environment variable
+      token: "${BITBUCKET_TOKEN}" # Environment variable
+      baseUrl: "${BITBUCKET_BASE_URL}" # Your Bitbucket server URL
 
 # Feature Configuration
 features:
@@ -26,7 +26,14 @@ features:
   codeReview:
     enabled: true
     severityLevels: ["CRITICAL", "MAJOR", "MINOR", "SUGGESTION"]
-    categories: ["security", "performance", "maintainability", "functionality", "error_handling"]
+    categories:
+      [
+        "security",
+        "performance",
+        "maintainability",
+        "functionality",
+        "error_handling",
+      ]
     excludePatterns:
       - "*.lock"
       - "*.svg"
@@ -38,7 +45,7 @@ features:
       - "dist/**"
       - "build/**"
       - "vendor/**"
-    contextLines: 3  # Lines of context around changes
+    contextLines: 3 # Lines of context around changes
     focusAreas:
       - "Security vulnerabilities"
       - "Performance bottlenecks"
@@ -48,7 +55,7 @@ features:
   # Description Enhancement Configuration
   descriptionEnhancement:
     enabled: true
-    preserveContent: true  # Always preserve existing content
+    preserveContent: true # Always preserve existing content
     autoFormat: true
     requiredSections:
       - key: "changelog"
@@ -65,36 +72,36 @@ features:
   diffStrategy:
     enabled: true
     thresholds:
-      wholeDiffMaxFiles: 2      # Use whole diff for ≤2 files
-      fileByFileMinFiles: 3     # Use file-by-file for ≥3 files
+      wholeDiffMaxFiles: 2 # Use whole diff for ≤2 files
+      fileByFileMinFiles: 3 # Use file-by-file for ≥3 files
     # Optional: Force a specific strategy regardless of file count
     # forceStrategy: "file-by-file"  # Options: whole, file-by-file, auto
 
   # Security Scan Configuration (Future)
   securityScan:
     enabled: false
-    level: "strict"  # Options: strict, moderate, basic
+    level: "strict" # Options: strict, moderate, basic
     scanTypes: ["dependencies", "secrets", "vulnerabilities"]
 
   # Analytics Configuration (Future)
   analytics:
     enabled: false
     trackMetrics: true
-    exportFormat: "json"  # Options: json, csv, yaml
+    exportFormat: "json" # Options: json, csv, yaml
 
 # Cache Configuration
 cache:
   enabled: true
-  ttl: "30m"  # Time to live for cache entries
+  ttl: "30m" # Time to live for cache entries
   maxSize: "100MB"
-  storage: "memory"  # Options: memory, redis, file
+  storage: "memory" # Options: memory, redis, file
 
 # Performance Configuration
 performance:
   batch:
     enabled: true
-    maxConcurrent: 5      # Max concurrent API calls
-    delayBetween: "1s"    # Delay between batches
+    maxConcurrent: 5 # Max concurrent API calls
+    delayBetween: "1s" # Delay between batches
   optimization:
     reuseConnections: true
     compressRequests: false
@@ -108,7 +115,7 @@ rules:
       severity: "CRITICAL"
       message: "Hardcoded secrets detected"
       suggestion: "Use environment variables or secure configuration"
-    
+
     - name: "SQL injection prevention"
       pattern: "query\\([^?]+\\+.*\\)"
       severity: "CRITICAL"
@@ -134,3 +141,12 @@ monitoring:
   metrics: ["api_calls", "cache_hits", "processing_time"]
   exportFormat: "prometheus"
   interval: "1m"
+
+# Memory Bank Configuration
+memoryBank:
+  enabled: true
+  path: "memory-bank"  # Primary path to look for memory bank files
+  fallbackPaths:       # Optional fallback paths if primary doesn't exist
+    - "docs/memory-bank"
+    - ".memory-bank"
+    - "project-docs/context"