fix: switch error code from 404 to 401 when api key is invalid

Author: brandonkachen

cli/src/components/referral-banner.tsx

@@ -9,14 +9,17 @@ export const ReferralBanner = () => {
   const theme = useTheme()
   const setInputMode = useChatStore((state) => state.setInputMode)
 
-  const { data: userDetails, isLoading, isError } = useUserDetailsQuery({
+  const { data: userDetails, isLoading, isError, error } = useUserDetailsQuery({
     fields: ['referral_link'] as const,
     enabled: true,
   })
 
   const referralLink = userDetails?.referral_link ?? null
+  const isAuthError = error?.message?.includes('401')
+
   let text = ''
   if (isLoading) text = 'Loading your referral link...'
+  else if (isAuthError) text = 'Session expired. Please log in again to view your referral link.'
   else if (isError) text = 'Failed to load your referral link. Please try again later.'
   else if (!referralLink) text = 'Your referral link is not available yet'
   else text = `Share this link with friends:\n${referralLink}`

web/src/app/api/v1/me/__tests__/me.test.ts

@@ -105,7 +105,7 @@ describe('/api/v1/me route', () => {
       expect(body).toEqual({ id: 'user-123' })
     })
 
-    test('returns 404 when API key is invalid', async () => {
+    test('returns 401 when API key is invalid', async () => {
       const req = new NextRequest('http://localhost:3000/api/v1/me', {
         headers: { Authorization: 'Bearer invalid-key' },
       })
@@ -114,7 +114,7 @@ describe('/api/v1/me route', () => {
         ...agentRuntimeImpl,
         req,
       })
-      expect(response.status).toBe(404)
+      expect(response.status).toBe(401)
       const body = await response.json()
       expect(body).toEqual({ error: 'Invalid API key or user not found' })
     })

web/src/app/api/v1/me/_get.ts

@@ -113,7 +113,7 @@ export async function getMe(params: {
   if (!userInfo) {
     return NextResponse.json(
       { error: 'Invalid API key or user not found' },
-      { status: 404 },
+      { status: 401 },
     )
   }