@discomedia/utils 1.0.46 → 1.0.51

package/dist/index.cjs

@@ -449,9 +449,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)));
@@ -965,6 +966,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,
@@ -1087,6 +1143,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',
@@ -2452,7 +2512,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 = () => {
@@ -8293,6 +8353,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;
@@ -9435,6 +9508,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,
@@ -9455,6 +9544,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.
@@ -9464,10 +9554,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;
 }
 /**
@@ -9896,6 +9988,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);
@@ -9908,7 +10004,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);
 }
 /**
@@ -9926,7 +10036,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);
 }
 /**
@@ -10167,6 +10286,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.
  *
@@ -10203,7 +10336,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) {
@@ -10224,7 +10362,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',
@@ -10248,7 +10386,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,
@@ -10261,8 +10399,9 @@ async function makeImagesCall(prompt, options = {}) {
                     total_tokens: 0,
                 }),
                 provider: 'openai',
-                model: 'gpt-image-1',
+                model: imageModel,
                 cost,
+                ...(supportedVisionModel ? { visionModel: supportedVisionModel } : {}),
             },
         };
         return enhancedResponse;
@@ -19389,6 +19528,7 @@ const disco = {
         getNYTimeZone: getNYTimeZone,
         getTradingDate: getTradingDate,
         getTradingStartAndEndDates: getTradingStartAndEndDates,
+        getTradingDaysBack: getTradingDaysBack,
         isMarketDay: isMarketDay,
         isWithinMarketHours: isWithinMarketHours,
         countTradingDays: countTradingDays,