@neurocode-ai/core 1.18.8

package/test/plugin/host.ts

@@ -0,0 +1,303 @@
+import type { PluginContext } from "@neurocode-ai/plugin/v2/effect"
+import { AgentV2 } from "@neurocode-ai/core/agent"
+import { Catalog } from "@neurocode-ai/core/catalog"
+import { Credential } from "@neurocode-ai/core/credential"
+import { Integration } from "@neurocode-ai/core/integration"
+import { ModelV2 } from "@neurocode-ai/core/model"
+import { ProviderV2 } from "@neurocode-ai/core/provider"
+import type { IntegrationEnvMethod, IntegrationKeyMethod, IntegrationOAuthMethod } from "@neurocode-ai/sdk/v2/types"
+import { Effect } from "effect"
+
+type Overrides = Partial<Omit<PluginContext, "options">>
+
+export function host(overrides: Overrides = {}): PluginContext {
+  return {
+    options: {},
+    agent: overrides.agent ?? {
+      transform: () => Effect.die("unused agent.transform"),
+      reload: () => Effect.die("unused agent.reload"),
+    },
+    aisdk: overrides.aisdk ?? {
+      sdk: () => Effect.die("unused aisdk.sdk"),
+      language: () => Effect.die("unused aisdk.language"),
+    },
+    catalog: overrides.catalog ?? {
+      transform: () => Effect.die("unused catalog.transform"),
+      reload: () => Effect.die("unused catalog.reload"),
+    },
+    command: overrides.command ?? {
+      transform: () => Effect.die("unused command.transform"),
+      reload: () => Effect.die("unused command.reload"),
+    },
+    integration: overrides.integration ?? {
+      transform: () => Effect.die("unused integration.transform"),
+      reload: () => Effect.die("unused integration.reload"),
+      connection: {
+        active: () => Effect.die("unused integration.connection.active"),
+        resolve: () => Effect.die("unused integration.connection.resolve"),
+      },
+    },
+    plugin: overrides.plugin ?? {
+      add: () => Effect.die("unused plugin.add"),
+      remove: () => Effect.die("unused plugin.remove"),
+    },
+    reference: overrides.reference ?? {
+      transform: () => Effect.die("unused reference.transform"),
+      reload: () => Effect.die("unused reference.reload"),
+    },
+    skill: overrides.skill ?? {
+      transform: () => Effect.die("unused skill.transform"),
+      reload: () => Effect.die("unused skill.reload"),
+    },
+  }
+}
+
+export function agentHost(agent: AgentV2.Interface): PluginContext["agent"] {
+  return {
+    reload: agent.reload,
+    transform: (callback) =>
+      agent.transform((draft) =>
+        callback({
+          list: () => draft.list().map(agentInfo),
+          get: (id) => {
+            const value = draft.get(AgentV2.ID.make(id))
+            return value && agentInfo(value)
+          },
+          default: (id) => draft.default(id === undefined ? undefined : AgentV2.ID.make(id)),
+          update: (id, update) =>
+            draft.update(AgentV2.ID.make(id), (value) => {
+              const current = agentInfo(value)
+              update(current)
+              Object.assign(value, current, { id: AgentV2.ID.make(current.id) })
+            }),
+          remove: (id) => draft.remove(AgentV2.ID.make(id)),
+        }),
+      ),
+  }
+}
+
+export function catalogHost(catalog: Catalog.Interface): PluginContext["catalog"] {
+  return {
+    reload: catalog.reload,
+    transform: (callback) =>
+      catalog.transform((draft) =>
+        callback({
+          provider: {
+            list: () =>
+              draft.provider.list().map((value) => ({
+                provider: providerInfo(value.provider),
+                models: new Map(Array.from(value.models, ([id, model]) => [id, modelInfo(model)])),
+              })),
+            get: (id) => {
+              const value = draft.provider.get(ProviderV2.ID.make(id))
+              return (
+                value && {
+                  provider: providerInfo(value.provider),
+                  models: new Map(Array.from(value.models, ([id, model]) => [id, modelInfo(model)])),
+                }
+              )
+            },
+            update: (id, update) =>
+              draft.provider.update(ProviderV2.ID.make(id), (value) => {
+                const current = providerInfo(value)
+                update(current)
+                Object.assign(value, current, { id: ProviderV2.ID.make(current.id) })
+              }),
+            remove: (id) => draft.provider.remove(ProviderV2.ID.make(id)),
+          },
+          model: {
+            get: (providerID, modelID) => {
+              const value = draft.model.get(ProviderV2.ID.make(providerID), ModelV2.ID.make(modelID))
+              return value && modelInfo(value)
+            },
+            update: (providerID, modelID, update) =>
+              draft.model.update(ProviderV2.ID.make(providerID), ModelV2.ID.make(modelID), (value) => {
+                const current = modelInfo(value)
+                update(current)
+                Object.assign(value, current, {
+                  id: ModelV2.ID.make(current.id),
+                  providerID: ProviderV2.ID.make(current.providerID),
+                  family: current.family === undefined ? undefined : ModelV2.Family.make(current.family),
+                  variants: current.variants.map((variant) => ({
+                    ...variant,
+                    id: ModelV2.VariantID.make(variant.id),
+                  })),
+                })
+              }),
+            remove: (providerID, modelID) =>
+              draft.model.remove(ProviderV2.ID.make(providerID), ModelV2.ID.make(modelID)),
+            default: {
+              get: () => {
+                const value = draft.model.default.get()
+                return value && { providerID: value.providerID, modelID: value.modelID }
+              },
+              set: (providerID, modelID) =>
+                draft.model.default.set(ProviderV2.ID.make(providerID), ModelV2.ID.make(modelID)),
+            },
+          },
+        }),
+      ),
+  }
+}
+
+export function integrationHost(integration: Integration.Interface): PluginContext["integration"] {
+  return {
+    reload: integration.reload,
+    connection: {
+      active: (id) => integration.connection.active(Integration.ID.make(id)),
+      resolve: (connection) =>
+        integration.connection.resolve(
+          connection.type === "credential" ? { ...connection, id: Credential.ID.make(connection.id) } : connection,
+        ),
+    },
+    transform: (callback) =>
+      integration.transform((draft) =>
+        callback({
+          list: () => draft.list().map((value) => ({ id: value.id, name: value.name })),
+          get: (id) => {
+            const value = draft.get(Integration.ID.make(id))
+            return value && { id: value.id, name: value.name }
+          },
+          update: (id, update) => draft.update(Integration.ID.make(id), update),
+          remove: (id) => draft.remove(Integration.ID.make(id)),
+          method: {
+            list: (id) => draft.method.list(Integration.ID.make(id)).map(method),
+            update: (input) => {
+              if ("authorize" in input) {
+                const methodID = Integration.MethodID.make(input.method.id)
+                const refresh = input.refresh
+                draft.method.update({
+                  integrationID: Integration.ID.make(input.integrationID),
+                  method: { ...input.method, id: methodID },
+                  authorize: (inputs) =>
+                    input.authorize(inputs).pipe(
+                      Effect.map((authorization) => {
+                        if (authorization.mode === "auto") {
+                          return {
+                            ...authorization,
+                            callback: authorization.callback.pipe(
+                              Effect.map((credential) =>
+                                Credential.OAuth.make({
+                                  ...credential,
+                                  methodID: Integration.MethodID.make(credential.methodID),
+                                }),
+                              ),
+                            ),
+                          }
+                        }
+                        return {
+                          ...authorization,
+                          callback: (code: string) =>
+                            authorization.callback(code).pipe(
+                              Effect.map((credential) =>
+                                Credential.OAuth.make({
+                                  ...credential,
+                                  methodID: Integration.MethodID.make(credential.methodID),
+                                }),
+                              ),
+                            ),
+                        }
+                      }),
+                    ),
+                  ...(refresh
+                    ? {
+                        refresh: (value: Credential.OAuth) =>
+                          refresh(value).pipe(
+                            Effect.map((next) =>
+                              Credential.OAuth.make({
+                                ...next,
+                                methodID: Integration.MethodID.make(next.methodID),
+                              }),
+                            ),
+                          ),
+                      }
+                    : {}),
+                  ...(input.label ? { label: input.label } : {}),
+                })
+                return
+              }
+              if (input.method.type === "env") {
+                draft.method.update({
+                  integrationID: Integration.ID.make(input.integrationID),
+                  method: { ...input.method, names: [...input.method.names] },
+                })
+                return
+              }
+              draft.method.update({
+                integrationID: Integration.ID.make(input.integrationID),
+                method: input.method,
+              })
+            },
+            remove: (id, item) => draft.method.remove(Integration.ID.make(id), internalMethod(item)),
+          },
+        }),
+      ),
+  }
+}
+
+function method(value: Integration.Method) {
+  if (value.type === "env") return { type: value.type, names: [...value.names] }
+  if (value.type === "key") return { type: value.type, label: value.label }
+  return {
+    type: value.type,
+    id: value.id,
+    label: value.label,
+    prompts: value.prompts?.map((prompt) => {
+      if (prompt.type === "text") return { ...prompt }
+      return { ...prompt, options: prompt.options.map((option) => ({ ...option })) }
+    }),
+  }
+}
+
+function internalMethod(
+  value: IntegrationOAuthMethod | IntegrationKeyMethod | IntegrationEnvMethod,
+): Integration.Method {
+  if (value.type === "env") return value
+  if (value.type === "key") return value
+  return {
+    ...value,
+    id: Integration.MethodID.make(value.id),
+  }
+}
+
+function agentInfo(value: AgentV2.Info) {
+  return {
+    ...value,
+    model: value.model && { ...value.model },
+    request: { headers: { ...value.request.headers }, body: { ...value.request.body } },
+    permissions: value.permissions.map((permission) => ({ ...permission })),
+  }
+}
+
+function providerInfo(value: ProviderV2.MutableInfo) {
+  return {
+    ...value,
+    api: { ...value.api, settings: value.api.settings && { ...value.api.settings } },
+    request: { headers: { ...value.request.headers }, body: { ...value.request.body } },
+  }
+}
+
+function modelInfo(value: ModelV2.Info | ModelV2.MutableInfo) {
+  return {
+    ...value,
+    api: { ...value.api, settings: value.api.settings && { ...value.api.settings } },
+    capabilities: {
+      ...value.capabilities,
+      input: [...value.capabilities.input],
+      output: [...value.capabilities.output],
+    },
+    request: {
+      ...value.request,
+      headers: { ...value.request.headers },
+      body: { ...value.request.body },
+    },
+    variants: value.variants.map((variant) => ({
+      ...variant,
+      headers: { ...variant.headers },
+      body: { ...variant.body },
+    })),
+    time: { ...value.time },
+    cost: value.cost.map((cost) => ({ ...cost, tier: cost.tier && { ...cost.tier }, cache: { ...cost.cache } })),
+    limit: { ...value.limit },
+  }
+}

