@react-grab/ami 0.0.55

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Aiden Bai
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/client.cjs

@@ -0,0 +1,202 @@
+'use strict';
+
+var amiSdk = require('ami-sdk');
+
+// src/client.ts
+var STORAGE_KEY = "react-grab:agent-sessions";
+var TOKEN_STORAGE_KEY = "react-grab:ami-token";
+var DEFAULT_PROJECT_ID = "react-grab-agent";
+var loadCachedToken = () => {
+  try {
+    return localStorage.getItem(TOKEN_STORAGE_KEY);
+  } catch {
+    return null;
+  }
+};
+var saveCachedToken = (token) => {
+  try {
+    localStorage.setItem(TOKEN_STORAGE_KEY, token);
+  } catch {
+  }
+};
+var authenticate = async () => {
+  const exchangeToken = await amiSdk.createExchangeToken();
+  const authUrl = amiSdk.getExchangeTokenAuthUrl(exchangeToken);
+  window.open(authUrl, "_blank");
+  const result = await amiSdk.getToken(exchangeToken, () => {
+  });
+  if (!result.success || !result.token) {
+    throw new Error("Authentication failed");
+  }
+  saveCachedToken(result.token);
+  return result.token;
+};
+var getOrCreateToken = async () => {
+  const cachedToken = loadCachedToken();
+  if (cachedToken) {
+    return cachedToken;
+  }
+  return authenticate();
+};
+var isToolPart = (part) => {
+  return part.type.startsWith("tool-");
+};
+var isToolCallComplete = (part) => {
+  return part.state === "output-available" && part.output !== void 0;
+};
+var hasIncompleteToolCalls = (message) => {
+  if (message.role !== "assistant") return false;
+  const toolParts = message.parts.filter(isToolPart);
+  if (toolParts.length === 0) return false;
+  return toolParts.some((part) => !isToolCallComplete(part));
+};
+var sanitizeMessages = (messages) => {
+  const sanitized = [];
+  for (const message of messages) {
+    if (hasIncompleteToolCalls(message)) {
+      break;
+    }
+    sanitized.push(message);
+  }
+  return sanitized;
+};
+var runAgent = async (context, token, projectId, onStatus) => {
+  const fullPrompt = `${context.prompt}
+
+${context.content}`;
+  const messages = [
+    amiSdk.convertToAmiMessage({
+      id: amiSdk.generateId(),
+      role: "user",
+      content: fullPrompt
+    })
+  ];
+  const chatId = amiSdk.generateId();
+  const environment = amiSdk.createMinimalEnvironment(window.location.href);
+  const upsertMessage = async (message) => {
+    const existingIndex = messages.findIndex((m) => m.id === message.id);
+    if (existingIndex >= 0) {
+      messages[existingIndex] = message;
+    } else {
+      messages.push(message);
+    }
+    for (const part of message.parts) {
+      if (part.type === "text" && part.text) {
+        const statusUpdate = part.text.length > 100 ? `${part.text.slice(0, 100)}...` : part.text;
+        onStatus(statusUpdate);
+      }
+    }
+  };
+  const status = await amiSdk.runAgentLoop({
+    messages: sanitizeMessages(messages),
+    context: {
+      environment,
+      systemContext: [],
+      attachments: []
+    },
+    url: `${amiSdk.AMI_BRIDGE_URL}/api/v1/agent-proxy`,
+    chatId,
+    projectId,
+    token,
+    upsertMessage,
+    getMessages: () => sanitizeMessages(messages)
+  });
+  return status === "completed" ? "Completed successfully" : `Task finished (${status})`;
+};
+var createAmiAgentProvider = (projectId = DEFAULT_PROJECT_ID) => ({
+  send: async function* (context) {
+    const token = await getOrCreateToken();
+    yield "Please wait...";
+    const statusQueue = [];
+    let resolveWait = null;
+    const onStatus = (status) => {
+      statusQueue.push(status);
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    };
+    const agentPromise = runAgent(context, token, projectId, onStatus);
+    let done = false;
+    agentPromise.then((finalStatus) => {
+      statusQueue.push(finalStatus);
+      done = true;
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    }).catch((error) => {
+      const errorMessage = error instanceof Error ? error.message : "Unknown error";
+      statusQueue.push(`Error: ${errorMessage}`);
+      done = true;
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    });
+    while (!done || statusQueue.length > 0) {
+      if (statusQueue.length > 0) {
+        yield statusQueue.shift();
+      } else if (!done) {
+        await new Promise((resolve) => {
+          resolveWait = resolve;
+        });
+      }
+    }
+  },
+  resume: async function* (sessionId) {
+    const savedSessions = sessionStorage.getItem(STORAGE_KEY);
+    if (!savedSessions) {
+      throw new Error("No sessions to resume");
+    }
+    const sessionsObject = JSON.parse(savedSessions);
+    const session = sessionsObject[sessionId];
+    if (!session) {
+      throw new Error(`Session ${sessionId} not found`);
+    }
+    const context = session.context;
+    const token = await getOrCreateToken();
+    yield "Resuming...";
+    const statusQueue = [];
+    let resolveWait = null;
+    const onStatus = (status) => {
+      statusQueue.push(status);
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    };
+    const agentPromise = runAgent(context, token, DEFAULT_PROJECT_ID, onStatus);
+    let done = false;
+    agentPromise.then((finalStatus) => {
+      statusQueue.push(finalStatus);
+      done = true;
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    }).catch((error) => {
+      const errorMessage = error instanceof Error ? error.message : "Unknown error";
+      statusQueue.push(`Error: ${errorMessage}`);
+      done = true;
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    });
+    while (!done || statusQueue.length > 0) {
+      if (statusQueue.length > 0) {
+        yield statusQueue.shift();
+      } else if (!done) {
+        await new Promise((resolve) => {
+          resolveWait = resolve;
+        });
+      }
+    }
+  },
+  supportsResume: true
+});
+var defaultAmiAgentProvider = createAmiAgentProvider();
+
+exports.createAmiAgentProvider = createAmiAgentProvider;
+exports.defaultAmiAgentProvider = defaultAmiAgentProvider;

