diff --git a/app/api/ai/conversations/[id]/route.ts b/app/api/ai/conversations/[id]/route.ts
index 1f5b673..a644e40 100644
--- a/app/api/ai/conversations/[id]/route.ts
+++ b/app/api/ai/conversations/[id]/route.ts
@@ -1,5 +1,5 @@
 import { NextRequest, NextResponse } from 'next/server'
-import { AIClient } from '@/lib/ai-assistant-openai'
+import { AIClient } from '@/lib/ai-assistant'
 import { getUser } from '@/lib/auth'
 import { ComponentService } from '@/lib/services/component-service'
 import { Component } from '@prisma/client'
diff --git a/app/api/ai/conversations/route.ts b/app/api/ai/conversations/route.ts
index c70fb98..9580fe7 100644
--- a/app/api/ai/conversations/route.ts
+++ b/app/api/ai/conversations/route.ts
@@ -1,5 +1,5 @@
 import { NextRequest, NextResponse } from 'next/server'
-import { AIClient } from '@/lib/ai-assistant-openai'
+import { AIClient } from '@/lib/ai-assistant'
 import { getUser } from '@/lib/auth'
 
 export async function GET(request: NextRequest) {