@discomedia/utils 1.0.46 → 1.0.51

package/dist/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.0.46",
+  "version": "1.0.51",
   "author": "Disco Media",
   "description": "Utility functions used in Disco Media apps",
   "always-build-npm": true,
@@ -33,7 +33,7 @@
   },
   "dependencies": {
     "dotenv": "^17.2.3",
-    "openai": "^6.9.1",
+    "openai": "^6.15.0",
     "p-limit": "^7.2.0",
     "tslib": "^2.8.1",
     "ws": "^8.18.3"
@@ -45,8 +45,8 @@
     "@rollup/plugin-node-resolve": "^16.0.3",
     "@rollup/plugin-typescript": "^12.3.0",
     "@types/ws": "^8.18.1",
-    "lightweight-charts": "^5.0.9",
-    "rollup": "^4.53.2",
+    "lightweight-charts": "^5.1.0",
+    "rollup": "^4.53.5",
     "typescript": "^5.9.3"
   }
 }

package/dist/test.js

@@ -447,9 +447,10 @@ function getLastFullTradingDateImpl(currentDate = new Date()) {
         if (calendar.isEarlyCloseDay(prevMarketDay)) {
             prevCloseMinutes = MARKET_CONFIG.TIMES.EARLY_CLOSE.hour * 60 + MARKET_CONFIG.TIMES.EARLY_CLOSE.minute;
         }
-        const year = prevMarketDay.getUTCFullYear();
-        const month = prevMarketDay.getUTCMonth();
-        const day = prevMarketDay.getUTCDate();
+        const prevNYDate = toNYTime(prevMarketDay);
+        const year = prevNYDate.getUTCFullYear();
+        const month = prevNYDate.getUTCMonth();
+        const day = prevNYDate.getUTCDate();
         const closeHour = Math.floor(prevCloseMinutes / 60);
         const closeMinute = prevCloseMinutes % 60;
         return fromNYTime(new Date(Date.UTC(year, month, day, closeHour, closeMinute, 0, 0)));
@@ -963,6 +964,61 @@ function countTradingDays(startDate, endDate = new Date()) {
         minutes: Math.round(minutes),
     };
 }
