@truefoundry/trueforge 0.2.0-rc.6 → 0.2.0-rc.8

package/dist/main.js

@@ -8107,7 +8107,8 @@ var ImportAgentItemSchema = z3.object({
   description: AgentDescriptionSchema.optional(),
   manifest: AgentSpecSchema5,
   tenant_id: z3.string().min(1).describe("Tenant to create the agent under."),
-  created_by_subject: CreatedBySubjectSchema3.describe("Original creator to persist on the agent.")
+  created_by_subject: CreatedBySubjectSchema3.describe("Original creator to persist on the agent."),
+  truefoundry_managed_agent_id: z3.string()
 }).strict().openapi("ImportAgentItem");
 var ImportAgentsRequestSchema = z3.object({
   agents: z3.array(ImportAgentItemSchema).describe("Agents to create; each carries tenant_id and created_by_subject.")
@@ -8729,6 +8730,7 @@ var TrueFoundryServiceFoundryServerClient = class {
   }
   /** PUT `/internal/tfg/agents` — create/reuse remote agent + sync model/MCP grants. */
   async putRemoteAgent(input) {
+    const hasTrueFoundryManagedAgentId = input.trueFoundryManagedAgentId !== void 0;
     const payload = await this.#requestJson({
       url: this.#url(TFG_AGENTS_PATH),
       accessToken: input.accessToken,
@@ -8738,7 +8740,8 @@ var TrueFoundryServiceFoundryServerClient = class {
         name: input.name,
         description: input.description,
         model: input.model,
-        mcp_servers: input.mcp_servers
+        mcp_servers: input.mcp_servers,
+        ...hasTrueFoundryManagedAgentId ? { trueFoundryManagedAgentId: input.trueFoundryManagedAgentId } : {}
       }
     });
     const parsed = PutRemoteAgentResponseSchema.safeParse(payload);
@@ -9125,7 +9128,10 @@ function createAgentImportRouter(deps) {
           description: agent.description ?? agent.name,
           manifest: agent.manifest,
           external_id: null,
-          created_by_subject: agent.created_by_subject
+          created_by_subject: agent.created_by_subject,
+          custom: {
+            trueFoundryManagedAgentId: agent.truefoundry_managed_agent_id
+          }
         });
         results.push({
           name: created.name,
@@ -17758,13 +17764,15 @@ function asError(value) {
 function toPutRemoteAgentPayload({
   name,
   description,
-  manifest
+  manifest,
+  trueFoundryManagedAgentId
 }) {
   return {
     name,
     description: (description || name).slice(0, AGENT_DESCRIPTION_MAX_LENGTH),
     model: manifest.model.name,
-    mcp_servers: (manifest.mcp_servers ?? []).map((server) => server.name)
+    mcp_servers: (manifest.mcp_servers ?? []).map((server) => server.name),
+    ...trueFoundryManagedAgentId ? { trueFoundryManagedAgentId } : {}
   };
 }
 var TrueFoundryAgentStore = class {
@@ -17806,12 +17814,14 @@ var TrueFoundryAgentStore = class {
     const created = await this.#inner.createAgent({ ...input, external_id: null }, transaction);
     let externalId;
     try {
+      const trueFoundryManagedAgentId = input.custom?.["trueFoundryManagedAgentId"];
       ({ externalId } = await this.#client.putRemoteAgent({
         accessToken: await this.#resolveAccessToken(),
         ...toPutRemoteAgentPayload({
           name: input.name,
           description: input.description,
-          manifest: input.manifest
+          manifest: input.manifest,
+          ...trueFoundryManagedAgentId ? { trueFoundryManagedAgentId } : {}
         })
       }));
       const updated = await this.#inner.updateAgent(