diff --git a/packages/ai/test/fixtures/oauth-barrel-import.ts b/packages/ai/test/fixtures/oauth-barrel-import.ts index 758a904c0..074377e70 100644 --- a/packages/ai/test/fixtures/oauth-barrel-import.ts +++ b/packages/ai/test/fixtures/oauth-barrel-import.ts @@ -1,7 +1,25 @@ -import { AnthropicOAuthFlow, loginAnthropic, refreshAnthropicToken } from "@oh-my-pi/pi-ai/registry/oauth"; +import { + AnthropicOAuthFlow as RootAnthropicOAuthFlow, + loginAnthropic as rootLoginAnthropic, + refreshAnthropicToken as rootRefreshAnthropicToken, +} from "@oh-my-pi/pi-ai"; +import { + AnthropicOAuthFlow as OAuthAnthropicOAuthFlow, + loginAnthropic as oauthLoginAnthropic, + refreshAnthropicToken as oauthRefreshAnthropicToken, +} from "@oh-my-pi/pi-ai/registry/oauth"; import "@oh-my-pi/pi-ai/providers/anthropic"; import "@oh-my-pi/pi-ai/auth-storage"; -if (!AnthropicOAuthFlow || !loginAnthropic || !refreshAnthropicToken) { +const publicExports = [ + RootAnthropicOAuthFlow, + rootLoginAnthropic, + rootRefreshAnthropicToken, + OAuthAnthropicOAuthFlow, + oauthLoginAnthropic, + oauthRefreshAnthropicToken, +]; + +if (publicExports.some(value => !value)) { throw new Error("Anthropic OAuth exports are unavailable"); }