AgentFlow | AI Orchestration Platform

\n\n\n`;\n}\n\n// --- HONO APPLICATION SETUP ---\nconst app = new Hono();\n\n// --- API ROUTES ---\n\n// Serves the main unified HTML page\napp.get(\"/\", (c: Context) => c.html(generateHtml(c.req.url)));\n\n// --- AgentFlow API Routes ---\napp.post(\"/api/company-lookup\", async (c: Context) => {\n const { companyName } = await c.req.json();\n const prompt = BUSINESS_RESEARCHER_AGENT.replace(\n \"{{company_name}}\",\n companyName,\n );\n try {\n const result = await services.callAI(\n \"You are a helpful assistant that only speaks JSON.\",\n prompt,\n {\n isJson: true,\n model: config.models.minor,\n },\n );\n return c.json(result);\n } catch (e) {\n console.error(\"Company lookup failed:\", e);\n return c.json({ company_name: companyName, description: null }, 500);\n }\n});\n\napp.post(\"/api/enrich-context\", async (c: Context) => {\n const { history, context } = await c.req.json();\n const historyString = history.map((h) => `${h.role}: ${h.content}`).join(\n \"\\n\",\n );\n const prompt = CONTEXT_ENRICHMENT_AGENT\n .replace(\"{{current_context}}\", JSON.stringify(context))\n .replace(\n \"{{history}}\",\n historyString || \"No history yet. Start with your first question.\",\n );\n try {\n const completion = await services.callAI(\n \"You are a helpful assistant following strict output rules.\",\n prompt,\n { model: config.models.minor },\n );\n const responseText = completion.trim();\n if (responseText.startsWith(\"__COMPLETE__\")) {\n const jsonPart = responseText.replace(\"__COMPLETE__\", \"\");\n return c.json({ status: \"complete\", data: JSON.parse(jsonPart) });\n } else {\n return c.json({ status: \"in-progress\", question: responseText });\n }\n } catch (e) {\n console.error(\"Context enrichment agent failed:\", e);\n return c.json({\n status: \"in-progress\",\n question: \"I'm having trouble. Could you please rephrase?\",\n }, 500);\n }\n});\n\napp.post(\"/api/gather\", async (c: Context) => {\n const { history, company_context } = await c.req.json();\n const historyString = history.map((h) => `${h.role}: ${h.content}`).join(\n \"\\n\",\n );\n const prompt = INFORMATION_GATHERER\n .replace(\"{{company_context}}\", JSON.stringify(company_context))\n .replace(\n \"{{history}}\",\n historyString ||\n \"No history yet. Start with your most important question.\",\n );\n try {\n const completion = await services.callAI(\n \"You are a helpful assistant following strict output rules.\",\n prompt,\n { model: config.models.minor },\n );\n const responseText = completion.trim();\n if (responseText.startsWith(\"__COMPLETE__\")) {\n const jsonPart = responseText.replace(\"__COMPLETE__\", \"\");\n return c.json({ status: \"complete\", data: JSON.parse(jsonPart) });\n } else {\n return c.json({ status: \"in-progress\", question: responseText });\n }\n } catch (e) {\n console.error(\"Gathering agent failed:\", e);\n return c.json({\n status: \"in-progress\",\n question:\n \"I'm having a little trouble with that. Could you please rephrase?\",\n }, 500);\n }\n});\n\napp.get(\"/api/execute-flow\", (c: Context) => {\n return streamSSE(c, async (streamApi) => {\n streamApi.onAbort(() => console.log(\"Client disconnected.\"));\n const goal = c.req.query(\"goal\") || \"No goal provided\";\n let company_context: CompanyContext;\n try {\n company_context = JSON.parse(c.req.query(\"context\") || \"{}\");\n if (!company_context.company_name) {\n throw new Error(\"Company context is missing or invalid.\");\n }\n } catch (e) {\n console.error(\"Context parsing failed:\", e);\n await streamApi.writeSSE({\n event: \"error\",\n data: `Invalid context provided: ${e.message}`,\n });\n return;\n }\n\n try {\n // Stage 1: Orchestrator plans the assembly line\n const planPrompt = ORCHESTRATOR_AGENT\n .replace(\"{{goal}}\", goal)\n .replace(\"{{company_context}}\", JSON.stringify(company_context));\n const planResult = await services.callAI(\n \"You are a helpful assistant that only speaks JSON.\",\n planPrompt,\n { isJson: true, model: config.models.minor },\n );\n const planMap = new Map();\n planResult.plan.forEach((step) =>\n planMap.set(step.id, {\n ...step,\n status: \"pending\",\n output: \"\",\n critique: null,\n })\n );\n const planArray = Array.from(planMap.values());\n await streamApi.writeSSE({\n event: \"plan_complete\",\n data: JSON.stringify(planArray),\n });\n\n // Stage 2 & 3: Execute and Critique steps with dependency checking\n let completedSteps = new Set();\n while (completedSteps.size < planArray.length) {\n const availableSteps = planArray.filter(\n (step) =>\n step.status === \"pending\" &&\n step.dependencies.every((depId) => completedSteps.has(depId)),\n );\n if (availableSteps.length === 0) {\n // If no steps are available but we are not done, there might be a cycle or error\n if (\n Array.from(planMap.values()).some((s) => s.status === \"pending\")\n ) {\n console.error(\n \"Execution stalled. Check for circular dependencies.\",\n );\n throw new Error(\"Execution stalled due to dependency issue.\");\n }\n break; // All running/completed\n }\n\n await Promise.all(availableSteps.map(async (step) => {\n planMap.get(step.id).status = \"running\";\n try {\n await streamApi.writeSSE({\n event: \"step_start\",\n data: JSON.stringify({\n id: step.id,\n message: `Executing: ${step.agent_persona}`,\n }),\n });\n const previousWork = step.dependencies\n .map((depId) => {\n const depStep = planMap.get(depId);\n return depStep.status === \"complete\"\n ? `\\n\\n---\\n\\n## Output from ${depStep.agent_persona} (Step ${depStep.id}):\\n\\n${depStep.output}`\n : \"\";\n }).join(\"\\n\");\n\n const executorPrompt = EXECUTOR_AGENT\n .replace(\"{{company_name}}\", company_context.company_name)\n .replace(\n \"{{company_description}}\",\n company_context.company_description,\n )\n .replace(\"{{use_case}}\", company_context.use_case)\n .replace(\"{{language}}\", company_context.language)\n .replace(\"{{agent_persona}}\", step.agent_persona)\n .replace(\n \"{{previous_work}}\",\n previousWork || \"This is the first step. Start from scratch.\",\n )\n .replace(\"{{task}}\", step.task);\n\n const stepResult = await services.callAIStreaming(\n executorPrompt,\n \"Proceed with your task. Generate the deliverable now.\",\n async (delta) => {\n await streamApi.writeSSE({\n event: \"stream\",\n data: JSON.stringify({ id: step.id, delta }),\n });\n },\n );\n planMap.get(step.id).output = stepResult;\n await streamApi.writeSSE({\n event: \"step_complete\",\n data: JSON.stringify({ id: step.id }),\n });\n\n const critiquePrompt = JSON.stringify({\n project_goal: goal,\n task: step.task,\n raw_output: stepResult,\n language: company_context.language,\n });\n const critiqueResult = await services.callAI(\n CRITIQUE_AGENT,\n critiquePrompt,\n { isJson: true, model: config.models.minor },\n );\n planMap.get(step.id).critique = critiqueResult;\n planMap.get(step.id).status = \"complete\";\n completedSteps.add(step.id);\n await streamApi.writeSSE({\n event: \"critique_complete\",\n data: JSON.stringify({ id: step.id, critique: critiqueResult }),\n });\n } catch (error) {\n console.error(`Step ${step.id} failed:`, error);\n planMap.get(step.id).status = \"error\";\n completedSteps.add(step.id); // Also add failed steps to avoid blocking\n await streamApi.writeSSE({\n event: \"step_error\",\n data: JSON.stringify({ id: step.id, error: error.message }),\n });\n }\n }));\n }\n await streamApi.writeSSE({\n event: \"project_complete\",\n data: \"Project finished\",\n });\n } catch (error) {\n console.error(\"Project flow failed:\", error);\n await streamApi.writeSSE({\n event: \"error\",\n data: `An error occurred: ${error.message}`,\n });\n }\n });\n});\n\n// --- Totem API Routes (Integrated into Hono) ---\napp.post(\"/api/totem/start\", async (c: Context) => {\n const { claim } = await c.req.json();\n if (!claim || typeof claim !== \"string\" || claim.trim().length === 0) {\n return c.json(\n { error: \"A valid, non-empty 'claim' string is required.\" },\n 400,\n );\n }\n const jobId = nanoid();\n const rootClaim: ClaimNode = {\n id: nanoid(),\n claim: claim,\n children: [],\n depth: 0,\n };\n await blob.setJSON(jobId, { status: \"processing\", totem: rootClaim });\n // This runs in the background and does not block the response\n buildTotemInBackground(jobId, rootClaim);\n return c.json({ jobId });\n});\n\napp.get(\"/api/totem/status\", async (c: Context) => {\n const jobId = c.req.query(\"jobId\");\n if (!jobId) {\n return c.json({ error: \"A 'jobId' query parameter is required.\" }, 400);\n }\n const state: JobState | undefined = await blob.getJSON(jobId);\n if (!state) {\n return c.json({ status: \"error\", error: \"Job not found.\" }, 404);\n }\n return c.json(state);\n});\n\napp.get(\"/api/totem/list\", async (c: Context) => {\n const index = await blob.getJSON(TOTEM_INDEX_KEY) || [];\n return c.json(index);\n});\n\napp.delete(\"/api/totem/delete\", async (c: Context) => {\n const jobId = c.req.query(\"jobId\");\n if (!jobId) {\n return c.json({ error: \"A 'jobId' query parameter is required.\" }, 400);\n }\n const currentIndex: TotemIndexEntry[] = await blob.getJSON(TOTEM_INDEX_KEY) ||\n [];\n const newIndex = currentIndex.filter((item) => item.jobId !== jobId);\n await blob.setJSON(TOTEM_INDEX_KEY, newIndex);\n await blob.delete(jobId);\n return c.json({ success: true, jobId });\n});\n\nexport default app.fetch;","deleted","label",[99],"permalink","https://www.val.town/x/volt/agentflow/code/main.ts","extension","actionData","errors"]