package/test/plugin/models-dev.test.ts

@@ -0,0 +1,170 @@
+import path from "path"
+import { describe, expect } from "bun:test"
+import { Effect, Layer } from "effect"
+import { Catalog } from "@neurocode-ai/core/catalog"
+import { Integration } from "@neurocode-ai/core/integration"
+import { AppNodeBuilder } from "@neurocode-ai/core/effect/app-node-builder"
+import { LayerNode } from "@neurocode-ai/core/effect/layer-node"
+import { EventV2 } from "@neurocode-ai/core/event"
+import { Flag } from "@neurocode-ai/core/flag/flag"
+import { Location } from "@neurocode-ai/core/location"
+import { ModelV2 } from "@neurocode-ai/core/model"
+import { ModelsDev } from "@neurocode-ai/core/models-dev"
+import { ModelsDevPlugin } from "@neurocode-ai/core/plugin/models-dev"
+import { ProviderV2 } from "@neurocode-ai/core/provider"
+import { AbsolutePath } from "@neurocode-ai/core/schema"
+import { location } from "../fixture/location"
+import { testEffect } from "../lib/effect"
+import { catalogHost, host, integrationHost } from "./host"
+
+const locationLayer = Layer.succeed(
+  Location.Service,
+  Location.Service.of(location({ directory: AbsolutePath.make(import.meta.dir) })),
+)
+const layer = AppNodeBuilder.build(LayerNode.group([Catalog.node, Integration.node, EventV2.node]), [
+  [Location.node, locationLayer],
+])
+const it = testEffect(layer)
+
+describe("ModelsDevPlugin", () => {
+  it.effect("projects models.dev modes as separate models instead of variants", () =>
+    Effect.gen(function* () {
+      const integrations = yield* Integration.Service
+      const catalog = yield* Catalog.Service
+      const models = ModelsDev.Service.of({
+        get: () =>
+          Effect.succeed({
+            acme: {
+              id: "acme",
+              name: "Acme",
+              env: [],
+              npm: "@ai-sdk/openai-compatible",
+              api: "https://api.acme.test/v1",
+              models: {
+                "gpt-5.4": {
+                  id: "gpt-5.4",
+                  name: "GPT-5.4",
+                  family: "gpt",
+                  release_date: "2026-01-01",
+                  attachment: false,
+                  reasoning: true,
+                  temperature: true,
+                  tool_call: true,
+                  cost: {
+                    input: 2.5,
+                    output: 15,
+                    tiers: [
+                      {
+                        tier: { type: "context", size: 272_000 },
+                        input: 3,
+                        output: 18,
+                        cache_read: 0.25,
+                      },
+                    ],
+                    context_over_200k: { input: 5, output: 22.5, cache_read: 0.5 },
+                  },
+                  limit: { context: 1_050_000, input: 922_000, output: 128_000 },
+                  experimental: {
+                    modes: {
+                      fast: {
+                        cost: { input: 5, output: 30, cache_read: 0.5 },
+                        provider: {
+                          headers: { "x-mode": "fast" },
+                          body: { service_tier: "priority" },
+                        },
+                      },
+                    },
+                  },
+                },
+              },
+            },
+          } satisfies Record<string, ModelsDev.Provider>),
+        refresh: () => Effect.void,
+      })
+
+      yield* ModelsDevPlugin.effect(
+        host({
+          catalog: catalogHost(catalog),
+          integration: integrationHost(integrations),
+        }),
+      ).pipe(Effect.provideService(ModelsDev.Service, models))
+
+      const providerID = ProviderV2.ID.make("acme")
+      const base = yield* catalog.model.get(providerID, ModelV2.ID.make("gpt-5.4"))
+      const fast = yield* catalog.model.get(providerID, ModelV2.ID.make("gpt-5.4-fast"))
+
+      expect(base?.variants).toEqual([])
+      expect(base?.request.body).toEqual({})
+      expect(fast).toMatchObject({
+        id: "gpt-5.4-fast",
+        providerID: "acme",
+        name: "GPT-5.4 Fast",
+        api: { id: "gpt-5.4" },
+        request: {
+          headers: { "x-mode": "fast" },
+          body: { service_tier: "priority" },
+        },
+        variants: [],
+      })
+      expect(fast?.cost).toEqual([
+        { input: 5, output: 30, cache: { read: 0.5, write: 0 } },
+        {
+          tier: { type: "context", size: 272_000 },
+          input: 3,
+          output: 18,
+          cache: { read: 0.25, write: 0 },
+        },
+        {
+          tier: { type: "context", size: 200_000 },
+          input: 5,
+          output: 22.5,
+          cache: { read: 0.5, write: 0 },
+        },
+      ])
+    }),
+  )
+
+  it.effect("registers key methods for providers with environment variables", () =>
+    Effect.acquireUseRelease(
+      Effect.sync(() => {
+        const previous = {
+          path: Flag.NEUROCODE_MODELS_PATH,
+          disabled: Flag.NEUROCODE_DISABLE_MODELS_FETCH,
+        }
+        Flag.NEUROCODE_MODELS_PATH = path.join(import.meta.dir, "fixtures", "models-dev.json")
+        Flag.NEUROCODE_DISABLE_MODELS_FETCH = true
+        return previous
+      }),
+      () =>
+        Effect.gen(function* () {
+          const integrations = yield* Integration.Service
+          const catalog = yield* Catalog.Service
+          yield* ModelsDevPlugin.effect(
+            host({
+              catalog: catalogHost(catalog),
+              integration: integrationHost(integrations),
+            }),
+          )
+          expect(yield* integrations.list()).toEqual([
+            new Integration.Info({
+              id: Integration.ID.make("acme"),
+              name: "Acme",
+              methods: [
+                { type: "key" },
+                {
+                  type: "env",
+                  names: ["ACME_API_KEY"],
+                },
+              ],
+              connections: [],
+            }),
+          ])
+        }).pipe(Effect.provide(AppNodeBuilder.build(ModelsDev.node))),
+      (previous) =>
+        Effect.sync(() => {
+          Flag.NEUROCODE_MODELS_PATH = previous.path
+          Flag.NEUROCODE_DISABLE_MODELS_FETCH = previous.disabled
+        }),
+    ),
+  )
+})

