@pipe2-ai/sdk 0.1.1 → 0.2.0

package/dist/src/index.d.ts

@@ -4,5 +4,5 @@ export type Pipe2SDK = ReturnType<typeof getSdk>;
  * Create a Pipe2.ai SDK client with full support for queries, mutations,
  * and real-time WebSocket subscriptions.
  */
-export declare function createClient(token: string, endpoint?: string): Pipe2SDK;
+export declare function createClient(token: string, endpoint?: string, wsEndpoint?: string): Pipe2SDK;
 export * from '../generated/sdk.js';

package/dist/src/index.js

@@ -1,28 +1,31 @@
 import { GraphQLClient } from 'graphql-request';
 import { createClient as createWsClient } from 'graphql-ws';
 import { print } from 'graphql';
-import WebSocket from 'ws';
 import { getSdk } from '../generated/sdk.js';
 /**
  * Create a Pipe2.ai SDK client with full support for queries, mutations,
  * and real-time WebSocket subscriptions.
  */
-export function createClient(token, endpoint = 'https://api.pipe2.ai/v1/graphql') {
+export function createClient(token, endpoint = 'https://api.pipe2.ai/v1/graphql', wsEndpoint) {
     const headers = {};
     if (token) {
         headers.Authorization = `Bearer ${token}`;
     }
     // HTTP client for queries and mutations
     const httpClient = new GraphQLClient(endpoint, { headers });
-    // WebSocket client for subscriptions (lazy — connects on first subscription)
-    const wsEndpoint = endpoint.replace(/^http/, 'ws');
+    // WebSocket client for subscriptions (lazy — connects on first subscription).
+    // wsEndpoint overrides the derived URL — needed in dev because Astro's
+    // dev-server doesn't proxy WS upgrades through Vite, so subscriptions must
+    // bypass the same-origin proxy and dial Hasura directly.
+    const wsEndpointResolved = wsEndpoint || endpoint.replace(/^http/, 'ws');
     let wsClient = null;
     const getWsClient = () => {
         if (!wsClient) {
             wsClient = createWsClient({
-                url: wsEndpoint,
+                url: wsEndpointResolved,
                 connectionParams: { headers },
-                webSocketImpl: WebSocket,
+                // Browsers have native WebSocket; only Node.js needs the 'ws' polyfill.
+                // graphql-ws uses globalThis.WebSocket by default when webSocketImpl is omitted.
             });
         }
         return wsClient;

package/package.json

@@ -1,8 +1,16 @@
 {
   "name": "@pipe2-ai/sdk",
-  "version": "0.1.1",
-  "main": "dist/src/index.js",
-  "types": "dist/src/index.d.ts",
+  "version": "0.2.0",
+  "type": "module",
+  "main": "./dist/src/index.js",
+  "types": "./dist/src/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/src/index.d.ts",
+      "import": "./dist/src/index.js",
+      "default": "./dist/src/index.js"
+    }
+  },
   "files": [
     "dist"
   ],