fix: invalidate models

stacklok · Feb 14, 2025 · c6bcee0 · c6bcee0
1 parent 375078f
commit c6bcee0
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/features/providers/hooks/use-invalidate-providers-queries.ts b/src/features/providers/hooks/use-invalidate-providers-queries.ts
@@ -1,4 +1,7 @@
-import { v1ListProviderEndpointsQueryKey } from '@/api/generated/@tanstack/react-query.gen'
+import {
+  v1ListAllModelsForAllProvidersQueryKey,
+  v1ListProviderEndpointsQueryKey,
+} from '@/api/generated/@tanstack/react-query.gen'
 import { useQueryClient } from '@tanstack/react-query'
 import { useCallback } from 'react'
 import { invalidateQueries } from '../../../lib/react-query-utils'
@@ -7,7 +10,10 @@ export function useInvalidateProvidersQueries() {
   const queryClient = useQueryClient()
 
   const invalidate = useCallback(async () => {
-    invalidateQueries(queryClient, [v1ListProviderEndpointsQueryKey])
+    invalidateQueries(queryClient, [
+      v1ListProviderEndpointsQueryKey,
+      v1ListAllModelsForAllProvidersQueryKey,
+    ])
   }, [queryClient])
 
   return invalidate