package/test/plugin/promise.test.ts

@@ -0,0 +1,67 @@
+import { describe, expect } from "bun:test"
+import { Effect } from "effect"
+import { AgentV2 } from "@neurocode-ai/core/agent"
+import { PluginV2 } from "@neurocode-ai/core/plugin"
+import { PluginHost } from "@neurocode-ai/core/plugin/host"
+import { PluginPromise } from "@neurocode-ai/core/plugin/promise"
+import { define } from "@neurocode-ai/plugin/v2/promise"
+import { testEffect } from "../lib/effect"
+import { PluginTestLayer } from "./fixture"
+
+const it = testEffect(PluginTestLayer)
+
+describe("fromPromise", () => {
+  it.effect("loads a promise plugin and registers a transform hook", () =>
+    Effect.gen(function* () {
+      const agents = yield* AgentV2.Service
+      const plugin = yield* PluginV2.Service
+      const host = yield* PluginHost.make(plugin)
+
+      const promisePlugin = define({
+        id: "promise-example",
+        setup: async (ctx) => {
+          expect(ctx.options.mode).toBe("strict")
+          await ctx.agent.transform((draft) => {
+            draft.update("reviewer", (item) => {
+              item.description = "Reviews code"
+              item.mode = "subagent"
+            })
+          })
+        },
+      })
+
+      const adapted = PluginPromise.fromPromise(promisePlugin)
+      yield* adapted.effect({ ...host, options: { mode: "strict" } })
+
+      expect(yield* agents.get(AgentV2.ID.make("reviewer"))).toMatchObject({
+        description: "Reviews code",
+        mode: "subagent",
+      })
+    }),
+  )
+
+  it.effect("disposes a hook registration on request", () =>
+    Effect.gen(function* () {
+      const agents = yield* AgentV2.Service
+      const plugin = yield* PluginV2.Service
+      const host = yield* PluginHost.make(plugin)
+
+      const promisePlugin = define({
+        id: "promise-dispose",
+        setup: async (ctx) => {
+          const registration = await ctx.agent.transform((draft) => {
+            draft.update("temp", (item) => {
+              item.description = "temporary"
+            })
+          })
+          await registration.dispose()
+        },
+      })
+
+      const adapted = PluginPromise.fromPromise(promisePlugin)
+      yield* adapted.effect(host)
+
+      expect(yield* agents.get(AgentV2.ID.make("temp"))).toBeUndefined()
+    }),
+  )
+})

