diff --git a/src/services/aiService.ts b/src/services/aiService.ts
index 08baf89..eabb3b8 100644
--- a/src/services/aiService.ts
+++ b/src/services/aiService.ts
@@ -4,7 +4,7 @@
  * Validates: Requirements 12.1-12.6
  */
 
-import api, { API_BASE_URL } from './api';
+import api, { API_BASE_URL, getAccessToken } from './api';
 import type {
   ApiResponse,
   AIChatResponse,
@@ -105,7 +105,7 @@ export async function streamChatMessage(
   onChunk: (text: string) => void
 ): Promise<AIChatResponse> {
   const sessionId = existingSessionId || getSessionId();
-  const token = localStorage.getItem('token'); // Assuming auth token is stored here
+  const token = getAccessToken(); // Get token from shared storage logic
 
   // Use fetch directly for streaming support
   const response = await fetch(`${API_BASE_URL}/ai/chat/stream`, {
diff --git a/src/services/api.ts b/src/services/api.ts
index 06e7926..5eb1935 100644
--- a/src/services/api.ts
+++ b/src/services/api.ts
@@ -25,7 +25,7 @@ interface RequestOptions extends RequestInit {
 /**
  * Get access token from storage
  */
-function getAccessToken(): string | null {
+export function getAccessToken(): string | null {
   return localStorage.getItem(ACCESS_TOKEN_KEY);
 }