Reset codebuff client on login

Author: jahooma

cli/src/hooks/use-auth-query.ts

@@ -8,6 +8,7 @@ import {
   logoutUser as logoutUserUtil,
   type User,
 } from '../utils/auth'
+import { resetCodebuffClient } from '../utils/codebuff-client'
 import { logger as defaultLogger } from '../utils/logger'
 
 import type { GetUserInfoFromApiKeyFn } from '@codebuff/common/types/contracts/database'
@@ -159,6 +160,8 @@ export function useLogoutMutation(deps: UseLogoutMutationDeps = {}) {
   return useMutation({
     mutationFn: logoutUser,
     onSuccess: () => {
+      // Reset the SDK client after logout
+      resetCodebuffClient()
       // Clear all auth-related cache
       queryClient.removeQueries({ queryKey: authQueryKeys.all })
     },

cli/src/hooks/use-auth-state.ts

@@ -2,7 +2,7 @@ import { useCallback, useEffect, useRef, useState } from 'react'
 
 import { useAuthQuery, useLogoutMutation } from './use-auth-query'
 import { getUserCredentials } from '../utils/auth'
-import { logger } from '../utils/logger'
+import { resetCodebuffClient } from '../utils/codebuff-client'
 
 import type { MultilineInputHandle } from '../components/multiline-input'
 import type { User } from '../utils/auth'
@@ -63,6 +63,8 @@ export const useAuthState = ({
   // Handle successful login
   const handleLoginSuccess = useCallback(
     (loggedInUser: User) => {
+      // Reset the SDK client to pick up new credentials
+      resetCodebuffClient()
       resetChatStore()
       setInputFocused(true)
       setUser(loggedInUser)

cli/src/utils/codebuff-client.ts

@@ -8,6 +8,14 @@ import { logger } from './logger'
 
 let clientInstance: CodebuffClient | null = null
 
+/**
+ * Reset the cached CodebuffClient instance.
+ * This should be called after login to ensure the client is re-initialized with new credentials.
+ */
+export function resetCodebuffClient(): void {
+  clientInstance = null
+}
+
 export function getCodebuffClient(): CodebuffClient | null {
   if (!clientInstance) {
     const { token: apiKey, source } = getAuthTokenDetails()