package/dist/client.d.cts

@@ -0,0 +1,6 @@
+import { AgentProvider } from 'react-grab/core';
+
+declare const createAmiAgentProvider: (projectId?: string) => AgentProvider;
+declare const defaultAmiAgentProvider: AgentProvider<unknown>;
+
+export { createAmiAgentProvider, defaultAmiAgentProvider };

package/dist/client.d.ts

@@ -0,0 +1,6 @@
+import { AgentProvider } from 'react-grab/core';
+
+declare const createAmiAgentProvider: (projectId?: string) => AgentProvider;
+declare const defaultAmiAgentProvider: AgentProvider<unknown>;
+
+export { createAmiAgentProvider, defaultAmiAgentProvider };

package/dist/client.js

@@ -0,0 +1,199 @@
+import { convertToAmiMessage, generateId, createMinimalEnvironment, runAgentLoop, AMI_BRIDGE_URL, createExchangeToken, getExchangeTokenAuthUrl, getToken } from 'ami-sdk';
+
+// src/client.ts
+var STORAGE_KEY = "react-grab:agent-sessions";
+var TOKEN_STORAGE_KEY = "react-grab:ami-token";
+var DEFAULT_PROJECT_ID = "react-grab-agent";
+var loadCachedToken = () => {
+  try {
+    return localStorage.getItem(TOKEN_STORAGE_KEY);
+  } catch {
+    return null;
+  }
+};
+var saveCachedToken = (token) => {
+  try {
+    localStorage.setItem(TOKEN_STORAGE_KEY, token);
+  } catch {
+  }
+};
+var authenticate = async () => {
+  const exchangeToken = await createExchangeToken();
+  const authUrl = getExchangeTokenAuthUrl(exchangeToken);
+  window.open(authUrl, "_blank");
+  const result = await getToken(exchangeToken, () => {
+  });
+  if (!result.success || !result.token) {
+    throw new Error("Authentication failed");
+  }
+  saveCachedToken(result.token);
+  return result.token;
+};
+var getOrCreateToken = async () => {
+  const cachedToken = loadCachedToken();
+  if (cachedToken) {
+    return cachedToken;
+  }
+  return authenticate();
+};
+var isToolPart = (part) => {
+  return part.type.startsWith("tool-");
+};
+var isToolCallComplete = (part) => {
+  return part.state === "output-available" && part.output !== void 0;
+};
+var hasIncompleteToolCalls = (message) => {
+  if (message.role !== "assistant") return false;
+  const toolParts = message.parts.filter(isToolPart);
+  if (toolParts.length === 0) return false;
+  return toolParts.some((part) => !isToolCallComplete(part));
+};
+var sanitizeMessages = (messages) => {
+  const sanitized = [];
+  for (const message of messages) {
+    if (hasIncompleteToolCalls(message)) {
+      break;
+    }
+    sanitized.push(message);
+  }
+  return sanitized;
+};
+var runAgent = async (context, token, projectId, onStatus) => {
+  const fullPrompt = `${context.prompt}
+
+${context.content}`;
+  const messages = [
+    convertToAmiMessage({
+      id: generateId(),
+      role: "user",
+      content: fullPrompt
+    })
+  ];
+  const chatId = generateId();
+  const environment = createMinimalEnvironment(window.location.href);
+  const upsertMessage = async (message) => {
+    const existingIndex = messages.findIndex((m) => m.id === message.id);
+    if (existingIndex >= 0) {
+      messages[existingIndex] = message;
+    } else {
+      messages.push(message);
+    }
+    for (const part of message.parts) {
+      if (part.type === "text" && part.text) {
+        const statusUpdate = part.text.length > 100 ? `${part.text.slice(0, 100)}...` : part.text;
+        onStatus(statusUpdate);
+      }
+    }
+  };
+  const status = await runAgentLoop({
+    messages: sanitizeMessages(messages),
+    context: {
+      environment,
+      systemContext: [],
+      attachments: []
+    },
+    url: `${AMI_BRIDGE_URL}/api/v1/agent-proxy`,
+    chatId,
+    projectId,
+    token,
+    upsertMessage,
+    getMessages: () => sanitizeMessages(messages)
+  });
+  return status === "completed" ? "Completed successfully" : `Task finished (${status})`;
+};
+var createAmiAgentProvider = (projectId = DEFAULT_PROJECT_ID) => ({
+  send: async function* (context) {
+    const token = await getOrCreateToken();
+    yield "Please wait...";
+    const statusQueue = [];
+    let resolveWait = null;
+    const onStatus = (status) => {
+      statusQueue.push(status);
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    };
+    const agentPromise = runAgent(context, token, projectId, onStatus);
+    let done = false;
+    agentPromise.then((finalStatus) => {
+      statusQueue.push(finalStatus);
+      done = true;
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    }).catch((error) => {
+      const errorMessage = error instanceof Error ? error.message : "Unknown error";
+      statusQueue.push(`Error: ${errorMessage}`);
+      done = true;
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    });
+    while (!done || statusQueue.length > 0) {
+      if (statusQueue.length > 0) {
+        yield statusQueue.shift();
+      } else if (!done) {
+        await new Promise((resolve) => {
+          resolveWait = resolve;
+        });
+      }
+    }
+  },
+  resume: async function* (sessionId) {
+    const savedSessions = sessionStorage.getItem(STORAGE_KEY);
+    if (!savedSessions) {
+      throw new Error("No sessions to resume");
+    }
+    const sessionsObject = JSON.parse(savedSessions);
+    const session = sessionsObject[sessionId];
+    if (!session) {
+      throw new Error(`Session ${sessionId} not found`);
+    }
+    const context = session.context;
+    const token = await getOrCreateToken();
+    yield "Resuming...";
+    const statusQueue = [];
+    let resolveWait = null;
+    const onStatus = (status) => {
+      statusQueue.push(status);
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    };
+    const agentPromise = runAgent(context, token, DEFAULT_PROJECT_ID, onStatus);
+    let done = false;
+    agentPromise.then((finalStatus) => {
+      statusQueue.push(finalStatus);
+      done = true;
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    }).catch((error) => {
+      const errorMessage = error instanceof Error ? error.message : "Unknown error";
+      statusQueue.push(`Error: ${errorMessage}`);
+      done = true;
+      if (resolveWait) {
+        resolveWait();
+        resolveWait = null;
+      }
+    });
+    while (!done || statusQueue.length > 0) {
+      if (statusQueue.length > 0) {
+        yield statusQueue.shift();
+      } else if (!done) {
+        await new Promise((resolve) => {
+          resolveWait = resolve;
+        });
+      }
+    }
+  },
+  supportsResume: true
+});
+var defaultAmiAgentProvider = createAmiAgentProvider();
+
+export { createAmiAgentProvider, defaultAmiAgentProvider };

package/package.json

@@ -0,0 +1,26 @@
+{
+  "name": "@react-grab/ami",
+  "version": "0.0.55",
+  "type": "module",
+  "exports": {
+    "./client": {
+      "types": "./dist/client.d.ts",
+      "import": "./dist/client.js",
+      "require": "./dist/client.cjs"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "devDependencies": {
+    "tsup": "^8.4.0"
+  },
+  "dependencies": {
+    "ami-sdk": "^0.0.1",
+    "react-grab": "0.0.55"
+  },
+  "scripts": {
+    "dev": "tsup --watch",
+    "build": "NODE_ENV=production tsup"
+  }
+}