provider-github-copilot.test.ts 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278
  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 advertised Copilot endpoint metadata before model ID fallbacks", () =>
  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("mai-code-1-flash-picker")),
  160. api: {
  161. id: ModelV2.ID.make("mai-code-1-flash-picker"),
  162. type: "aisdk",
  163. package: "test-provider",
  164. settings: { endpoint: "responses" },
  165. },
  166. }),
  167. sdk: fakeSelectorSdk(calls),
  168. options: { endpoint: "responses" },
  169. })
  170. yield* aisdk.runLanguage({
  171. model: ModelV2.Info.make({
  172. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5")),
  173. api: {
  174. id: ModelV2.ID.make("gpt-5"),
  175. type: "aisdk",
  176. package: "test-provider",
  177. settings: { endpoint: "chat" },
  178. },
  179. }),
  180. sdk: fakeSelectorSdk(calls),
  181. options: { endpoint: "chat" },
  182. })
  183. expect(calls).toEqual(["responses:mai-code-1-flash-picker", "chat:gpt-5"])
  184. }),
  185. )
  186. it.effect("uses the API model ID when selecting responses or chat", () =>
  187. Effect.gen(function* () {
  188. const plugin = yield* PluginV2.Service
  189. const aisdk = yield* AISDK.Service
  190. const calls: string[] = []
  191. yield* addPlugin()
  192. yield* aisdk.runLanguage({
  193. model: ModelV2.Info.make({
  194. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("default")),
  195. api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" },
  196. }),
  197. sdk: fakeSelectorSdk(calls),
  198. options: {},
  199. })
  200. yield* aisdk.runLanguage({
  201. model: ModelV2.Info.make({
  202. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("small")),
  203. api: { id: ModelV2.ID.make("gpt-5-mini"), type: "aisdk", package: "test-provider" },
  204. }),
  205. sdk: fakeSelectorSdk(calls),
  206. options: {},
  207. })
  208. yield* aisdk.runLanguage({
  209. model: ModelV2.Info.make({
  210. ...ModelV2.Info.empty(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("sonnet")),
  211. api: { id: ModelV2.ID.make("claude-sonnet-4"), type: "aisdk", package: "test-provider" },
  212. }),
  213. sdk: fakeSelectorSdk(calls),
  214. options: {},
  215. })
  216. expect(calls).toEqual(["responses:gpt-5", "chat:gpt-5-mini", "chat:claude-sonnet-4"])
  217. }),
  218. )
  219. it.effect("disables gpt-5-chat-latest before Copilot language selection", () =>
  220. Effect.gen(function* () {
  221. const catalog = yield* Catalog.Service
  222. yield* catalog.transform((catalog) => {
  223. catalog.provider.update(ProviderV2.ID.make("github-copilot"), () => {})
  224. catalog.model.update(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5-chat-latest"), () => {})
  225. })
  226. yield* addPlugin()
  227. expect(
  228. required(yield* catalog.model.get(ProviderV2.ID.make("github-copilot"), ModelV2.ID.make("gpt-5-chat-latest")))
  229. .enabled,
  230. ).toBe(false)
  231. }),
  232. )
  233. it.effect("does not disable gpt-5-chat-latest for non-Copilot providers", () =>
  234. Effect.gen(function* () {
  235. const catalog = yield* Catalog.Service
  236. yield* catalog.transform((catalog) => {
  237. catalog.provider.update(ProviderV2.ID.make("custom-copilot"), () => {})
  238. catalog.model.update(ProviderV2.ID.make("custom-copilot"), ModelV2.ID.make("gpt-5-chat-latest"), () => {})
  239. })
  240. yield* addPlugin()
  241. expect(
  242. required(yield* catalog.model.get(ProviderV2.ID.make("custom-copilot"), ModelV2.ID.make("gpt-5-chat-latest")))
  243. .enabled,
  244. ).toBe(true)
  245. }),
  246. )
  247. it.effect("ignores non-Copilot providers", () =>
  248. Effect.gen(function* () {
  249. const plugin = yield* PluginV2.Service
  250. const aisdk = yield* AISDK.Service
  251. const calls: string[] = []
  252. yield* addPlugin()
  253. const result = yield* aisdk.runLanguage({
  254. model: ModelV2.Info.make({
  255. ...ModelV2.Info.empty(ProviderV2.ID.make("openai"), ModelV2.ID.make("gpt-5")),
  256. api: { id: ModelV2.ID.make("gpt-5"), type: "aisdk", package: "test-provider" },
  257. }),
  258. sdk: fakeSelectorSdk(calls),
  259. options: {},
  260. })
  261. expect(calls).toEqual([])
  262. expect(result.language).toBeUndefined()
  263. }),
  264. )
  265. })