@objectstack/service-ai 6.7.1 → 6.8.0

package/dist/index.js

@@ -754,7 +754,20 @@ function createDeleteFieldHandler(ctx) {
 function createListObjectsHandler(ctx) {
   return async (args) => {
     const { filter, includeFields } = args ?? {};
-    const objects = await ctx.metadataService.listObjects();
+    let objects = [];
+    if (ctx.protocol?.getMetaItems) {
+      try {
+        const fromProtocol = await ctx.protocol.getMetaItems({ type: "object" });
+        const arr = Array.isArray(fromProtocol) ? fromProtocol : fromProtocol && typeof fromProtocol === "object" && Array.isArray(fromProtocol.items) ? fromProtocol.items : null;
+        objects = arr ?? await ctx.metadataService.listObjects();
+      } catch {
+        objects = await ctx.metadataService.listObjects();
+      }
+    } else {
+      objects = await ctx.metadataService.listObjects();
+    }
+    if (!Array.isArray(objects)) objects = [];
+    if (!Array.isArray(objects)) objects = [];
     let result = objects.map((o) => {
       const base = {
         name: o.name,
@@ -791,7 +804,15 @@ function createDescribeObjectHandler(ctx) {
     if (!isSnakeCase(objectName)) {
       return JSON.stringify({ error: `Invalid object name "${objectName}". Must be snake_case.` });
     }
-    const objectDef = await ctx.metadataService.getObject(objectName);
+    let objectDef = await ctx.metadataService.getObject(objectName);
+    if (!objectDef && ctx.protocol?.getMetaItems) {
+      try {
+        const all = await ctx.protocol.getMetaItems({ type: "object" });
+        const arr = Array.isArray(all) ? all : all && typeof all === "object" && Array.isArray(all.items) ? all.items : [];
+        objectDef = arr.find((o) => o?.name === objectName);
+      } catch {
+      }
+    }
     if (!objectDef) {
       return JSON.stringify({ error: `Object "${objectName}" not found` });
     }
@@ -1329,6 +1350,16 @@ var InMemoryConversationService = class {
     conversation.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
     return conversation;
   }
+  async update(conversationId, patch) {
+    const conversation = this.store.get(conversationId);
+    if (!conversation) {
+      throw new Error(`Conversation "${conversationId}" not found`);
+    }
+    if (patch.title !== void 0) conversation.title = patch.title;
+    if (patch.metadata !== void 0) conversation.metadata = patch.metadata;
+    conversation.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
+    return conversation;
+  }
   async delete(conversationId) {
     this.store.delete(conversationId);
   }
@@ -1456,6 +1487,30 @@ var _AIService = class _AIService {
       this.logger.info(`[AI] LLM adapter swapped: ${prev} \u2192 ${next.name}`);
     }
   }
+  /**
+   * Best-effort auto-creation of a conversation when the caller did not
+   * supply one but did supply an actor we can attribute the chat to.
+   * Returns the new id on success, or `undefined` if creation failed (in
+   * which case we silently fall back to non-persisted chat).
+   */
+  async autoCreateConversation(ctx) {
+    const actorId = ctx?.actor?.id;
+    if (!actorId) return void 0;
+    try {
+      const conv = await this.conversationService.create({
+        userId: actorId,
+        metadata: ctx?.environmentId ? { environmentId: ctx.environmentId } : void 0
+      });
+      this.logger.debug("[AI] auto-created conversation", { conversationId: conv.id, actorId });
+      return conv.id;
+    } catch (err) {
+      this.logger.warn("[AI] auto-create conversation failed", {
+        actorId,
+        error: err instanceof Error ? err.message : String(err)
+      });
+      return void 0;
+    }
+  }
   /**
    * Best-effort persistence of a single chat message to the conversation
    * store. Failures are logged at warn level and swallowed — chat requests
@@ -1594,7 +1649,12 @@ var _AIService = class _AIService {
     } = options ?? {};
     const maxIterations = maxIter ?? _AIService.DEFAULT_MAX_ITERATIONS;
     const registeredTools = this.toolRegistry.getAll();
-    const conversationId = toolExecutionContext?.conversationId;
+    let conversationId = toolExecutionContext?.conversationId;
+    let autoCreatedConversationId;
+    if (!conversationId) {
+      autoCreatedConversationId = await this.autoCreateConversation(toolExecutionContext);
+      conversationId = autoCreatedConversationId;
+    }
     const mergedTools = [
       ...registeredTools,
       ...restOptions.tools ?? []
@@ -1628,7 +1688,7 @@ var _AIService = class _AIService {
             content: result.content
           });
         }
-        return result;
+        return autoCreatedConversationId ? { ...result, conversationId: autoCreatedConversationId } : result;
       }
       this.logger.debug("[AI] chatWithTools tool calls", {
         iteration,
@@ -1695,7 +1755,7 @@ var _AIService = class _AIService {
         content: finalResult.content
       });
     }
-    return finalResult;
+    return autoCreatedConversationId ? { ...finalResult, conversationId: autoCreatedConversationId } : finalResult;
   }
   /**
    * Stream chat with automatic tool call resolution.
@@ -1713,7 +1773,12 @@ var _AIService = class _AIService {
     } = options ?? {};
     const maxIterations = maxIter ?? _AIService.DEFAULT_MAX_ITERATIONS;
     const registeredTools = this.toolRegistry.getAll();
-    const conversationId = toolExecutionContext?.conversationId;
+    let conversationId = toolExecutionContext?.conversationId;
+    let autoCreatedConversationId;
+    if (!conversationId) {
+      autoCreatedConversationId = await this.autoCreateConversation(toolExecutionContext);
+      conversationId = autoCreatedConversationId;
+    }
     const mergedTools = [
       ...registeredTools,
       ...restOptions.tools ?? []
@@ -1725,6 +1790,14 @@ var _AIService = class _AIService {
     };
     const conversation = [...messages];
     let abortedByCallback = false;
+    if (autoCreatedConversationId) {
+      yield {
+        type: "tool-call",
+        toolCallId: "__conversation_meta__",
+        toolName: "__conversation_meta__",
+        input: { conversationId: autoCreatedConversationId }
+      };
+    }
     if (conversationId && messages.length > 0) {
       const last = messages[messages.length - 1];
       if (last && last.role === "user") {
@@ -2376,6 +2449,56 @@ function buildAIRoutes(aiService, conversationService, logger) {
         }
       }
     },
+    {
+      method: "PATCH",
+      path: "/api/v1/ai/conversations/:id",
+      description: "Update mutable conversation fields (title, metadata)",
+      auth: true,
+      permissions: ["ai:conversations"],
+      handler: async (req) => {
+        const id = req.params?.id;
+        if (!id) {
+          return { status: 400, body: { error: "conversation id is required" } };
+        }
+        const body = req.body ?? {};
+        const patch = {};
+        if (body.title !== void 0) {
+          if (typeof body.title !== "string") {
+            return { status: 400, body: { error: "title must be a string" } };
+          }
+          patch.title = body.title;
+        }
+        if (body.metadata !== void 0) {
+          if (typeof body.metadata !== "object" || body.metadata === null || Array.isArray(body.metadata)) {
+            return { status: 400, body: { error: "metadata must be an object" } };
+          }
+          patch.metadata = body.metadata;
+        }
+        if (patch.title === void 0 && patch.metadata === void 0) {
+          return { status: 400, body: { error: "at least one of title or metadata is required" } };
+        }
+        try {
+          if (req.user?.userId) {
+            const existing = await conversationService.get(id);
+            if (!existing) {
+              return { status: 404, body: { error: `Conversation "${id}" not found` } };
+            }
+            if (existing.userId && existing.userId !== req.user.userId) {
+              return { status: 403, body: { error: "You do not have access to this conversation" } };
+            }
+          }
+          const conversation = await conversationService.update(id, patch);
+          return { status: 200, body: conversation };
+        } catch (err) {
+          const msg = err instanceof Error ? err.message : String(err);
+          if (msg.includes("not found")) {
+            return { status: 404, body: { error: msg } };
+          }
+          logger.error("[AI Route] PATCH /conversations/:id error", err instanceof Error ? err : void 0);
+          return { status: 500, body: { error: "Internal AI service error" } };
+        }
+      }
+    },
     {
       method: "DELETE",
       path: "/api/v1/ai/conversations/:id",
@@ -3139,6 +3262,22 @@ var ObjectQLConversationService = class {
     });
     return await this.get(conversationId);
   }
+  async update(conversationId, patch) {
+    const row = await this.engine.findOne(CONVERSATIONS_OBJECT, {
+      where: { id: conversationId }
+    });
+    if (!row) {
+      throw new Error(`Conversation "${conversationId}" not found`);
+    }
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    const updates = { id: conversationId, updated_at: now };
+    if (patch.title !== void 0) updates.title = patch.title;
+    if (patch.metadata !== void 0) updates.metadata = JSON.stringify(patch.metadata);
+    await this.engine.update(CONVERSATIONS_OBJECT, updates, {
+      where: { id: conversationId }
+    });
+    return await this.get(conversationId);
+  }
   async delete(conversationId) {
     await this.engine.delete(MESSAGES_OBJECT, {
       where: { conversation_id: conversationId },
@@ -3922,8 +4061,9 @@ import { z } from "zod";
 
 // src/schema-retriever.ts
 var SchemaRetriever = class {
-  constructor(metadata, options = {}) {
+  constructor(metadata, options = {}, protocol) {
     this.metadata = metadata;
+    this.protocol = protocol;
     this.options = {
       limit: options.limit ?? 3,
       minScore: options.minScore ?? 1,
@@ -3940,7 +4080,19 @@ var SchemaRetriever = class {
   async retrieve(query) {
     const terms = tokenise(query);
     if (terms.length === 0) return [];
-    const objects = await this.metadata.listObjects();
+    let objects = [];
+    if (this.protocol?.getMetaItems) {
+      try {
+        const fromProtocol = await this.protocol.getMetaItems({ type: "object" });
+        const arr = Array.isArray(fromProtocol) ? fromProtocol : fromProtocol && typeof fromProtocol === "object" && Array.isArray(fromProtocol.items) ? fromProtocol.items : null;
+        objects = arr ?? await this.metadata.listObjects();
+      } catch {
+        objects = await this.metadata.listObjects();
+      }
+    } else {
+      objects = await this.metadata.listObjects();
+    }
+    if (!Array.isArray(objects)) objects = [];
     const hits = [];
     for (const raw of objects) {
       const obj = raw;
@@ -4110,7 +4262,7 @@ var QUERY_DATA_TOOL = {
   }
 };
 function createQueryDataHandler(ctx) {
-  const retriever = new SchemaRetriever(ctx.metadata);
+  const retriever = new SchemaRetriever(ctx.metadata, {}, ctx.protocol);
   const maxLimit = ctx.maxLimit ?? 100;
   return async (args, execCtx) => {
     const { request } = args;
@@ -5377,16 +5529,18 @@ var AIServicePlugin = class {
       return { adapter: new MemoryLLMAdapter(), description: "MemoryLLMAdapter (echo mode)" };
     }
     if (provider === "gateway") {
-      const gatewayModel = String(values.gateway_model ?? "").trim();
+      const gatewayModel = String(values.gateway_model ?? "").trim() || String(process.env.AI_GATEWAY_MODEL ?? "").trim();
       if (!gatewayModel) return null;
+      const gatewayApiKey = String(values.gateway_api_key ?? "").trim() || String(process.env.AI_GATEWAY_API_KEY ?? "").trim();
       try {
         const gatewayPkg = "@ai-sdk/gateway";
-        const { gateway } = await import(
+        const mod = await import(
           /* webpackIgnore: true */
           gatewayPkg
         );
+        const gw = gatewayApiKey ? mod.createGateway({ apiKey: gatewayApiKey }) : mod.gateway;
         return {
-          adapter: new VercelLLMAdapter({ model: gateway(gatewayModel) }),
+          adapter: new VercelLLMAdapter({ model: gw(gatewayModel) }),
           description: `Vercel AI Gateway (model: ${gatewayModel})`
         };
       } catch (err) {
@@ -5404,7 +5558,12 @@ var AIServicePlugin = class {
     };
     const spec = providerSpecs[provider];
     if (!spec) return null;
-    const apiKey = String(values[`${provider}_api_key`] ?? "").trim();
+    const apiKey = String(values[`${provider}_api_key`] ?? "").trim() || // Fall back to the corresponding env var so operators who only
+    // configured env credentials (and didn't paste the key into the
+    // settings form) can still validate the connection.
+    String(
+      process.env[provider === "openai" ? "OPENAI_API_KEY" : provider === "anthropic" ? "ANTHROPIC_API_KEY" : "GOOGLE_GENERATIVE_AI_API_KEY"] ?? ""
+    ).trim();
     if (!apiKey) return null;
     const envKey = provider === "openai" ? "OPENAI_API_KEY" : provider === "anthropic" ? "ANTHROPIC_API_KEY" : "GOOGLE_GENERATIVE_AI_API_KEY";
     process.env[envKey] = apiKey;
@@ -5680,6 +5839,13 @@ var AIServicePlugin = class {
         metadataService = void 0;
       }
     }
+    let protocolService;
+    try {
+      const p = ctx.getService("protocol");
+      if (p && typeof p.getMetaItems === "function") protocolService = p;
+    } catch {
+      protocolService = void 0;
+    }
     try {
       const dataEngine = ctx.getService("data");
       if (dataEngine) {
@@ -5689,7 +5855,8 @@ var AIServicePlugin = class {
           registerQueryDataTool(this.service.toolRegistry, {
             ai: this.service,
             metadata: metadataService,
-            dataEngine
+            dataEngine,
+            protocol: protocolService
           });
           ctx.logger.info("[AI] query_data tool registered");
           try {
@@ -5801,7 +5968,7 @@ var AIServicePlugin = class {
     }
     if (metadataService) {
       try {
-        registerMetadataTools(this.service.toolRegistry, { metadataService });
+        registerMetadataTools(this.service.toolRegistry, { metadataService, protocol: protocolService });
         ctx.logger.info("[AI] Built-in metadata tools registered");
         const { METADATA_TOOL_DEFINITIONS: METADATA_TOOL_DEFINITIONS2 } = await Promise.resolve().then(() => (init_metadata_tools(), metadata_tools_exports));
         for (const toolDef of METADATA_TOOL_DEFINITIONS2) {
@@ -5996,7 +6163,7 @@ var AIServicePlugin = class {
     }
     if (typeof settings.registerAction === "function") {
       settings.registerAction("ai", "test_embedder", async ({ values, payload }) => {
-        const overrides = payload && typeof payload === "object" && payload !== null && "values" in payload ? payload.values ?? {} : {};
+        const overrides = extractOverrides(payload);
         const merged = { ...values ?? {}, ...overrides };
         const provider = String(merged.embedder_provider ?? "none");
         if (provider === "none") {
@@ -6041,10 +6208,33 @@ var AIServicePlugin = class {
     }
     if (typeof settings.registerAction === "function") {
       settings.registerAction("ai", "test", async ({ values, payload }) => {
-        const overrides = payload && typeof payload === "object" && payload !== null && "values" in payload ? payload.values ?? {} : {};
+        const overrides = extractOverrides(payload);
         const merged = { ...values ?? {}, ...overrides };
         const provider = String(merged.provider ?? "memory");
         if (provider === "memory") {
+          const liveName = this.service?.adapterName ?? "";
+          if (this.service && liveName && liveName !== "memory") {
+            const started2 = Date.now();
+            try {
+              const result = await this.service.chat(
+                [{ role: "user", content: "ping" }],
+                { maxTokens: 8 }
+              );
+              const latency = Date.now() - started2;
+              const preview = String(result?.text ?? "").slice(0, 60);
+              return {
+                ok: true,
+                severity: "info",
+                message: `Env-configured adapter "${liveName}" responded in ${latency}ms${preview ? ` \u2014 "${preview}"` : ""}.`
+              };
+            } catch (err) {
+              return {
+                ok: false,
+                severity: "error",
+                message: `Env-configured adapter "${liveName}" request failed: ${err?.message ?? String(err)}`
+              };
+            }
+          }
           return {
             ok: true,
             severity: "warning",
@@ -6061,7 +6251,7 @@ var AIServicePlugin = class {
           return {
             ok: false,
             severity: "error",
-            message: `Could not build adapter for provider=${provider}. Check API key and that the provider SDK package is installed.`
+            message: `Could not build adapter for provider=${provider}. Check API key (or the corresponding env var) and that the provider SDK package is installed.`
           };
         }
         const started = Date.now();
@@ -6078,6 +6268,15 @@ var AIServicePlugin = class {
             message: `${built.description} responded in ${latency}ms${preview ? ` \u2014 "${preview}"` : ""}.`
           };
         } catch (err) {
+          const isGwAuth = err?.name === "GatewayAuthenticationError";
+          const keyWasProvided = provider === "gateway" && String(merged.gateway_api_key ?? process.env.AI_GATEWAY_API_KEY ?? "").trim().length > 0;
+          if (isGwAuth && keyWasProvided) {
+            return {
+              ok: false,
+              severity: "error",
+              message: `${built.description}: API key was rejected by the AI Gateway (invalid, expired, or lacking access to model "${String(merged.gateway_model)}"). Create a new key at https://vercel.com/dashboard/ai-gateway/api-keys and re-save the settings.`
+            };
+          }
           return {
             ok: false,
             severity: "error",
@@ -6092,6 +6291,14 @@ var AIServicePlugin = class {
     this.service = void 0;
   }
 };
+function extractOverrides(payload) {
+  if (!payload || typeof payload !== "object") return {};
+  const p = payload;
+  if (p.values && typeof p.values === "object" && p.values !== null) {
+    return p.values;
+  }
+  return p;
+}
 
 // src/index.ts
 init_data_tools();