| 1883 | } |
| 1884 | |
| 1885 | function oauthAgentOptions( |
| 1886 | getAccessToken: (options?: { readonly force?: boolean }) => Promise<string>, |
| 1887 | capabilities?: readonly string[] | undefined, |
| 1888 | ): Pick<TestAgentOptions, 'initialConfig' | 'providerManagerOverrides'> { |
| 1889 | return { |
| 1890 | initialConfig: { |
| 1891 | defaultModel: 'kimi-code', |
| 1892 | providers: { |
| 1893 | 'managed:kimi-code': { |
| 1894 | type: 'vertexai', |
| 1895 | baseUrl: 'https://api.example/v1', |
| 1896 | oauth: { storage: 'file', key: 'oauth/kimi-code' }, |
| 1897 | }, |
| 1898 | }, |
| 1899 | models: { |
| 1900 | 'kimi-code': { |
| 1901 | provider: 'managed:kimi-code', |
| 1902 | model: 'kimi-for-coding', |
| 1903 | maxContextSize: 1_000_000, |
| 1904 | capabilities: capabilities === undefined ? undefined : [...capabilities], |
| 1905 | }, |
| 1906 | }, |
| 1907 | }, |
| 1908 | providerManagerOverrides: { |
| 1909 | resolveOAuthTokenProvider: vi.fn(() => ({ getAccessToken })), |
| 1910 | }, |
| 1911 | }; |
| 1912 | } |
| 1913 | |
| 1914 | function textResult(text: string): Awaited<ReturnType<GenerateFn>> { |
| 1915 | return { |