more

Author: waleedlatif1

apps/sim/app/api/a2a/serve/[agentId]/route.ts

@@ -5,7 +5,7 @@ import { createLogger } from '@sim/logger'
 import { eq } from 'drizzle-orm'
 import { type NextRequest, NextResponse } from 'next/server'
 import { v4 as uuidv4 } from 'uuid'
-import { A2A_DEFAULT_TIMEOUT } from '@/lib/a2a/constants'
+import { A2A_DEFAULT_TIMEOUT, A2A_MAX_HISTORY_LENGTH } from '@/lib/a2a/constants'
 import { notifyTaskStateChange } from '@/lib/a2a/push-notifications'
 import {
   createAgentMessage,
@@ -332,6 +332,11 @@ async function handleMessageSend(
 
     history.push(message)
 
+    // Truncate history to prevent unbounded JSONB growth
+    if (history.length > A2A_MAX_HISTORY_LENGTH) {
+      history.splice(0, history.length - A2A_MAX_HISTORY_LENGTH)
+    }
+
     if (existingTask) {
       await db
         .update(a2aTask)
@@ -537,6 +542,11 @@ async function handleMessageStream(
 
   history.push(message)
 
+  // Truncate history to prevent unbounded JSONB growth
+  if (history.length > A2A_MAX_HISTORY_LENGTH) {
+    history.splice(0, history.length - A2A_MAX_HISTORY_LENGTH)
+  }
+
   if (existingTask) {
     await db
       .update(a2aTask)

apps/sim/app/workspace/[workspaceId]/w/[workflowId]/components/panel/components/deploy/components/deploy-modal/components/a2a/a2a.tsx

@@ -617,12 +617,13 @@ export function A2aDeploy({
 
       case 'python':
         return requiresAuth
-          ? `import requests
+          ? `import os
+import requests
 
 response = requests.post(
     "${endpoint}",
     headers={
-        "X-API-Key": SIM_API_KEY,
+        "X-API-Key": os.environ.get("SIM_API_KEY"),
         "Content-Type": "application/json"
     },
     json=${JSON.stringify(payload, null, 4).replace(/\n/g, '\n    ')}
@@ -644,7 +645,7 @@ print(response.json())`
           ? `const response = await fetch("${endpoint}", {
   method: "POST",
   headers: {
-    "X-API-Key": SIM_API_KEY,
+    "X-API-Key": process.env.SIM_API_KEY,
     "Content-Type": "application/json"
   },
   body: JSON.stringify(${JSON.stringify(payload)})
@@ -666,7 +667,7 @@ console.log(data);`
           ? `const response = await fetch("${endpoint}", {
   method: "POST",
   headers: {
-    "X-API-Key": SIM_API_KEY,
+    "X-API-Key": process.env.SIM_API_KEY,
     "Content-Type": "application/json"
   },
   body: JSON.stringify(${JSON.stringify(payload)})

apps/sim/app/workspace/[workspaceId]/w/[workflowId]/components/panel/components/deploy/components/deploy-modal/components/api/api.tsx

@@ -125,12 +125,13 @@ export function ApiDeploy({
   ${endpoint}`
 
       case 'python':
-        return `import requests
+        return `import os
+import requests
 
 response = requests.post(
     "${endpoint}",
     headers={
-        "X-API-Key": SIM_API_KEY,
+        "X-API-Key": os.environ.get("SIM_API_KEY"),
         "Content-Type": "application/json"
     },
     json=${JSON.stringify(payload, null, 4).replace(/\n/g, '\n    ')}
@@ -142,7 +143,7 @@ print(response.json())`
         return `const response = await fetch("${endpoint}", {
   method: "POST",
   headers: {
-    "X-API-Key": SIM_API_KEY,
+    "X-API-Key": process.env.SIM_API_KEY,
     "Content-Type": "application/json"
   },
   body: JSON.stringify(${JSON.stringify(payload)})
@@ -155,7 +156,7 @@ console.log(data);`
         return `const response = await fetch("${endpoint}", {
   method: "POST",
   headers: {
-    "X-API-Key": SIM_API_KEY,
+    "X-API-Key": process.env.SIM_API_KEY,
     "Content-Type": "application/json"
   },
   body: JSON.stringify(${JSON.stringify(payload)})
@@ -183,12 +184,13 @@ console.log(data);`
   ${endpoint}`
 
       case 'python':
-        return `import requests
+        return `import os
+import requests
 
 response = requests.post(
     "${endpoint}",
     headers={
-        "X-API-Key": SIM_API_KEY,
+        "X-API-Key": os.environ.get("SIM_API_KEY"),
         "Content-Type": "application/json"
     },
     json=${JSON.stringify(payload, null, 4).replace(/\n/g, '\n    ')},
@@ -203,7 +205,7 @@ for line in response.iter_lines():
         return `const response = await fetch("${endpoint}", {
   method: "POST",
   headers: {
-    "X-API-Key": SIM_API_KEY,
+    "X-API-Key": process.env.SIM_API_KEY,
     "Content-Type": "application/json"
   },
   body: JSON.stringify(${JSON.stringify(payload)})
@@ -222,7 +224,7 @@ while (true) {
         return `const response = await fetch("${endpoint}", {
   method: "POST",
   headers: {
-    "X-API-Key": SIM_API_KEY,
+    "X-API-Key": process.env.SIM_API_KEY,
     "Content-Type": "application/json"
   },
   body: JSON.stringify(${JSON.stringify(payload)})
@@ -265,7 +267,7 @@ while (true) {
 response = requests.post(
     "${endpoint}",
     headers={
-        "X-API-Key": SIM_API_KEY,
+        "X-API-Key": process.env.SIM_API_KEY,
         "Content-Type": "application/json",
         "X-Execution-Mode": "async"
     },
@@ -279,7 +281,7 @@ print(job)  # Contains job_id for status checking`
             return `const response = await fetch("${endpoint}", {
   method: "POST",
   headers: {
-    "X-API-Key": SIM_API_KEY,
+    "X-API-Key": process.env.SIM_API_KEY,
     "Content-Type": "application/json",
     "X-Execution-Mode": "async"
   },
@@ -293,7 +295,7 @@ console.log(job); // Contains job_id for status checking`
             return `const response = await fetch("${endpoint}", {
   method: "POST",
   headers: {
-    "X-API-Key": SIM_API_KEY,
+    "X-API-Key": process.env.SIM_API_KEY,
     "Content-Type": "application/json",
     "X-Execution-Mode": "async"
   },