feat: add workspaces impacted by provider deletion (#340)

* chore: update openapi

* feat: add workspaces by provider

* fix: query refetch

Author: peppescg

src/api/generated/@tanstack/react-query.gen.ts

@@ -28,6 +28,7 @@ import {
   v1DeleteWorkspaceCustomInstructions,
   v1GetWorkspaceMuxes,
   v1SetWorkspaceMuxes,
+  v1ListWorkspacesByProvider,
   v1StreamSse,
   v1VersionCheck,
   v1GetWorkspaceTokenUsage,
@@ -76,6 +77,7 @@ import type {
   V1SetWorkspaceMuxesData,
   V1SetWorkspaceMuxesError,
   V1SetWorkspaceMuxesResponse,
+  V1ListWorkspacesByProviderData,
   V1GetWorkspaceTokenUsageData,
 } from '../types.gen'
 
@@ -687,6 +689,27 @@ export const v1SetWorkspaceMuxesMutation = (
   return mutationOptions
 }
 
+export const v1ListWorkspacesByProviderQueryKey = (
+  options: OptionsLegacyParser<V1ListWorkspacesByProviderData>
+) => [createQueryKey('v1ListWorkspacesByProvider', options)]
+
+export const v1ListWorkspacesByProviderOptions = (
+  options: OptionsLegacyParser<V1ListWorkspacesByProviderData>
+) => {
+  return queryOptions({
+    queryFn: async ({ queryKey, signal }) => {
+      const { data } = await v1ListWorkspacesByProvider({
+        ...options,
+        ...queryKey[0],
+        signal,
+        throwOnError: true,
+      })
+      return data
+    },
+    queryKey: v1ListWorkspacesByProviderQueryKey(options),
+  })
+}
+
 export const v1StreamSseQueryKey = (options?: OptionsLegacyParser) => [
   createQueryKey('v1StreamSse', options),
 ]

src/api/generated/sdk.gen.ts

@@ -73,6 +73,9 @@ import type {
   V1SetWorkspaceMuxesData,
   V1SetWorkspaceMuxesError,
   V1SetWorkspaceMuxesResponse,
+  V1ListWorkspacesByProviderData,
+  V1ListWorkspacesByProviderError,
+  V1ListWorkspacesByProviderResponse,
   V1StreamSseError,
   V1StreamSseResponse,
   V1VersionCheckError,
@@ -512,6 +515,25 @@ export const v1SetWorkspaceMuxes = <ThrowOnError extends boolean = false>(
   })
 }
 
+/**
+ * List Workspaces By Provider
+ * List workspaces by provider ID.
+ */
+export const v1ListWorkspacesByProvider = <
+  ThrowOnError extends boolean = false,
+>(
+  options: OptionsLegacyParser<V1ListWorkspacesByProviderData, ThrowOnError>
+) => {
+  return (options?.client ?? client).get<
+    V1ListWorkspacesByProviderResponse,
+    V1ListWorkspacesByProviderError,
+    ThrowOnError
+  >({
+    ...options,
+    url: '/api/v1/workspaces/{provider_id}',
+  })
+}
+
 /**
  * Stream Sse
  * Send alerts event

src/api/generated/types.gen.ts

@@ -35,11 +35,11 @@ export type Alert = {
     | {
         [key: string]: unknown
       }
-    | null;
-  trigger_type: string;
-  trigger_category: AlertSeverity;
-  timestamp: string;
-};
+    | null
+  trigger_type: string
+  trigger_category: AlertSeverity
+  timestamp: string
+}
 
 /**
  * Represents an alert with it's respective conversation.
@@ -60,8 +60,8 @@ export type AlertConversation = {
 }
 
 export enum AlertSeverity {
-  INFO = "info",
-  CRITICAL = "critical",
+  INFO = 'info',
+  CRITICAL = 'critical',
 }
 
 /**
@@ -111,6 +111,7 @@ export type Conversation = {
 
 export type CreateOrRenameWorkspaceRequest = {
   name: string
+  config?: WorkspaceConfig | null
   rename_to?: string | null
 }
 
@@ -145,15 +146,16 @@ export type ModelByProvider = {
  * Represents the different types of matchers we support.
  */
 export enum MuxMatcherType {
-  CATCH_ALL = "catch_all",
-  FILENAME_MATCH = "filename_match",
-  REQUEST_TYPE_MATCH = "request_type_match",
+  CATCH_ALL = 'catch_all',
+  FILENAME_MATCH = 'filename_match',
+  REQUEST_TYPE_MATCH = 'request_type_match',
 }
 
 /**
  * Represents a mux rule for a provider.
  */
 export type MuxRule = {
+  provider_name?: string | null
   provider_id: string
   model: string
   matcher_type: MuxMatcherType
@@ -251,6 +253,20 @@ export type Workspace = {
   is_active: boolean
 }
 
+export type WorkspaceConfig = {
+  system_prompt: string
+  muxing_rules: Array<MuxRule>
+}
+
+/**
+ * Returns a workspace ID with model name
+ */
+export type WorkspaceWithModel = {
+  id: string
+  name: string
+  provider_model_name: string
+}
+
 export type HealthCheckHealthGetResponse = unknown
 
 export type HealthCheckHealthGetError = unknown
@@ -462,6 +478,16 @@ export type V1SetWorkspaceMuxesResponse = void
 
 export type V1SetWorkspaceMuxesError = HTTPValidationError
 
+export type V1ListWorkspacesByProviderData = {
+  path: {
+    provider_id: string
+  }
+}
+
+export type V1ListWorkspacesByProviderResponse = Array<WorkspaceWithModel>
+
+export type V1ListWorkspacesByProviderError = HTTPValidationError
+
 export type V1StreamSseResponse = unknown
 
 export type V1StreamSseError = unknown

src/api/openapi.json

@@ -989,6 +989,55 @@
         }
       }
     },
