@discomedia/utils 1.0.39 → 1.0.41

package/dist/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.0.39",
+  "version": "1.0.41",
   "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.1.0",
+    "openai": "^6.2.0",
     "p-limit": "^7.1.1",
     "tslib": "^2.8.1",
     "ws": "^8.18.3"

package/dist/test.js

@@ -799,7 +799,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.1.0'; // x-release-please-version
+const VERSION = '6.2.0'; // x-release-please-version
 
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 const isRunningInBrowser = () => {
@@ -2200,6 +2200,15 @@ function getName(value) {
         .pop() || undefined);
 }
 const isAsyncIterable = (value) => value != null && typeof value === 'object' && typeof value[Symbol.asyncIterator] === 'function';
+/**
+ * Returns a multipart/form-data request if any part of the given request body contains a File / Blob value.
+ * Otherwise returns the request as is.
+ */
+const maybeMultipartFormRequestOptions = async (opts, fetch) => {
+    if (!hasUploadableValue(opts.body))
+        return opts;
+    return { ...opts, body: await createForm(opts.body, fetch) };
+};
 const multipartFormRequestOptions = async (opts, fetch) => {
     return { ...opts, body: await createForm(opts.body, fetch) };
 };
@@ -2245,6 +2254,22 @@ const createForm = async (body, fetch) => {
 // We check for Blob not File because Bun.File doesn't inherit from File,
 // but they both inherit from Blob and have a `name` property at runtime.
 const isNamedBlob = (value) => value instanceof Blob && 'name' in value;
+const isUploadable = (value) => typeof value === 'object' &&
+    value !== null &&
+    (value instanceof Response || isAsyncIterable(value) || isNamedBlob(value));
+const hasUploadableValue = (value) => {
+    if (isUploadable(value))
+        return true;
+    if (Array.isArray(value))
+        return value.some(hasUploadableValue);
+    if (value && typeof value === 'object') {
+        for (const k in value) {
+            if (hasUploadableValue(value[k]))
+                return true;
+        }
+    }
+    return false;
+};
 const addFormValue = async (form, key, value) => {
     if (value === undefined)
         return;
@@ -2299,7 +2324,7 @@ const isResponseLike = (value) => value != null &&
     typeof value.blob === 'function';
 /**
  * Helper for creating a {@link File} to pass to an SDK upload method from a variety of different data formats
- * @param value the raw content of the file.  Can be an {@link Uploadable}, {@link BlobLikePart}, or {@link AsyncIterable} of {@link BlobLikePart}s
+ * @param value the raw content of the file. Can be an {@link Uploadable}, BlobLikePart, or AsyncIterable of BlobLikeParts
  * @param {string=} name the name of the file. If omitted, toFile will try to determine a file name from bits if possible
  * @param {Object=} options additional properties
  * @param {string=} options.type the MIME type of the content
@@ -4150,7 +4175,7 @@ class Assistants extends APIResource {
 }
 
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-class Sessions extends APIResource {
+let Sessions$1 = class Sessions extends APIResource {
     /**
      * Create an ephemeral API token for use in client-side applications with the
      * Realtime API. Can be configured with the same session parameters as the
@@ -4173,7 +4198,7 @@ class Sessions extends APIResource {
             headers: buildHeaders([{ 'OpenAI-Beta': 'assistants=v2' }, options?.headers]),
         });
     }
-}
+};
 
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 class TranscriptionSessions extends APIResource {
@@ -4208,13 +4233,144 @@ class TranscriptionSessions extends APIResource {
 let Realtime$1 = class Realtime extends APIResource {
     constructor() {
         super(...arguments);
-        this.sessions = new Sessions(this._client);
+        this.sessions = new Sessions$1(this._client);
         this.transcriptionSessions = new TranscriptionSessions(this._client);
     }
 };
-Realtime$1.Sessions = Sessions;
+Realtime$1.Sessions = Sessions$1;
 Realtime$1.TranscriptionSessions = TranscriptionSessions;
 
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+class Sessions extends APIResource {
+    /**
+     * Create a ChatKit session
+     *
+     * @example
+     * ```ts
+     * const chatSession =
+     *   await client.beta.chatkit.sessions.create({
+     *     user: 'x',
+     *     workflow: { id: 'id' },
+     *   });
+     * ```
+     */
+    create(body, options) {
+        return this._client.post('/chatkit/sessions', {
+            body,
+            ...options,
+            headers: buildHeaders([{ 'OpenAI-Beta': 'chatkit_beta=v1' }, options?.headers]),
+        });
+    }
+    /**
+     * Cancel a ChatKit session
+     *
+     * @example
+     * ```ts
+     * const chatSession =
+     *   await client.beta.chatkit.sessions.cancel('cksess_123');
+     * ```
+     */
+    cancel(sessionID, options) {
+        return this._client.post(path `/chatkit/sessions/${sessionID}/cancel`, {
+            ...options,
+            headers: buildHeaders([{ 'OpenAI-Beta': 'chatkit_beta=v1' }, options?.headers]),
+        });
+    }
+}
+
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+let Threads$1 = class Threads extends APIResource {
+    /**
+     * Retrieve a ChatKit thread
+     *
+     * @example
+     * ```ts
+     * const chatkitThread =
+     *   await client.beta.chatkit.threads.retrieve('cthr_123');
+     * ```
+     */
+    retrieve(threadID, options) {
+        return this._client.get(path `/chatkit/threads/${threadID}`, {
+            ...options,
+            headers: buildHeaders([{ 'OpenAI-Beta': 'chatkit_beta=v1' }, options?.headers]),
+        });
+    }
+    /**
+     * List ChatKit threads
+     *
+     * @example
+     * ```ts
+     * // Automatically fetches more pages as needed.
+     * for await (const chatkitThread of client.beta.chatkit.threads.list()) {
+     *   // ...
+     * }
+     * ```
+     */
+    list(query = {}, options) {
+        return this._client.getAPIList('/chatkit/threads', (ConversationCursorPage), {
+            query,
+            ...options,
+            headers: buildHeaders([{ 'OpenAI-Beta': 'chatkit_beta=v1' }, options?.headers]),
+        });
+    }
+    /**
+     * Delete a ChatKit thread
+     *
+     * @example
+     * ```ts
+     * const thread = await client.beta.chatkit.threads.delete(
+     *   'cthr_123',
+     * );
+     * ```
+     */
+    delete(threadID, options) {
+        return this._client.delete(path `/chatkit/threads/${threadID}`, {
+            ...options,
+            headers: buildHeaders([{ 'OpenAI-Beta': 'chatkit_beta=v1' }, options?.headers]),
+        });
+    }
+    /**
+     * List ChatKit thread items
+     *
+     * @example
+     * ```ts
+     * // Automatically fetches more pages as needed.
+     * for await (const thread of client.beta.chatkit.threads.listItems(
+     *   'cthr_123',
+     * )) {
+     *   // ...
+     * }
+     * ```
+     */
+    listItems(threadID, query = {}, options) {
+        return this._client.getAPIList(path `/chatkit/threads/${threadID}/items`, (ConversationCursorPage), { query, ...options, headers: buildHeaders([{ 'OpenAI-Beta': 'chatkit_beta=v1' }, options?.headers]) });
+    }
+};
+
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+class ChatKit extends APIResource {
+    constructor() {
+        super(...arguments);
+        this.sessions = new Sessions(this._client);
+        this.threads = new Threads$1(this._client);
+    }
+    /**
+     * Upload a ChatKit file
+     *
+     * @example
+     * ```ts
+     * const response = await client.beta.chatkit.uploadFile({
+     *   file: fs.createReadStream('path/to/file'),
+     * });
+     * ```
+     */
+    uploadFile(body, options) {
+        return this._client.post('/chatkit/files', maybeMultipartFormRequestOptions({ body, ...options, headers: buildHeaders([{ 'OpenAI-Beta': 'chatkit_beta=v1' }, options?.headers]) }, this._client));
+    }
+}
+ChatKit.Sessions = Sessions;
+ChatKit.Threads = Threads$1;
+
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 /**
  * @deprecated The Assistants API is deprecated in favor of the Responses API
@@ -5152,11 +5308,13 @@ class Beta extends APIResource {
     constructor() {
         super(...arguments);
         this.realtime = new Realtime$1(this._client);
+        this.chatkit = new ChatKit(this._client);
         this.assistants = new Assistants(this._client);
         this.threads = new Threads(this._client);
     }
 }
 Beta.Realtime = Realtime$1;
+Beta.ChatKit = ChatKit;
 Beta.Assistants = Assistants;
 Beta.Threads = Threads;
 
@@ -6909,6 +7067,51 @@ class VectorStores extends APIResource {
 VectorStores.Files = Files;
 VectorStores.FileBatches = FileBatches;
 
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+class Videos extends APIResource {
+    /**
+     * Create a video
+     */
+    create(body, options) {
+        return this._client.post('/videos', maybeMultipartFormRequestOptions({ body, ...options }, this._client));
+    }
+    /**
+     * Retrieve a video
+     */
+    retrieve(videoID, options) {
+        return this._client.get(path `/videos/${videoID}`, options);
+    }
+    /**
+     * List videos
+     */
+    list(query = {}, options) {
+        return this._client.getAPIList('/videos', (ConversationCursorPage), { query, ...options });
+    }
+    /**
+     * Delete a video
+     */
+    delete(videoID, options) {
+        return this._client.delete(path `/videos/${videoID}`, options);
+    }
+    /**
+     * Download video content
+     */
+    downloadContent(videoID, query = {}, options) {
+        return this._client.get(path `/videos/${videoID}/content`, {
+            query,
+            ...options,
+            headers: buildHeaders([{ Accept: 'application/binary' }, options?.headers]),
+            __binaryResponse: true,
+        });
+    }
+    /**
+     * Create a video remix
+     */
+    remix(videoID, body, options) {
+        return this._client.post(path `/videos/${videoID}/remix`, maybeMultipartFormRequestOptions({ body, ...options }, this._client));
+    }
+}
+
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 var _Webhooks_instances, _Webhooks_validateSecret, _Webhooks_getRequiredHeader;
 class Webhooks extends APIResource {
@@ -7047,6 +7250,7 @@ class OpenAI {
         this.conversations = new Conversations(this);
         this.evals = new Evals(this);
         this.containers = new Containers(this);
+        this.videos = new Videos(this);
         if (apiKey === undefined) {
             throw new OpenAIError('Missing credentials. Please pass an `apiKey`, or set the `OPENAI_API_KEY` environment variable.');
         }
@@ -7517,6 +7721,7 @@ OpenAI.Realtime = Realtime;
 OpenAI.Conversations = Conversations;
 OpenAI.Evals = Evals;
 OpenAI.Containers = Containers;
+OpenAI.Videos = Videos;
 
 function getDefaultExportFromCjs (x) {
 	return x && x.__esModule && Object.prototype.hasOwnProperty.call(x, 'default') ? x['default'] : x;
@@ -13526,14 +13731,25 @@ class AlpacaMarketDataAPI extends EventEmitter {
             `Avg Volume: ${avgVolume.toLocaleString()}`);
     }
     /**
-     * Get all assets available for trade and data consumption from Alpaca
-     * @param params Optional query params: status (e.g. 'active'), asset_class (e.g. 'us_equity', 'crypto')
+     * Get assets available for trade and data consumption from Alpaca
+     * @param params Optional query params
+     * - status: 'active' | 'inactive' (default 'active')
+     * - asset_class: 'us_equity' | 'us_option' | 'crypto' (default 'us_equity')
+     * - shortable: if true (default), filters to assets with shortable=true and easy_to_borrow=true
      * @returns Array of AlpacaAsset objects
      * @see https://docs.alpaca.markets/reference/get-v2-assets-1
      */
     async getAssets(params) {
         // Endpoint: GET /v2/assets
-        return this.makeRequest('/assets', 'GET', params, 'api'); // use apiURL
+        const { status = 'active', asset_class = 'us_equity', shortable = true, } = {
+            status: params?.status ?? 'active',
+            asset_class: params?.asset_class ?? 'us_equity',
+            shortable: params?.shortable ?? true,
+        };
+        const assets = await this.makeRequest('/assets', 'GET', { status, asset_class }, 'api');
+        if (!shortable)
+            return assets;
+        return assets.filter((a) => a.shortable === true && a.easy_to_borrow === true);
     }
     /**
      * Get a single asset by symbol or asset_id
@@ -14177,88 +14393,84 @@ class AlpacaMarketDataAPI extends EventEmitter {
     }
 }
 // Export the singleton instance
-AlpacaMarketDataAPI.getInstance();
+const marketDataAPI = AlpacaMarketDataAPI.getInstance();
+
+var alpacaMarketDataApi = /*#__PURE__*/Object.freeze({
+    __proto__: null,
+    AlpacaMarketDataAPI: AlpacaMarketDataAPI,
+    marketDataAPI: marketDataAPI
+});
 
 // Test file for context functionality
-//testOpenRouter();
-//testGetMarketStatus();
-//testCryptoMarketData();
-//testGetPortfolioDailyHistory();
-async function testWebSocketConnectAndDisconnect() {
-    console.log('\n--- Testing WebSocket Connect and Disconnect ---');
+async function testGetAssetsShortableFilter() {
+    console.log('\n--- Testing getAssets params and shortable filter ---');
     const log = (message, options = { type: 'info' }) => {
         log$1(message, { ...options, source: 'Test' });
     };
-    // Ensure market data env vars are populated from TRADING_* if needed
-    if (!process.env.ALPACA_API_KEY && process.env.ALPACA_TRADING_API_KEY) {
-        process.env.ALPACA_API_KEY = process.env.ALPACA_TRADING_API_KEY;
-    }
-    if (!process.env.ALPACA_SECRET_KEY && process.env.ALPACA_TRADING_SECRET_KEY) {
-        process.env.ALPACA_SECRET_KEY = process.env.ALPACA_TRADING_SECRET_KEY;
-    }
-    if (!process.env.ALPACA_ACCOUNT_TYPE && process.env.ALPACA_TRADING_ACCOUNT_TYPE) {
-        process.env.ALPACA_ACCOUNT_TYPE = process.env.ALPACA_TRADING_ACCOUNT_TYPE;
-    }
-    const apiKey = process.env.ALPACA_API_KEY;
-    const secretKey = process.env.ALPACA_SECRET_KEY;
-    if (!apiKey || !secretKey) {
-        console.error('Missing Alpaca API credentials. Check .env ALPACA_TRADING_API_KEY/SECRET or ALPACA_API_KEY/SECRET.');
+    if (!process.env.ALPACA_API_KEY || !process.env.ALPACA_SECRET_KEY) {
+        console.log('Skipping getAssets test: Missing environment variables (ALPACA_API_KEY, ALPACA_SECRET_KEY)');
         return;
     }
-    const md = AlpacaMarketDataAPI.getInstance();
-    const symbols = ['SPY', 'TSLA', 'AAPL'];
-    // Track received minute bars per symbol
-    const barCounts = Object.fromEntries(symbols.map((s) => [s, 0]));
-    const firstSeenAt = {};
-    // Connect and subscribe to minute bars
-    log(`Connecting stock stream and subscribing to minute bars for ${symbols.join(', ')}`);
-    md.connectStockStream();
-    md.subscribe('stock', { bars: symbols });
-    const formatNY = (d) => d.toLocaleString('en-US', { timeZone: 'America/New_York' });
-    const onBar = (msg) => {
-        if (!symbols.includes(msg.S))
-            return;
-        barCounts[msg.S] += 1;
-        if (!firstSeenAt[msg.S]) {
-            firstSeenAt[msg.S] = formatNY(new Date(msg.t));
+    try {
+        const { marketDataAPI } = await Promise.resolve().then(function () { return alpacaMarketDataApi; });
+        // 1) Default call: should filter to shortable && easy_to_borrow and default to active us_equity
+        log('Calling getAssets() with defaults');
+        const filtered = await marketDataAPI.getAssets();
+        console.log(`Received ${filtered.length} filtered assets (default params)`);
+        const allFilteredValid = filtered.every((a) => a.shortable === true && a.easy_to_borrow === true);
+        if (!allFilteredValid) {
+            console.error('✗ Default getAssets() included assets that are not shortable and easy_to_borrow');
         }
-        log(`Bar ${msg.S} o=${msg.o.toFixed(2)} h=${msg.h.toFixed(2)} l=${msg.l.toFixed(2)} c=${msg.c.toFixed(2)} @ ${formatNY(new Date(msg.t))}`, { type: 'debug', symbol: msg.S });
-    };
-    md.on('stock-b', onBar);
-    // Wait until we have at least 2 bars per symbol, or timeout after 3 minutes
-    const minBarsPerSymbol = 2;
-    const timeoutMs = 3 * 60 * 1000;
-    const done = await new Promise((resolve) => {
-        const startedAt = Date.now();
-        const interval = setInterval(() => {
-            const hasAll = symbols.every((s) => barCounts[s] >= minBarsPerSymbol);
-            const elapsed = Date.now() - startedAt;
-            if (hasAll) {
-                clearInterval(interval);
-                resolve(true);
-            }
-            else if (elapsed >= timeoutMs) {
-                clearInterval(interval);
-                resolve(false);
-            }
-        }, 1000);
-    });
-    // Unsubscribe and disconnect
-    md.unsubscribe('stock', { bars: symbols });
-    md.disconnectStockStream();
-    md.removeListener('stock-b', onBar);
-    // Report results
-    const nowNY = formatNY(new Date());
-    console.log('\nStock bars summary:');
-    symbols.forEach((s) => {
-        console.log(`  ${s}: ${barCounts[s]} bars (first at ${firstSeenAt[s] ?? 'n/a'}; finished at ${nowNY})`);
-    });
-    if (done) {
-        console.log('✓ Received minimum bars for all symbols and disconnected.');
-    }
-    else {
-        console.warn('Completed with timeout before receiving minimum bars for all symbols.');
-    }
-}
-testWebSocketConnectAndDisconnect();
+        else {
+            console.log('✓ Default getAssets() correctly filtered shortable and easy_to_borrow assets');
+        }
+        // 2) Unfiltered call: shortable=false
+        log('Calling getAssets({ shortable: false })');
+        const unfiltered = await marketDataAPI.getAssets({ shortable: false });
+        console.log(`Received ${unfiltered.length} unfiltered assets`);
+        if (unfiltered.length < filtered.length) {
+            console.warn('Unfiltered list is smaller than filtered. This is unexpected but not fatal.');
+        }
+        else {
+            console.log('✓ Unfiltered list length is >= filtered list length');
+        }
+        // 3) Specific constraints: inactive crypto, not asserting non-empty
+        log("Calling getAssets({ status: 'inactive', asset_class: 'crypto', shortable: false })");
+        const inactiveCrypto = await marketDataAPI.getAssets({ status: 'inactive', asset_class: 'crypto', shortable: false });
+        console.log(`Received ${inactiveCrypto.length} inactive crypto assets`);
+        if (inactiveCrypto.length > 0) {
+            const valid = inactiveCrypto.every((a) => a.class === 'crypto' && a.status === 'inactive');
+            if (!valid) {
+                console.error('✗ Inactive crypto request returned assets with mismatched class or status');
+            }
+            else {
+                console.log('✓ Inactive crypto request returned only crypto with inactive status');
+            }
+        }
+        else {
+            console.log('No inactive crypto assets returned; skipping class/status assertions.');
+        }
+    }
+    catch (error) {
+        console.error('✗ getAssets test failed:', error);
+        if (error instanceof Error) {
+            console.error('Error message:', error.message);
+            console.error('Stack trace:', error.stack);
+        }
+    }
+}
+// testGetTradingDate();
+// testGetTradingStartAndEndDates();
+// testGetLastFullTradingDate();
+// testGetMarketOpenClose();
+// testGetNYTimeZone();
+// testGetNextMarketDay();
+// testCountTradingDays();
+// testGetPreviousMarketDay();
+// testOpenRouter();
+// testGetMarketStatus();
+// testCryptoMarketData();
+// testGetPortfolioDailyHistory();
+// testWebSocketConnectAndDisconnect();
+testGetAssetsShortableFilter();
 //# sourceMappingURL=test.js.map