openAgent

1import { Effect, Layer, LayerMap } from "effect"
2import { AgentV2 } from "./agent"
3import { AISDK } from "./aisdk"
4import { Catalog } from "./catalog"
5import { CommandV2 } from "./command"
6import { Config } from "./config"
7import { LayerNode } from "./effect/layer-node"
8import { Node } from "./effect/app-node"
9import { FileMutation } from "./file-mutation"
10import { FileSystem } from "./filesystem"
11import { FileSystemSearch } from "./filesystem/search"
12import { Watcher } from "./filesystem/watcher"
13import { Image } from "./image"
14import { Integration } from "./integration"
15import { Location } from "./location"
16import { LocationMutation } from "./location-mutation"
17import { LocationServiceMap } from "./location-service-map"
18import { PermissionV2 } from "./permission"
19import { PluginV2 } from "./plugin"
20import { PluginInternal } from "./plugin/internal"
21import { Policy } from "./policy"
22import { ProjectCopy } from "./project/copy"
23import { Pty } from "./pty"
24import { QuestionV2 } from "./question"
25import { Reference } from "./reference"
26import { ReferenceGuidance } from "./reference/guidance"
27import * as SessionRunnerLLM from "./session/runner/llm"
28import { SessionRunnerModel } from "./session/runner/model"
29import { SessionTodo } from "./session/todo"
30import { SkillV2 } from "./skill"
31import { SkillGuidance } from "./skill/guidance"
32import { Snapshot } from "./snapshot"
33import { SystemContextBuiltIns } from "./system-context/builtins"
34import { SystemContextRegistry } from "./system-context/registry"
35import { BuiltInTools } from "./tool/builtins"
36import { ReadToolFileSystem } from "./tool/read-filesystem"
37import { ToolRegistry } from "./tool/registry"
38import { ToolOutputStore } from "./tool-output-store"
39 
40export { LocationServiceMap } from "./location-service-map"
41 
42export const locationServices = LayerNode.group([
43  Location.node,
44  Policy.node,
45  Config.node,
46  AgentV2.node,
47  CommandV2.node,
48  Reference.node,
49  Integration.node,
50  Catalog.node,
51  AISDK.node,
52  PluginV2.node,
53  PluginInternal.node,
54  ProjectCopy.node,
55  ProjectCopy.refreshNode,
56  FileSystemSearch.node,
57  FileSystem.node,
58  Watcher.node,
59  Pty.node,
60  SkillV2.node,
61  SystemContextRegistry.node,
62  SystemContextBuiltIns.node,
63  LocationMutation.node,
64  FileMutation.node,
65  PermissionV2.node,
66  ToolOutputStore.node,
67  ToolRegistry.node,
68  ToolRegistry.toolsNode,
69  Image.node,
70  SkillGuidance.node,
71  ReferenceGuidance.node,
72  SessionTodo.node,
73  QuestionV2.node,
74  ReadToolFileSystem.node,
75  BuiltInTools.node,
76  SessionRunnerModel.node,
77  Snapshot.node,
78  SessionRunnerLLM.node,
79])
80 
81export type LocationServices = LayerNode.Output<typeof locationServices>
82export type LocationError = LayerNode.Error<typeof locationServices>
83 
84export function buildLocationServiceMap(
85  replacements: LayerNode.Replacements = [],
86): Layer.Layer<LocationServiceMap.Service> {
87  return Layer.effect(
88    LocationServiceMap.Service,
89    LayerMap.make(
90      (ref: Location.Ref) => {
91        const allReplacements = replacements.concat([[Location.node, Location.boundNode(ref)]])
92        // Apply replacements during hoist, not afterward: replacements can
93        // introduce new tagged dependencies (Location.boundNode depends on
94        // Project), and the hoist walk is the only pass that can still slice
95        // those back out.
96        const location = LayerNode.hoist(locationServices, Node.tags.values.global, allReplacements)
97 
98        return LayerNode.compile(location.node).pipe(
99          Layer.fresh,
100          Layer.tap(() =>
101            Effect.logInfo("booting location services", {
102              directory: ref.directory,
103              workspaceID: ref.workspaceID,
104            }),
105          ),
106          Layer.provide(LayerNode.compile(location.hoisted)),
107        )
108      },
109      { idleTimeToLive: "60 minutes" },
110    ),
111  )
112}
113 
114// This is temporary for backwards compatibility
115export const locationServiceMapLayer = buildLocationServiceMap()
116