@pgnio/helios-mcp 0.2.0 → 0.2.1

package/dist/cli.js

@@ -920,11 +920,15 @@ var HeliosAgentMcpAuthService = class {
       requestedConstraints
     });
     if (pendingLogin) {
-      return {
-        ...pendingLogin,
-        connected: false,
-        message: formatHeliosLoginPrompt(pendingLogin)
-      };
+      const polled = await this.pollLogin(pendingLogin.deviceCode).catch(() => null);
+      if (polled?.status === "connected") return polled;
+      if (polled?.status !== "rejected" && polled?.status !== "expired") {
+        return {
+          ...pendingLogin,
+          connected: false,
+          message: formatHeliosLoginPrompt(pendingLogin)
+        };
+      }
     }
     const started = await this.startLogin({
       clientName: options.clientName ?? "Helios MCP",
@@ -937,7 +941,24 @@ var HeliosAgentMcpAuthService = class {
       message: formatHeliosLoginPrompt(started)
     };
   }
+  /** Polls the given login, or the stored pending one when no device code is given. */
   async pollLogin(deviceCode) {
+    if (!deviceCode) {
+      const pending = await this.sessionStore.getPendingLogin();
+      if (!pending) {
+        const session = await this.getValidSession();
+        if (session) return { ...sanitizeAgentSession(session), connected: true, status: "connected" };
+        return {
+          status: "failed",
+          connected: false,
+          message: "No Helios login is in progress. Call helios_auth_ensure_login to start one."
+        };
+      }
+      deviceCode = pending.deviceCode;
+    } else {
+      const pending = await this.sessionStore.getPendingLogin();
+      if (pending && pending.userCode.toUpperCase() === deviceCode.trim().toUpperCase()) deviceCode = pending.deviceCode;
+    }
     try {
       const response = await this.client.exchangeDeviceCode(deviceCode);
       const session = createStoredAgentSession(response, this.now());
@@ -1072,7 +1093,7 @@ User code: ${result.userCode}
 
 Requested scopes: ${result.requestedScope.join(", ")}
 
-After approving access, ask me to continue and I will finish connecting.`;
+After approving access, call helios_auth_ensure_login (or helios_auth_poll_login) again to finish connecting.`;
 }
 function isPendingLoginCompatible(pendingLogin, requestedScope, requestedConstraints) {
   const pendingScope = new Set(pendingLogin.requestedScope);
@@ -2016,7 +2037,8 @@ function asPending2(login) {
 // src/tools.ts
 var import_zod5 = require("zod");
 var pollLoginInputSchema = import_zod5.z.object({
-  deviceCode: import_zod5.z.string().min(1)
+  /** Optional: defaults to the login started by helios_auth_ensure_login / helios_auth_start_login. */
+  deviceCode: import_zod5.z.string().min(1).optional()
 });
 var startLoginOutputSchema = import_zod5.z.object({
   status: import_zod5.z.literal("pending"),
@@ -2325,7 +2347,7 @@ function createHeliosAuthToolDefinitions(service) {
     {
       name: "helios_auth_ensure_login",
       title: "Ensure Helios Login",
-      description: "Ensures a delegated Helios session exists, or starts a login flow and returns a ready-to-display approval prompt.",
+      description: "Ensures a delegated Helios session exists, or starts a login flow and returns a ready-to-display approval prompt. Call it again after the person approves: it finishes connecting.",
       inputSchema: ensureLoginInputSchema,
       async execute(input) {
         const result = ensureLoginOutputSchema.parse(await service.ensureLogin(input));
@@ -2347,7 +2369,7 @@ function createHeliosAuthToolDefinitions(service) {
     {
       name: "helios_auth_poll_login",
       title: "Poll Helios Login",
-      description: "Polls the device flow until the delegated Helios login is approved or rejected.",
+      description: "Checks whether the Helios login was approved and finishes connecting. The device code is optional; without it, the login in progress is checked.",
       inputSchema: pollLoginInputSchema,
       async execute(input) {
         const result = pollLoginOutputSchema.parse(await service.pollLogin(input.deviceCode));
@@ -2786,7 +2808,7 @@ function withManualLoginFields(result) {
   return {
     actionRequired: true,
     loginUrl: result.verificationUriComplete,
-    nextStep: "Open the login URL in your browser, approve Helios access, then ask me to continue."
+    nextStep: "Open the login URL in your browser, approve Helios access, then call helios_auth_ensure_login again to finish connecting."
   };
 }
 function buildPortfolioText(result) {
@@ -2799,7 +2821,7 @@ ${result.verificationUriComplete}
 
 User code: ${result.userCode}
 
-After approving access, ask me to continue and I will finish connecting.`;
+After approving access, call helios_auth_ensure_login (or helios_auth_poll_login) again to finish connecting.`;
 }
 
 // src/server.ts
@@ -2813,7 +2835,7 @@ async function createHeliosAgentMcpServer(options) {
   const { McpServer } = await import("@modelcontextprotocol/sdk/server/mcp.js");
   const server = new McpServer({
     name: "helios",
-    version: "0.1.0"
+    version: "0.2.1"
   });
   const sessionStore = options.sessionStore ?? new FileAgentSessionStore(options.sessionPath ?? getDefaultHeliosAgentMcpSessionPath());
   const client = new HeliosAgentAuthClient({

package/dist/cli.mjs

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 import {
   createHeliosAgentMcpServer
-} from "./chunk-NWGWXYTS.mjs";
+} from "./chunk-NCF7ASJB.mjs";
 
 // src/cli.ts
 async function main() {

package/dist/index.d.mts

@@ -795,7 +795,8 @@ declare class HeliosAgentMcpAuthService {
     invalidateIfUnauthorized(error: unknown, options?: EnsureLoginOptions): Promise<PendingLoginResult | null>;
     refreshSessionAfterUnauthorized(error: unknown): Promise<StoredAgentSession | null>;
     ensureLogin(options?: EnsureLoginOptions): Promise<EnsureLoginResult>;
-    pollLogin(deviceCode: string): Promise<PollLoginResult>;
+    /** Polls the given login, or the stored pending one when no device code is given. */
+    pollLogin(deviceCode?: string): Promise<PollLoginResult>;
     whoami(): Promise<AgentSessionView>;
     getAccessToken(): Promise<string | null>;
     requireSession(): Promise<StoredAgentSession>;
@@ -1176,10 +1177,10 @@ declare function createHeliosAuthToolDefinitions(service: HeliosAgentMcpAuthServ
     title: string;
     description: string;
     inputSchema: z.ZodObject<{
-        deviceCode: z.ZodString;
+        deviceCode: z.ZodOptional<z.ZodString>;
     }, z.core.$strip>;
     execute(input: {
-        deviceCode: string;
+        deviceCode?: string | undefined;
     }): Promise<{
         content: {
             type: "text";

package/dist/index.d.ts

@@ -795,7 +795,8 @@ declare class HeliosAgentMcpAuthService {
     invalidateIfUnauthorized(error: unknown, options?: EnsureLoginOptions): Promise<PendingLoginResult | null>;
     refreshSessionAfterUnauthorized(error: unknown): Promise<StoredAgentSession | null>;
     ensureLogin(options?: EnsureLoginOptions): Promise<EnsureLoginResult>;
-    pollLogin(deviceCode: string): Promise<PollLoginResult>;
+    /** Polls the given login, or the stored pending one when no device code is given. */
+    pollLogin(deviceCode?: string): Promise<PollLoginResult>;
     whoami(): Promise<AgentSessionView>;
     getAccessToken(): Promise<string | null>;
     requireSession(): Promise<StoredAgentSession>;
@@ -1176,10 +1177,10 @@ declare function createHeliosAuthToolDefinitions(service: HeliosAgentMcpAuthServ
     title: string;
     description: string;
     inputSchema: z.ZodObject<{
-        deviceCode: z.ZodString;
+        deviceCode: z.ZodOptional<z.ZodString>;
     }, z.core.$strip>;
     execute(input: {
-        deviceCode: string;
+        deviceCode?: string | undefined;
     }): Promise<{
         content: {
             type: "text";

package/dist/index.js

@@ -954,11 +954,15 @@ var HeliosAgentMcpAuthService = class {
       requestedConstraints
     });
     if (pendingLogin) {
-      return {
-        ...pendingLogin,
-        connected: false,
-        message: formatHeliosLoginPrompt(pendingLogin)
-      };
+      const polled = await this.pollLogin(pendingLogin.deviceCode).catch(() => null);
+      if (polled?.status === "connected") return polled;
+      if (polled?.status !== "rejected" && polled?.status !== "expired") {
+        return {
+          ...pendingLogin,
+          connected: false,
+          message: formatHeliosLoginPrompt(pendingLogin)
+        };
+      }
     }
     const started = await this.startLogin({
       clientName: options.clientName ?? "Helios MCP",
@@ -971,7 +975,24 @@ var HeliosAgentMcpAuthService = class {
       message: formatHeliosLoginPrompt(started)
     };
   }
+  /** Polls the given login, or the stored pending one when no device code is given. */
   async pollLogin(deviceCode) {
+    if (!deviceCode) {
+      const pending = await this.sessionStore.getPendingLogin();
+      if (!pending) {
+        const session = await this.getValidSession();
+        if (session) return { ...sanitizeAgentSession(session), connected: true, status: "connected" };
+        return {
+          status: "failed",
+          connected: false,
+          message: "No Helios login is in progress. Call helios_auth_ensure_login to start one."
+        };
+      }
+      deviceCode = pending.deviceCode;
+    } else {
+      const pending = await this.sessionStore.getPendingLogin();
+      if (pending && pending.userCode.toUpperCase() === deviceCode.trim().toUpperCase()) deviceCode = pending.deviceCode;
+    }
     try {
       const response = await this.client.exchangeDeviceCode(deviceCode);
       const session = createStoredAgentSession(response, this.now());
@@ -1106,7 +1127,7 @@ User code: ${result.userCode}
 
 Requested scopes: ${result.requestedScope.join(", ")}
 
-After approving access, ask me to continue and I will finish connecting.`;
+After approving access, call helios_auth_ensure_login (or helios_auth_poll_login) again to finish connecting.`;
 }
 function isPendingLoginCompatible(pendingLogin, requestedScope, requestedConstraints) {
   const pendingScope = new Set(pendingLogin.requestedScope);
@@ -2084,7 +2105,8 @@ function asPending2(login) {
 // src/tools.ts
 var import_zod5 = require("zod");
 var pollLoginInputSchema = import_zod5.z.object({
-  deviceCode: import_zod5.z.string().min(1)
+  /** Optional: defaults to the login started by helios_auth_ensure_login / helios_auth_start_login. */
+  deviceCode: import_zod5.z.string().min(1).optional()
 });
 var startLoginOutputSchema = import_zod5.z.object({
   status: import_zod5.z.literal("pending"),
@@ -2393,7 +2415,7 @@ function createHeliosAuthToolDefinitions(service) {
     {
       name: "helios_auth_ensure_login",
       title: "Ensure Helios Login",
-      description: "Ensures a delegated Helios session exists, or starts a login flow and returns a ready-to-display approval prompt.",
+      description: "Ensures a delegated Helios session exists, or starts a login flow and returns a ready-to-display approval prompt. Call it again after the person approves: it finishes connecting.",
       inputSchema: ensureLoginInputSchema,
       async execute(input) {
         const result = ensureLoginOutputSchema.parse(await service.ensureLogin(input));
@@ -2415,7 +2437,7 @@ function createHeliosAuthToolDefinitions(service) {
     {
       name: "helios_auth_poll_login",
       title: "Poll Helios Login",
-      description: "Polls the device flow until the delegated Helios login is approved or rejected.",
+      description: "Checks whether the Helios login was approved and finishes connecting. The device code is optional; without it, the login in progress is checked.",
       inputSchema: pollLoginInputSchema,
       async execute(input) {
         const result = pollLoginOutputSchema.parse(await service.pollLogin(input.deviceCode));
@@ -2854,7 +2876,7 @@ function withManualLoginFields(result) {
   return {
     actionRequired: true,
     loginUrl: result.verificationUriComplete,
-    nextStep: "Open the login URL in your browser, approve Helios access, then ask me to continue."
+    nextStep: "Open the login URL in your browser, approve Helios access, then call helios_auth_ensure_login again to finish connecting."
   };
 }
 function buildPortfolioText(result) {
@@ -2867,7 +2889,7 @@ ${result.verificationUriComplete}
 
 User code: ${result.userCode}
 
-After approving access, ask me to continue and I will finish connecting.`;
+After approving access, call helios_auth_ensure_login (or helios_auth_poll_login) again to finish connecting.`;
 }
 
 // src/server.ts
@@ -2881,7 +2903,7 @@ async function createHeliosAgentMcpServer(options) {
   const { McpServer } = await import("@modelcontextprotocol/sdk/server/mcp.js");
   const server = new McpServer({
     name: "helios",
-    version: "0.1.0"
+    version: "0.2.1"
   });
   const sessionStore = options.sessionStore ?? new FileAgentSessionStore(options.sessionPath ?? getDefaultHeliosAgentMcpSessionPath());
   const client = new HeliosAgentAuthClient({

package/dist/index.mjs

@@ -23,7 +23,7 @@ import {
   pendingAgentLoginSchema,
   retryAfterUnauthorizedRefresh,
   toolAnnotations
-} from "./chunk-NWGWXYTS.mjs";
+} from "./chunk-NCF7ASJB.mjs";
 export {
   FileAgentSessionStore,
   HOSTED_RECONNECT_MESSAGE,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pgnio/helios-mcp",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Helios MCP server: connect an AI agent to your Helios account to read markets and your portfolio and to trade on Hyperliquid. Most apps can use the hosted URL https://helios.pgn.io/mcp instead.",
   "license": "UNLICENSED",
   "homepage": "https://helios.pgn.io/?connect=ai",