provider-github-copilot.test.ts 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242
  1. import { AISDK } from "@opencode-ai/core/aisdk"
  2. import { describe, expect } from "bun:test"
  3. import { Effect } from "effect"
  4. import { Catalog } from "@opencode-ai/core/catalog"
  5. import { ModelV2 } from "@opencode-ai/core/model"
  6. import { PluginV2 } from "@opencode-ai/core/plugin"
  7. import { PluginHost } from "@opencode-ai/core/plugin/host"
  8. import { GithubCopilotPlugin } from "@opencode-ai/core/plugin/provider/github-copilot"
  9. import { ProviderV2 } from "@opencode-ai/core/provider"
  10. import type { LanguageModelV3 } from "@ai-sdk/provider"
  11. import { testEffect } from "../lib/effect"
  12. import { PluginTestLayer } from "./fixture"
  13. const it = testEffect(PluginTestLayer)
  14. const addPlugin = Effect.fn(function* () {
  15. const plugin = yield* PluginV2.Service
  16. const aisdk = yield* AISDK.Service
  17. const host = yield* PluginHost.make(plugin)
  18. yield* GithubCopilotPlugin.effect(host)
  19. })
  20. function required<T>(value: T | undefined): T {
  21. if (value === undefined) throw new Error("Expected value")
  22. return value
  23. }
  24. function fakeSelectorSdk(calls: string[]) {
  25. const make = (method: string) => (id: string) => {
  26. calls.push(`${method}:${id}`)
  27. return { modelId: id, provider: method, specificationVersion: "v3" } as unknown as LanguageModelV3
  28. }
  29. return {
  30. responses: make("responses"),
  31. messages: make("messages"),
  32. chat: make("chat"),
  33. languageModel: make("languageModel"),
  34. }
  35. }
  36. describe("GithubCopilotPlugin", () => {
  37. it.effect("creates the bundled Copilot SDK for the GitHub Copilot package", () =>
  38. Effect.gen(function* () {
  39. const plugin = yield* PluginV2.Service
  40. const aisdk = yield* AISDK.Service
  41. yield* addPlugin()
  42. const ignored = yield* aisdk.runSDK({
  43. model: ModelV2.Info.make({
  44. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5")),
  45. api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" },
  46. }),
  47. package: "@ai-sdk/openai-compatible",
  48. options: { name: "github-copilot" },
  49. })
  50. const result = yield* aisdk.runSDK({
  51. model: ModelV2.Info.make({
  52. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5")),
  53. api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" },
  54. }),
  55. package: "@ai-sdk/github-copilot",
  56. options: { name: "github-copilot" },
  57. })
  58. expect(ignored.sdk).toBeUndefined()
  59. expect(result.sdk).toBeDefined()
  60. }),
  61. )
  62. it.effect("selects languageModel when responses and chat are absent", () =>
  63. Effect.gen(function* () {
  64. const plugin = yield* PluginV2.Service
  65. const aisdk = yield* AISDK.Service
  66. const calls: string[] = []
  67. yield* addPlugin()
  68. yield* aisdk.runLanguage({
  69. model: ModelV2.Info.make({
  70. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("claude-sonnet-4")),
  71. api: { id: ModelV2.ID.make("claude-sonnet-4"), type: "aisdk", package: "test-provider" },
  72. }),
  73. sdk: { languageModel: fakeSelectorSdk(calls).languageModel },
  74. options: {},
  75. })
  76. expect(calls).toEqual(["languageModel:claude-sonnet-4"])
  77. }),
  78. )
  79. it.effect("selects languageModel with the API model ID when responses and chat are absent", () =>
  80. Effect.gen(function* () {
  81. const plugin = yield* PluginV2.Service
  82. const aisdk = yield* AISDK.Service
  83. const calls: string[] = []
  84. yield* addPlugin()
  85. yield* aisdk.runLanguage({
  86. model: ModelV2.Info.make({
  87. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("alias")),
  88. api: { id: ModelV2.ID.make("claude-sonnet-4"), type: "aisdk", package: "test-provider" },
  89. }),
  90. sdk: { languageModel: fakeSelectorSdk(calls).languageModel },
  91. options: {},
  92. })
  93. expect(calls).toEqual(["languageModel:claude-sonnet-4"])
  94. }),
  95. )
  96. it.effect("uses responses for gpt-5 models except gpt-5-mini", () =>
  97. Effect.gen(function* () {
  98. const plugin = yield* PluginV2.Service
  99. const aisdk = yield* AISDK.Service
  100. const calls: string[] = []
  101. yield* addPlugin()
  102. yield* aisdk.runLanguage({
  103. model: ModelV2.Info.make({
  104. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5")),
  105. api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" },
  106. }),
  107. sdk: fakeSelectorSdk(calls),
  108. options: {},
  109. })
  110. yield* aisdk.runLanguage({
  111. model: ModelV2.Info.make({
  112. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5.1-codex")),
  113. api: { id: ModelV2.ID.make("gpt-5.1-codex"), type: "aisdk", package: "test-provider" },
  114. }),
  115. sdk: fakeSelectorSdk(calls),
  116. options: {},
  117. })
  118. yield* aisdk.runLanguage({
  119. model: ModelV2.Info.make({
  120. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-4o")),
  121. api: { id: ModelV2.ID.make("gpt-4o"), type: "aisdk", package: "test-provider" },
  122. }),
  123. sdk: fakeSelectorSdk(calls),
  124. options: {},
  125. })
  126. yield* aisdk.runLanguage({
  127. model: ModelV2.Info.make({
  128. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5-mini")),
  129. api: { id: ModelV2.ID.make("gpt-5-mini"), type: "aisdk", package: "test-provider" },
  130. }),
  131. sdk: fakeSelectorSdk(calls),
  132. options: {},
  133. })
  134. yield* aisdk.runLanguage({
  135. model: ModelV2.Info.make({
  136. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5-mini-2025-08-07")),
  137. api: { id: ModelV2.ID.make("gpt-5-mini-2025-08-07"), type: "aisdk", package: "test-provider" },
  138. }),
  139. sdk: fakeSelectorSdk(calls),
  140. options: {},
  141. })
  142. expect(calls).toEqual([
  143. "responses:gpt-5",
  144. "responses:gpt-5.1-codex",
  145. "chat:gpt-4o",
  146. "chat:gpt-5-mini",
  147. "chat:gpt-5-mini-2025-08-07",
  148. ])
  149. }),
  150. )
  151. it.effect("uses the API model ID when selecting responses or chat", () =>
  152. Effect.gen(function* () {
  153. const plugin = yield* PluginV2.Service
  154. const aisdk = yield* AISDK.Service
  155. const calls: string[] = []
  156. yield* addPlugin()
  157. yield* aisdk.runLanguage({
  158. model: ModelV2.Info.make({
  159. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("default")),
  160. api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" },
  161. }),
  162. sdk: fakeSelectorSdk(calls),
  163. options: {},
  164. })
  165. yield* aisdk.runLanguage({
  166. model: ModelV2.Info.make({
  167. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("small")),
  168. api: { id: ModelV2.ID.make("gpt-5-mini"), type: "aisdk", package: "test-provider" },
  169. }),
  170. sdk: fakeSelectorSdk(calls),
  171. options: {},
  172. })
  173. yield* aisdk.runLanguage({
  174. model: ModelV2.Info.make({
  175. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("sonnet")),
  176. api: { id: ModelV2.ID.make("claude-sonnet-4"), type: "aisdk", package: "test-provider" },
  177. }),
  178. sdk: fakeSelectorSdk(calls),
  179. options: {},
  180. })
  181. expect(calls).toEqual(["responses:gpt-5", "chat:gpt-5-mini", "chat:claude-sonnet-4"])
  182. }),
  183. )
  184. it.effect("disables gpt-5-chat-latest before Copilot language selection", () =>
  185. Effect.gen(function* () {
  186. const catalog = yield* Catalog.Service
  187. yield* catalog.transform((catalog) => {
  188. catalog.provider.update(ProviderV2.ID.make("github-copilot"), () => {})
  189. catalog.model.update(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5-chat-latest"), () => {})
  190. })
  191. yield* addPlugin()
  192. expect(
  193. required(yield* catalog.model.get(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5-chat-latest")))
  194. .enabled,
  195. ).toBe(false)
  196. }),
  197. )
  198. it.effect("does not disable gpt-5-chat-latest for non-Copilot providers", () =>
  199. Effect.gen(function* () {
  200. const catalog = yield* Catalog.Service
  201. yield* catalog.transform((catalog) => {
  202. catalog.provider.update(ProviderV2.ID.make("custom-copilot"), () => {})
  203. catalog.model.update(ProviderV2.ID.make("custom-copilot"), ModelV2.ID.make("gpt-5-chat-latest"), () => {})
  204. })
  205. yield* addPlugin()
  206. expect(
  207. required(yield* catalog.model.get(ProviderV2.ID.make("custom-copilot"), ModelV2.ID.make("gpt-5-chat-latest")))
  208. .enabled,
  209. ).toBe(true)
  210. }),
  211. )
  212. it.effect("ignores non-Copilot providers", () =>
  213. Effect.gen(function* () {
  214. const plugin = yield* PluginV2.Service
  215. const aisdk = yield* AISDK.Service
  216. const calls: string[] = []
  217. yield* addPlugin()
  218. const result = yield* aisdk.runLanguage({
  219. model: ModelV2.Info.make({
  220. ...ModelV2.Info.empty(ProviderV2.ID.make("openai"), ModelV2.ID.make("gpt-5")),
  221. api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" },
  222. }),
  223. sdk: fakeSelectorSdk(calls),
  224. options: {},
  225. })
  226. expect(calls).toEqual([])
  227. expect(result.language).toBeUndefined()
  228. }),
  229. )
  230. })