progress

packages/opencode/src/agent/agent.ts

@@ -1,10 +1,11 @@
-import { App } from "../app/app"
 import { Config } from "../config/config"
 import z from "zod"
 import { Provider } from "../provider/provider"
 import { generateObject, type ModelMessage } from "ai"
 import PROMPT_GENERATE from "./generate.txt"
 import { SystemPrompt } from "../session/system"
+import { State } from "../project/state"
+import { Session } from "../session"
 
 export namespace Agent {
   export const Info = z
@@ -24,46 +25,49 @@ export namespace Agent {
       ref: "Agent",
     })
   export type Info = z.infer<typeof Info>
-  const state = App.state("agent", async () => {
-    const cfg = await Config.get()
-    const result: Record<string, Info> = {
-      general: {
-        name: "general",
-        description:
-          "General-purpose agent for researching complex questions, searching for code, and executing multi-step tasks. When you are searching for a keyword or file and are not confident that you will find the right match in the first few tries use this agent to perform the search for you.",
-        tools: {
-          todoread: false,
-          todowrite: false,
-        },
-      },
-    }
-    for (const [key, value] of Object.entries(cfg.agent ?? {})) {
-      if (value.disable) {
-        delete result[key]
-        continue
-      }
-      let item = result[key]
-      if (!item)
-        item = result[key] = {
-          name: key,
-          description: "",
+  const state = State.create(
+    () => Session.use().directory,
+    async () => {
+      const cfg = await Config.get()
+      const result: Record<string, Info> = {
+        general: {
+          name: "general",
+          description:
+            "General-purpose agent for researching complex questions, searching for code, and executing multi-step tasks. When you are searching for a keyword or file and are not confident that you will find the right match in the first few tries use this agent to perform the search for you.",
           tools: {
-            todowrite: false,
             todoread: false,
+            todowrite: false,
           },
+        },
+      }
+      for (const [key, value] of Object.entries(cfg.agent ?? {})) {
+        if (value.disable) {
+          delete result[key]
+          continue
         }
-      const model = value.model ?? cfg.model
-      if (model) item.model = Provider.parseModel(model)
-      if (value.prompt) item.prompt = value.prompt
-      if (value.tools)
-        item.tools = {
-          ...item.tools,
-          ...value.tools,
-        }
-      if (value.description) item.description = value.description
-    }
-    return result
-  })
+        let item = result[key]
+        if (!item)
+          item = result[key] = {
+            name: key,
+            description: "",
+            tools: {
+              todowrite: false,
+              todoread: false,
+            },
+          }
+        const model = value.model ?? cfg.model
+        if (model) item.model = Provider.parseModel(model)
+        if (value.prompt) item.prompt = value.prompt
+        if (value.tools)
+          item.tools = {
+            ...item.tools,
+            ...value.tools,
+          }
+        if (value.description) item.description = value.description
+      }
+      return result
+    },
+  )
 
   export async function get(agent: string) {
     return state().then((x) => x[agent])