+/**
+ * Returns the trading day N days back from a reference date, along with its market open time.
+ * Trading days are counted as full or half trading days (days that end count as 1 full trading day).
+ *
+ * @param options - Object with:
+ *   - referenceDate: Date to count back from (default: now)
+ *   - days: Number of trading days to go back (must be >= 1)
+ * @returns Object containing:
+ *   - date: Trading date in YYYY-MM-DD format
+ *   - marketOpenISO: Market open time as ISO string (e.g., "2025-11-15T13:30:00.000Z")
+ *   - unixTimestamp: Market open time as Unix timestamp in seconds
+ * @example
+ * ```typescript
+ * // Get the trading day 1 day back (most recent full trading day)
+ * const result = getTradingDaysBack({ days: 1 });
+ * console.log(result.date); // "2025-11-01"
+ * console.log(result.marketOpenISO); // "2025-11-01T13:30:00.000Z"
+ * console.log(result.unixTimestamp); // 1730466600
+ *
+ * // Get the trading day 5 days back from a specific date
+ * const result2 = getTradingDaysBack({
+ *   referenceDate: new Date('2025-11-15T12:00:00-05:00'),
+ *   days: 5
+ * });
+ * ```
+ */
+function getTradingDaysBack(options) {
+    const calendar = new MarketCalendar();
+    const refDate = options.referenceDate || new Date();
+    const daysBack = options.days;
+    if (daysBack < 1) {
+        throw new Error('days must be at least 1');
+    }
+    // Start from the last full trading date relative to reference
+    let targetDate = getLastFullTradingDateImpl(refDate);
+    // Go back the specified number of days (we're already at day 1, so go back days-1 more)
+    for (let i = 1; i < daysBack; i++) {
+        targetDate = calendar.getPreviousMarketDay(targetDate);
+    }
+    // Get market open time for this date
+    const marketTimes = getMarketTimes(targetDate);
+    if (!marketTimes.open) {
+        throw new Error(`No market open time for target date`);
+    }
+    // Format the date string (YYYY-MM-DD) in NY time
+    const nyDate = toNYTime(marketTimes.open);
+    const dateStr = `${nyDate.getUTCFullYear()}-${String(nyDate.getUTCMonth() + 1).padStart(2, '0')}-${String(nyDate.getUTCDate()).padStart(2, '0')}`;
+    const marketOpenISO = marketTimes.open.toISOString();
+    const unixTimestamp = Math.floor(marketTimes.open.getTime() / 1000);
+    return {
+        date: dateStr,
+        marketOpenISO,
+        unixTimestamp,
+    };
+}
 // Export MARKET_TIMES for compatibility
 const MARKET_TIMES = {
     TIMEZONE: MARKET_CONFIG.TIMEZONE,
@@ -1085,6 +1141,10 @@ function isOpenRouterModel(model) {
         'openai/gpt-5-mini',
         'openai/gpt-5-nano',
         'openai/gpt-5.1',
+        'openai/gpt-5.2',
+        'openai/gpt-5.2-pro',
+        'openai/gpt-5.1-codex',
+        'openai/gpt-5.1-codex-max',
         'openai/gpt-oss-120b',
         'z.ai/glm-4.5',
         'z.ai/glm-4.5-air',
@@ -2450,7 +2510,7 @@ const safeJSON = (text) => {
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
 
-const VERSION = '6.9.1'; // x-release-please-version
+const VERSION = '6.15.0'; // x-release-please-version
 
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 const isRunningInBrowser = () => {
@@ -8291,6 +8351,19 @@ class Responses extends APIResource {
     cancel(responseID, options) {
         return this._client.post(path `/responses/${responseID}/cancel`, options);
     }
+    /**
+     * Compact conversation
+     *
+     * @example
+     * ```ts
+     * const compactedResponse = await client.responses.compact({
+     *   model: 'gpt-5.2',
+     * });
+     * ```
+     */
+    compact(body, options) {
+        return this._client.post('/responses/compact', { body, ...options });
+    }
 }
 Responses.InputItems = InputItems;
 Responses.InputTokens = InputTokens;
@@ -9433,6 +9506,22 @@ const openAiModelCosts = {
         inputCost: 1.25 / 1_000_000,
         outputCost: 10 / 1_000_000,
     },
+    'gpt-5.2': {
+        inputCost: 1.5 / 1_000_000,
+        outputCost: 12 / 1_000_000,
+    },
+    'gpt-5.2-pro': {
+        inputCost: 3 / 1_000_000,
+        outputCost: 24 / 1_000_000,
+    },
+    'gpt-5.1-codex': {
+        inputCost: 1.1 / 1_000_000,
+        outputCost: 8.8 / 1_000_000,
+    },
+    'gpt-5.1-codex-max': {
+        inputCost: 1.8 / 1_000_000,
+        outputCost: 14.4 / 1_000_000,
+    },
     'o4-mini': {
         inputCost: 1.1 / 1_000_000,
         outputCost: 4.4 / 1_000_000,
@@ -9453,6 +9542,7 @@ const deepseekModelCosts = {
 /** Image generation costs in USD per image. Based on OpenAI pricing as of Feb 2025. */
 const openAiImageCosts = {
     'gpt-image-1': 0.0075, // $0.0075 per image for gpt-image-1
+    'gpt-image-1.5': 0.0075, // Assumes parity pricing with gpt-image-1 until OpenAI publishes updated rates
 };
 /**
  * Calculates the cost of generating images using OpenAI's Images API.
@@ -9462,10 +9552,12 @@ const openAiImageCosts = {
  * @returns The cost of generating the images in USD.
  */
 function calculateImageCost(model, imageCount) {
-    if (typeof imageCount !== 'number' || imageCount <= 0) {
+    if (typeof model !== 'string' || typeof imageCount !== 'number' || imageCount <= 0) {
         return 0;
     }
     const costPerImage = openAiImageCosts[model];
+    if (!costPerImage)
+        return 0;
     return imageCount * costPerImage;
 }
 /**
@@ -9894,6 +9986,10 @@ const isSupportedModel = (model) => {
         'gpt-5-mini',
         'gpt-5-nano',
         'gpt-5.1',
+        'gpt-5.2',
+        'gpt-5.2-pro',
+        'gpt-5.1-codex',
+        'gpt-5.1-codex-max',
         'o4-mini',
         'o3',
     ].includes(model);
@@ -9906,7 +10002,21 @@ const isSupportedModel = (model) => {
 function supportsTemperature(model) {
     // Reasoning models don't support temperature
     // GPT-5 models also do not support temperature
-    const reasoningAndGPT5Models = ['o1', 'o1-mini', 'o3-mini', 'o4-mini', 'o3', 'gpt-5', 'gpt-5-mini', 'gpt-5-nano', 'gpt-5.1'];
+    const reasoningAndGPT5Models = [
+        'o1',
+        'o1-mini',
+        'o3-mini',
+        'o4-mini',
+        'o3',
+        'gpt-5',
+        'gpt-5-mini',
+        'gpt-5-nano',
+        'gpt-5.1',
+        'gpt-5.2',
+        'gpt-5.2-pro',
+        'gpt-5.1-codex',
+        'gpt-5.1-codex-max',
+    ];
     return !reasoningAndGPT5Models.includes(model);
 }
 /**
@@ -9924,7 +10034,16 @@ function isReasoningModel(model) {
  * @returns True if the model is a GPT-5 model, false otherwise.
  */
 function isGPT5Model(model) {
-    const gpt5Models = ['gpt-5', 'gpt-5-mini', 'gpt-5-nano', 'gpt-5.1'];
+    const gpt5Models = [
+        'gpt-5',
+        'gpt-5-mini',
+        'gpt-5-nano',
+        'gpt-5.1',
+        'gpt-5.2',
+        'gpt-5.2-pro',
+        'gpt-5.1-codex',
+        'gpt-5.1-codex-max',
+    ];
     return gpt5Models.includes(model);
 }
 /**
@@ -10165,6 +10284,20 @@ async function makeLLMCall(input, options = {}) {
     return await makeResponsesAPICall(processedInput, responsesOptions);
 }
 
+const DEFAULT_IMAGE_MODEL = 'gpt-image-1.5';
+const resolveImageModel = (model) => model ?? DEFAULT_IMAGE_MODEL;
+const MULTIMODAL_VISION_MODELS = new Set([
+    'gpt-4o-mini',
+    'gpt-4o',
+    'gpt-5',
+    'gpt-5-mini',
+    'gpt-5-nano',
+    'gpt-5.1',
+    'gpt-5.2',
+    'gpt-5.2-pro',
+    'gpt-5.1-codex',
+    'gpt-5.1-codex-max',
+]);
 /**
  * Makes a call to the OpenAI Images API to generate images based on a text prompt.
  *
@@ -10201,7 +10334,12 @@ async function makeLLMCall(input, options = {}) {
  * @throws Error if the API call fails or invalid parameters are provided
  */
 async function makeImagesCall(prompt, options = {}) {
-    const { size = 'auto', outputFormat = 'webp', compression = 50, quality = 'high', count = 1, background = 'auto', moderation = 'auto', apiKey } = options;
+    const { model, size = 'auto', outputFormat = 'webp', compression = 50, quality = 'high', count = 1, background = 'auto', moderation = 'auto', apiKey, visionModel, } = options;
+    const imageModel = resolveImageModel(model);
+    const supportedVisionModel = visionModel && MULTIMODAL_VISION_MODELS.has(visionModel) ? visionModel : undefined;
+    if (visionModel && !supportedVisionModel) {
+        console.warn(`Vision model ${visionModel} is not recognized as a multimodal OpenAI model. Ignoring for image usage metadata.`);
+    }
     // Get API key
     const effectiveApiKey = apiKey || process.env.OPENAI_API_KEY;
     if (!effectiveApiKey) {
@@ -10222,7 +10360,7 @@ async function makeImagesCall(prompt, options = {}) {
     });
     // Build the request parameters using OpenAI's type
     const requestParams = {
-        model: 'gpt-image-1',
+        model: imageModel,
         prompt,
         n: count || 1,
         size: size || 'auto',
@@ -10246,7 +10384,7 @@ async function makeImagesCall(prompt, options = {}) {
             throw new Error('No images returned from OpenAI Images API');
         }
         // Calculate cost
-        const cost = calculateImageCost('gpt-image-1', count || 1);
+        const cost = calculateImageCost(imageModel, count || 1);
         // Return the response with enhanced usage information
         const enhancedResponse = {
             ...response,
@@ -10259,8 +10397,9 @@ async function makeImagesCall(prompt, options = {}) {
                     total_tokens: 0,
                 }),
                 provider: 'openai',
-                model: 'gpt-image-1',
+                model: imageModel,
                 cost,
+                ...(supportedVisionModel ? { visionModel: supportedVisionModel } : {}),
             },
         };
         return enhancedResponse;
@@ -19387,6 +19526,7 @@ const disco = {
         getNYTimeZone: getNYTimeZone,
         getTradingDate: getTradingDate,
         getTradingStartAndEndDates: getTradingStartAndEndDates,
+        getTradingDaysBack: getTradingDaysBack,
         isMarketDay: isMarketDay,
         isWithinMarketHours: isWithinMarketHours,
         countTradingDays: countTradingDays,
@@ -19401,47 +19541,151 @@ const disco = {
 };
 
 // Test file for context functionality
-async function testLLM() {
-    //const models: LLMModel[] = ['gpt-5', 'gpt-5-mini', 'gpt-5-nano'];
-    const models = ['gpt-5.1'];
-    for (const model of models) {
-        console.log(`\nTesting model: ${model}`);
-        // 1. Basic call
-        try {
-            const basic = await disco.llm.call('What is the capital of France?', { model });
-            if (!basic || !basic.response) {
-                throw new Error('No response from LLM');
-            }
-            console.log(`Response: ${basic.response}`);
-        }
-        catch (e) {
-            console.error(`  Basic call error:`, e);
-        }
-        // 2. JSON call
+function testGetTradingDaysBack() {
+    const testCases = [
+        {
+            label: '1 day back from Friday after close',
+            referenceDate: '2025-07-11T18:00:00-04:00',
+            days: 1,
+            expected: {
+                date: '2025-07-11',
+                marketOpenISO: '2025-07-11T13:30:00.000Z',
+            },
+        },
+        {
+            label: '1 day back from Friday during market',
+            referenceDate: '2025-07-11T10:00:00-04:00',
+            days: 1,
+            expected: {
+                date: '2025-07-10',
+                marketOpenISO: '2025-07-10T13:30:00.000Z',
+            },
+        },
+        {
+            label: '2 days back from Friday after close',
+            referenceDate: '2025-07-11T18:00:00-04:00',
+            days: 2,
+            expected: {
+                date: '2025-07-10',
+                marketOpenISO: '2025-07-10T13:30:00.000Z',
+            },
+        },
+        {
+            label: '5 days back from Friday after close',
+            referenceDate: '2025-07-11T18:00:00-04:00',
+            days: 5,
+            expected: {
+                date: '2025-07-07',
+                marketOpenISO: '2025-07-07T13:30:00.000Z',
+            },
+        },
+        {
+            label: '6 days back from Friday after close (skips weekend and July 4)',
+            referenceDate: '2025-07-11T18:00:00-04:00',
+            days: 6,
+            expected: {
+                date: '2025-07-03',
+                marketOpenISO: '2025-07-03T13:30:00.000Z',
+            },
+        },
+        {
+            label: '1 day back from Monday after close (should be Monday)',
+            referenceDate: '2025-07-14T18:00:00-04:00',
+            days: 1,
+            expected: {
+                date: '2025-07-14',
+                marketOpenISO: '2025-07-14T13:30:00.000Z',
+            },
+        },
+        {
+            label: '1 day back from Saturday (should be Friday)',
+            referenceDate: '2025-07-12T12:00:00-04:00',
+            days: 1,
+            expected: {
+                date: '2025-07-11',
+                marketOpenISO: '2025-07-11T13:30:00.000Z',
+            },
+        },
+        {
+            label: '3 days back from Monday (skips weekend)',
+            referenceDate: '2025-07-14T18:00:00-04:00',
+            days: 3,
+            expected: {
+                date: '2025-07-10',
+                marketOpenISO: '2025-07-10T13:30:00.000Z',
+            },
+        },
+        {
+            label: '1 day back from day after holiday (July 7, after July 4)',
+            referenceDate: '2025-07-07T18:00:00-04:00',
+            days: 1,
+            expected: {
+                date: '2025-07-07',
+                marketOpenISO: '2025-07-07T13:30:00.000Z',
+            },
+        },
+        {
+            label: '2 days back from day after holiday (skips July 4)',
+            referenceDate: '2025-07-07T18:00:00-04:00',
+            days: 2,
+            expected: {
+                date: '2025-07-03',
+                marketOpenISO: '2025-07-03T13:30:00.000Z',
+            },
+        },
+        {
+            label: 'Early close day (July 3) - 1 day back from July 7',
+            referenceDate: '2025-07-07T18:00:00-04:00',
+            days: 2,
+            expected: {
+                date: '2025-07-03',
+                marketOpenISO: '2025-07-03T13:30:00.000Z', // Still opens at normal time
+            },
+        },
+    ];
+    console.log('\n=== Testing getTradingDaysBack ===\n');
+    for (const { label, referenceDate, days, expected } of testCases) {
         try {
-            const jsonPrompt = 'Return a JSON object with keys country and capital for France.';
-            const json = await disco.llm.call(jsonPrompt, { model, responseFormat: 'json' });
-            if (!json || !json.response) {
-                throw new Error('No response from LLM');
+            const refDate = new Date(referenceDate);
+            const result = disco.time.getTradingDaysBack({ referenceDate: refDate, days });
+            const dateMatches = result.date === expected.date;
+            const isoMatches = result.marketOpenISO === expected.marketOpenISO;
+            const unixValid = typeof result.unixTimestamp === 'number' && result.unixTimestamp > 0;
+            // Verify unix timestamp matches ISO
+            const dateFromUnix = new Date(result.unixTimestamp * 1000).toISOString();
+            const unixMatches = dateFromUnix === result.marketOpenISO;
+            const allPass = dateMatches && isoMatches && unixValid && unixMatches;
+            console.log(`${allPass ? '✅' : '❌'} ${label}`);
+            if (!allPass) {
+                console.log(`  Expected: ${JSON.stringify(expected)}`);
+                console.log(`  Got: ${JSON.stringify(result)}`);
+                if (!dateMatches)
+                    console.log(`  ❌ Date mismatch`);
+                if (!isoMatches)
+                    console.log(`  ❌ ISO mismatch`);
+                if (!unixValid)
+                    console.log(`  ❌ Unix timestamp invalid`);
+                if (!unixMatches)
+                    console.log(`  ❌ Unix timestamp doesn't match ISO`);
             }
-            console.log(`Response: ${JSON.stringify(json.response)}`);
         }
-        catch (e) {
-            console.error(`  JSON call error:`, e);
+        catch (error) {
+            console.log(`❌ ${label}`);
+            console.log(`  Error: ${error instanceof Error ? error.message : String(error)}`);
         }
-        // 3. Web search
+    }
+    const totalTests = testCases.length;
+    const passedTests = testCases.filter((tc) => {
         try {
-            const searchPrompt = 'What is the latest news about artificial intelligence? Respond with 3 sentences max.';
-            const tool = await disco.llm.call(searchPrompt, { model, useWebSearch: true });
-            if (!tool || !tool.response) {
-                throw new Error('No response from LLM');
-            }
-            console.log(`Response: ${tool.response}`);
+            const refDate = new Date(tc.referenceDate);
+            const result = disco.time.getTradingDaysBack({ referenceDate: refDate, days: tc.days });
+            return result.date === tc.expected.date && result.marketOpenISO === tc.expected.marketOpenISO;
         }
-        catch (e) {
-            console.error(`  Web search error:`, e);
+        catch {
+            return false;
         }
-    }
+    }).length;
+    console.log(`\n=== Summary: ${passedTests}/${totalTests} tests passed ===\n`);
 }
 // testGetTradingDate();
 // testGetTradingStartAndEndDates();
@@ -19458,5 +19702,7 @@ async function testLLM() {
 // testWebSocketConnectAndDisconnect();
 // testGetAssetsShortableFilter();
 // testMarketDataAPI();
-testLLM();
+// testLLM();
+// testImageModelDefaults();
+testGetTradingDaysBack();
 //# sourceMappingURL=test.js.map