diff --git a/apps/gateway/src/agent/provider.service.ts b/apps/gateway/src/agent/provider.service.ts
index 9b84df3..8c73fe6 100644
--- a/apps/gateway/src/agent/provider.service.ts
+++ b/apps/gateway/src/agent/provider.service.ts
@@ -92,15 +92,16 @@ export class ProviderService implements OnModuleInit {
       .map((modelId: string) => modelId.trim())
       .filter(Boolean);
 
-    this.registerCustomProvider({
-      id: 'ollama',
-      name: 'Ollama',
+    this.registry.registerProvider('ollama', {
       baseUrl: `${ollamaUrl}/v1`,
       apiKey: 'ollama',
+      api: 'openai-completions' as never,
       models: modelIds.map((id) => ({
         id,
         name: id,
         reasoning: false,
+        input: ['text'] as ('text' | 'image')[],
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
         contextWindow: 8192,
         maxTokens: 4096,
       })),