session-skill.test.ts 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. import path from "path"
  2. import { describe, expect } from "bun:test"
  3. import { Effect, Layer, LayerMap } from "effect"
  4. import { Database } from "@opencode-ai/core/database/database"
  5. import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
  6. import { LayerNode } from "@opencode-ai/util/effect/layer-node"
  7. import { EventV2 } from "@opencode-ai/core/event"
  8. import { Location } from "@opencode-ai/core/location"
  9. import { LocationServiceMap } from "@opencode-ai/core/location-service-map"
  10. import type { LocationServices } from "@opencode-ai/core/location-services"
  11. import { ProjectV2 } from "@opencode-ai/core/project"
  12. import { AbsolutePath } from "@opencode-ai/core/schema"
  13. import { SessionV2 } from "@opencode-ai/core/session"
  14. import { SessionExecution } from "@opencode-ai/core/session/execution"
  15. import { SessionMessage } from "@opencode-ai/core/session/message"
  16. import { SessionProjector } from "@opencode-ai/core/session/projector"
  17. import { SessionStore } from "@opencode-ai/core/session/store"
  18. import { SkillV2 } from "@opencode-ai/core/skill"
  19. import { testEffect } from "./lib/effect"
  20. const location = Location.Ref.make({ directory: AbsolutePath.make("/project") })
  21. const projects = Layer.mock(ProjectV2.Service, {
  22. resolve: (directory) => Effect.succeed({ id: ProjectV2.ID.global, directory }),
  23. })
  24. const skills = Layer.mock(SkillV2.Service, {
  25. list: () =>
  26. Effect.succeed([
  27. SkillV2.Info.make({
  28. id: SkillV2.ID.make("effect"),
  29. name: SkillV2.Name.make("Effect"),
  30. description: "Effect guidance",
  31. location: AbsolutePath.make(path.resolve("/skills/effect/SKILL.md")),
  32. content: "Use Effect",
  33. }),
  34. ]),
  35. })
  36. const locations = Layer.effect(
  37. LocationServiceMap.Service,
  38. LayerMap.make(
  39. () =>
  40. // The skill endpoint only needs the location-scoped Skill service.
  41. // oxlint-disable-next-line typescript-eslint/no-unsafe-type-assertion
  42. skills as unknown as Layer.Layer<LocationServices>,
  43. ),
  44. )
  45. const it = testEffect(
  46. AppNodeBuilder.build(
  47. LayerNode.group([Database.node, EventV2.node, SessionProjector.node, SessionStore.node, SessionV2.node]),
  48. [
  49. [LocationServiceMap.node, locations],
  50. [ProjectV2.node, projects],
  51. [SessionExecution.node, SessionExecution.noopLayer],
  52. ],
  53. ),
  54. )
  55. describe("SessionV2.skill", () => {
  56. it.effect("projects the caller-supplied message ID", () =>
  57. Effect.gen(function* () {
  58. const sessions = yield* SessionV2.Service
  59. const session = yield* sessions.create({ location })
  60. const id = SessionMessage.ID.make("msg_caller_skill")
  61. yield* sessions.skill({ id, sessionID: session.id, skill: SkillV2.ID.make("effect"), resume: false })
  62. expect(yield* sessions.messages({ sessionID: session.id })).toContainEqual(
  63. expect.objectContaining({ id, type: "skill", skill: "effect", name: "Effect", text: "Use Effect" }),
  64. )
  65. }),
  66. )
  67. })