package/test/plugin/provider-alibaba.test.ts

@@ -0,0 +1,92 @@
+import { AISDK } from "@neurocode-ai/core/aisdk"
+import { describe, expect } from "bun:test"
+import { createAlibaba } from "@ai-sdk/alibaba"
+import { Effect } from "effect"
+import { ModelV2 } from "@neurocode-ai/core/model"
+import { PluginV2 } from "@neurocode-ai/core/plugin"
+import { PluginHost } from "@neurocode-ai/core/plugin/host"
+import { AlibabaPlugin } from "@neurocode-ai/core/plugin/provider/alibaba"
+import { ProviderV2 } from "@neurocode-ai/core/provider"
+import { testEffect } from "../lib/effect"
+import { PluginTestLayer } from "./fixture"
+
+const it = testEffect(PluginTestLayer)
+
+const addPlugin = Effect.fn(function* () {
+  const plugin = yield* PluginV2.Service
+  const aisdk = yield* AISDK.Service
+  const host = yield* PluginHost.make(plugin)
+  yield* AlibabaPlugin.effect(host)
+})
+
+describe("AlibabaPlugin", () => {
+  it.effect("creates an Alibaba SDK for @ai-sdk/alibaba", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const result = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("alibaba"), ModelV2.ID.make("qwen")),
+          api: { id: ModelV2.ID.make("qwen"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "@ai-sdk/alibaba",
+        options: { name: "alibaba" },
+      })
+      expect(result.sdk).toBeDefined()
+    }),
+  )
+
+  it.effect("ignores non-Alibaba SDK packages", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const result = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("alibaba"), ModelV2.ID.make("qwen")),
+          api: { id: ModelV2.ID.make("qwen"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "@ai-sdk/openai-compatible",
+        options: { name: "alibaba" },
+      })
+      expect(result.sdk).toBeUndefined()
+    }),
+  )
+
+  it.effect("matches the old bundled Alibaba SDK provider naming", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const result = yield* aisdk.runSDK({
+        model: ModelV2.Info.make({
+          ...ModelV2.Info.empty(ProviderV2.ID.make("custom-alibaba"), ModelV2.ID.make("qwen")),
+          api: { id: ModelV2.ID.make("qwen"), type: "aisdk", package: "test-provider" },
+        }),
+        package: "@ai-sdk/alibaba",
+        options: { name: "custom-alibaba", apiKey: "test" },
+      })
+      const expected = createAlibaba({ apiKey: "test", ...{ name: "custom-alibaba" } }).languageModel("qwen")
+      const actual = result.sdk?.languageModel("qwen")
+      expect(actual?.provider).toBe(expected.provider)
+      expect(actual?.modelId).toBe(expected.modelId)
+    }),
+  )
+
+  it.effect("uses the old default languageModel(api.id) behavior", () =>
+    Effect.gen(function* () {
+      const plugin = yield* PluginV2.Service
+      const aisdk = yield* AISDK.Service
+      yield* addPlugin()
+      const item = ModelV2.Info.make({
+        ...ModelV2.Info.empty(ProviderV2.ID.make("alibaba"), ModelV2.ID.make("alias")),
+        api: { id: ModelV2.ID.make("qwen-plus"), type: "aisdk", package: "test-provider" },
+      })
+      const result = yield* aisdk.runSDK({ model: item, package: "@ai-sdk/alibaba", options: {} })
+      const language = result.sdk?.languageModel(item.api.id)
+      expect(language?.modelId).toBe("qwen-plus")
+      expect(language?.provider).toBe("alibaba.chat")
+    }),
+  )
+})