diff --git a/packages/provider-inference/package.json b/packages/provider-inference/package.json
index e49145fb8..5570e3254 100644
--- a/packages/provider-inference/package.json
+++ b/packages/provider-inference/package.json
@@ -53,6 +53,7 @@
     "zod": "catalog:"
   },
   "devDependencies": {
+    "@proj-airi/core-agent": "workspace:^",
     "@types/node": "catalog:",
     "@vitest/browser-playwright": "catalog:vitest",
     "@xsai/generate-speech": "catalog:",
diff --git a/packages/stage-ui/src/libs/providers/providers/azure-openai/index.test.ts b/packages/provider-inference/src/providers/cloud/azure-openai/index.test.ts
similarity index 90%
rename from packages/stage-ui/src/libs/providers/providers/azure-openai/index.test.ts
rename to packages/provider-inference/src/providers/cloud/azure-openai/index.test.ts
index 22ea2823e..2d8ea4e2d 100644
--- a/packages/stage-ui/src/libs/providers/providers/azure-openai/index.test.ts
+++ b/packages/provider-inference/src/providers/cloud/azure-openai/index.test.ts
@@ -1,9 +1,10 @@
 import type { JsonSchema } from 'xsschema'
 
 import { createSparkCommandTool } from '@proj-airi/core-agent/agents/spark-command'
-import { getDefinedProvider } from '@proj-airi/provider-inference'
 import { afterEach, describe, expect, it, vi } from 'vitest'
 
+import { providerAzureOpenAI } from './index'
+
 interface ChatRequestBody {
   tools: Array<{
     function: {
@@ -40,11 +41,7 @@ describe('providerAzureOpenAI tool schemas', () => {
     const fetchMock = vi.fn<typeof fetch>().mockResolvedValue(new Response('{}'))
     vi.stubGlobal('fetch', fetchMock)
 
-    const providerDefinition = getDefinedProvider('azure-openai')
-    if (!providerDefinition)
-      throw new Error('Azure OpenAI provider definition is not registered.')
-
-    const provider = await providerDefinition.createProvider({
+    const provider = await providerAzureOpenAI.createProvider({
       apiKey: 'test-key',
       baseUrl: 'https://example.openai.azure.com/openai/',
     })
diff --git a/packages/stage-ui/src/libs/providers/providers/openrouter-ai/index.test.ts b/packages/provider-inference/src/providers/cloud/openrouter-ai/index.test.ts
similarity index 83%
rename from packages/stage-ui/src/libs/providers/providers/openrouter-ai/index.test.ts
rename to packages/provider-inference/src/providers/cloud/openrouter-ai/index.test.ts
index 039c57da2..d8580aab9 100644
--- a/packages/stage-ui/src/libs/providers/providers/openrouter-ai/index.test.ts
+++ b/packages/provider-inference/src/providers/cloud/openrouter-ai/index.test.ts
@@ -1,11 +1,13 @@
-import type { ChatRequestOptions } from '@proj-airi/provider-inference'
 import type { ChatProviderWithExtraOptions } from '@xsai-ext/providers/utils'
 import type { JsonSchema } from 'xsschema'
 
+import type { ChatRequestOptions } from '../../../types'
+
 import { createSparkCommandTool } from '@proj-airi/core-agent/agents/spark-command'
-import { getDefinedProvider } from '@proj-airi/provider-inference'
 import { afterEach, describe, expect, it, vi } from 'vitest'
 
+import { providerOpenRouterAI } from './index'
+
 interface ChatRequestBody {
   tools: Array<{
     function: {
@@ -35,11 +37,7 @@ describe('providerOpenRouterAI tool schemas', () => {
   })
 
   it('maps AIRI reasoning modes to OpenRouter request fields', async () => {
-    const providerDefinition = getDefinedProvider('openrouter-ai')
-    if (!providerDefinition)
-      throw new Error('OpenRouter provider definition is not registered.')
-
-    const provider = await providerDefinition.createProvider({
+    const provider = await providerOpenRouterAI.createProvider({
       apiKey: 'test-key',
     }) as ChatProviderWithExtraOptions<string, ChatRequestOptions>
 
@@ -58,11 +56,7 @@ describe('providerOpenRouterAI tool schemas', () => {
     const fetchMock = vi.fn<typeof fetch>().mockResolvedValue(new Response('{}'))
     vi.stubGlobal('fetch', fetchMock)
 
-    const providerDefinition = getDefinedProvider('openrouter-ai')
-    if (!providerDefinition)
-      throw new Error('OpenRouter provider definition is not registered.')
-
-    const provider = await providerDefinition.createProvider({
+    const provider = await providerOpenRouterAI.createProvider({
       apiKey: 'test-key',
     })
     if (!('chat' in provider))
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index 1680826ee..653c12e6c 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -3917,6 +3917,9 @@ importers:
         specifier: 'catalog:'
         version: 4.4.3
     devDependencies:
+      '@proj-airi/core-agent':
+        specifier: workspace:^
+        version: link:../core-agent
       '@types/node':
         specifier: 'catalog:'
         version: 26.2.0