+    "/api/v1/workspaces/{provider_id}": {
+      "get": {
+        "tags": [
+          "CodeGate API",
+          "Workspaces"
+        ],
+        "summary": "List Workspaces By Provider",
+        "description": "List workspaces by provider ID.",
+        "operationId": "v1_list_workspaces_by_provider",
+        "parameters": [
+          {
+            "name": "provider_id",
+            "in": "path",
+            "required": true,
+            "schema": {
+              "type": "string",
+              "format": "uuid",
+              "title": "Provider Id"
+            }
+          }
+        ],
+        "responses": {
+          "200": {
+            "description": "Successful Response",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "type": "array",
+                  "items": {
+                    "$ref": "#/components/schemas/WorkspaceWithModel"
+                  },
+                  "title": "Response V1 List Workspaces By Provider"
+                }
+              }
+            }
+          },
+          "422": {
+            "description": "Validation Error",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/HTTPValidationError"
+                }
+              }
+            }
+          }
+        }
+      }
+    },
     "/api/v1/alerts_notification": {
       "get": {
         "tags": [
@@ -1478,6 +1527,16 @@
             "type": "string",
             "title": "Name"
           },
+          "config": {
+            "anyOf": [
+              {
+                "$ref": "#/components/schemas/WorkspaceConfig"
+              },
+              {
+                "type": "null"
+              }
+            ]
+          },
           "rename_to": {
             "anyOf": [
               {
@@ -1590,6 +1649,17 @@
       },
       "MuxRule": {
         "properties": {
+          "provider_name": {
+            "anyOf": [
+              {
+                "type": "string"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "title": "Provider Name"
+          },
           "provider_id": {
             "type": "string",
             "title": "Provider Id"
@@ -1842,6 +1912,52 @@
           "is_active"
         ],
         "title": "Workspace"
+      },
+      "WorkspaceConfig": {
+        "properties": {
+          "system_prompt": {
+            "type": "string",
+            "title": "System Prompt"
+          },
+          "muxing_rules": {
+            "items": {
+              "$ref": "#/components/schemas/MuxRule"
+            },
+            "type": "array",
+            "title": "Muxing Rules"
+          }
+        },
+        "type": "object",
+        "required": [
+          "system_prompt",
+          "muxing_rules"
+        ],
+        "title": "WorkspaceConfig"
+      },
+      "WorkspaceWithModel": {
+        "properties": {
+          "id": {
+            "type": "string",
+            "title": "Id"
+          },
+          "name": {
+            "type": "string",
+            "pattern": "^[a-zA-Z0-9_-]+$",
+            "title": "Name"
+          },
+          "provider_model_name": {
+            "type": "string",
+            "title": "Provider Model Name"
+          }
+        },
+        "type": "object",
+        "required": [
+          "id",
+          "name",
+          "provider_model_name"
+        ],
+        "title": "WorkspaceWithModel",
+        "description": "Returns a workspace ID with model name"
       }
     }
   }

src/features/providers/components/table-providers.tsx

@@ -51,12 +51,12 @@ const COLUMNS: Column[] = [
 function CellRenderer({
   column,
   row,
-  deleteProvider,
 }: {
   column: Column
   row: ProviderEndpoint
-  deleteProvider: () => void
 }) {
+  const deleteProvider = useConfirmDeleteProvider(row.id)
+
   return match(column.id)
     .with(COLUMN_MAP.provider, () => (
       <>
@@ -90,7 +90,15 @@ function CellRenderer({
       </div>
     ))
     .with(COLUMN_MAP.configuration, () => (
-      <Button isIcon variant="tertiary" onPress={deleteProvider}>
+      <Button
+        isIcon
+        variant="tertiary"
+        onPress={() =>
+          deleteProvider({
+            path: { provider_id: row.id as string },
+          })
+        }
+      >
         <Trash01 />
       </Button>
     ))
@@ -99,7 +107,6 @@ function CellRenderer({
 
 export function TableProviders() {
   const { data: providers = [] } = useProviders()
-  const deleteProvider = useConfirmDeleteProvider()
 
   return (
     <ResizableTableContainer>
@@ -117,15 +124,7 @@ export function TableProviders() {
                   id={column.id}
                   alignment={column.alignment}
                 >
-                  <CellRenderer
-                    column={column}
-                    row={row}
-                    deleteProvider={() => {
-                      deleteProvider({
-                        path: { provider_id: row.id as string },
-                      })
-                    }}
-                  />
+                  <CellRenderer column={column} row={row} />
                 </Cell>
               )}
             </Row>