chore: update api

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

@@ -12,6 +12,7 @@ import {
   v1GetProviderEndpoint,
   v1UpdateProviderEndpoint,
   v1DeleteProviderEndpoint,
+  v1ConfigureAuthMaterial,
   v1ListWorkspaces,
   v1CreateWorkspace,
   v1ListActiveWorkspaces,
@@ -44,6 +45,9 @@ import type {
   V1DeleteProviderEndpointData,
   V1DeleteProviderEndpointError,
   V1DeleteProviderEndpointResponse,
+  V1ConfigureAuthMaterialData,
+  V1ConfigureAuthMaterialError,
+  V1ConfigureAuthMaterialResponse,
   V1CreateWorkspaceData,
   V1CreateWorkspaceError,
   V1CreateWorkspaceResponse,
@@ -293,6 +297,26 @@ export const v1DeleteProviderEndpointMutation = (
   return mutationOptions;
 };
 
+export const v1ConfigureAuthMaterialMutation = (
+  options?: Partial<OptionsLegacyParser<V1ConfigureAuthMaterialData>>,
+) => {
+  const mutationOptions: UseMutationOptions<
+    V1ConfigureAuthMaterialResponse,
+    V1ConfigureAuthMaterialError,
+    OptionsLegacyParser<V1ConfigureAuthMaterialData>
+  > = {
+    mutationFn: async (localOptions) => {
+      const { data } = await v1ConfigureAuthMaterial({
+        ...options,
+        ...localOptions,
+        throwOnError: true,
+      });
+      return data;
+    },
+  };
+  return mutationOptions;
+};
+
 export const v1ListWorkspacesQueryKey = (options?: OptionsLegacyParser) => [
   createQueryKey("v1ListWorkspaces", options),
 ];

src/api/generated/sdk.gen.ts

@@ -28,6 +28,9 @@ import type {
   V1DeleteProviderEndpointData,
   V1DeleteProviderEndpointError,
   V1DeleteProviderEndpointResponse,
+  V1ConfigureAuthMaterialData,
+  V1ConfigureAuthMaterialError,
+  V1ConfigureAuthMaterialResponse,
   V1ListWorkspacesError,
   V1ListWorkspacesResponse,
   V1CreateWorkspaceData,
@@ -218,6 +221,23 @@ export const v1DeleteProviderEndpoint = <ThrowOnError extends boolean = false>(
   });
 };
 
+/**
+ * Configure Auth Material
+ * Configure auth material for a provider.
+ */
+export const v1ConfigureAuthMaterial = <ThrowOnError extends boolean = false>(
+  options: OptionsLegacyParser<V1ConfigureAuthMaterialData, ThrowOnError>,
+) => {
+  return (options?.client ?? client).put<
+    V1ConfigureAuthMaterialResponse,
+    V1ConfigureAuthMaterialError,
+    ThrowOnError
+  >({
+    ...options,
+    url: "/api/v1/provider-endpoints/{provider_id}/auth-material",
+  });
+};
+
 /**
  * List Workspaces
  * List all workspaces.

src/api/generated/types.gen.ts

@@ -44,6 +44,14 @@ export type CodeSnippet = {
   libraries?: Array<string>;
 };
 
+/**
+ * Represents a request to configure auth material for a provider.
+ */
+export type ConfigureAuthMaterial = {
+  auth_type: ProviderAuthType;
+  api_key?: string | null;
+};
+
 /**
  * Represents a conversation.
  */
@@ -100,10 +108,10 @@ export enum MuxMatcherType {
  * Represents a mux rule for a provider.
  */
 export type MuxRule = {
-  provider: string;
+  provider_id: string;
   model: string;
   matcher_type: MuxMatcherType;
-  matcher: string | null;
+  matcher?: string | null;
 };
 
 /**
@@ -126,7 +134,7 @@ export type ProviderEndpoint = {
   description?: string;
   provider_type: ProviderType;
   endpoint: string;
-  auth_type: ProviderAuthType;
+  auth_type?: ProviderAuthType | null;
 };
 
 /**
@@ -263,6 +271,17 @@ export type V1DeleteProviderEndpointResponse = unknown;
 
 export type V1DeleteProviderEndpointError = HTTPValidationError;
 
+export type V1ConfigureAuthMaterialData = {
+  body: ConfigureAuthMaterial;
+  path: {
+    provider_id: string;
+  };
+};
+
+export type V1ConfigureAuthMaterialResponse = void;
+
+export type V1ConfigureAuthMaterialError = HTTPValidationError;
+
 export type V1ListWorkspacesResponse = ListWorkspacesResponse;
 
 export type V1ListWorkspacesError = unknown;

src/api/openapi.json

@@ -336,6 +336,54 @@
         }
       }
     },
+    "/api/v1/provider-endpoints/{provider_id}/auth-material": {
+      "put": {
+        "tags": [
+          "CodeGate API",
+          "Providers"
+        ],
+        "summary": "Configure Auth Material",
+        "description": "Configure auth material for a provider.",
+        "operationId": "v1_configure_auth_material",
+        "parameters": [
+          {
+            "name": "provider_id",
+            "in": "path",
+            "required": true,
+            "schema": {
+              "type": "string",
+              "format": "uuid",
+              "title": "Provider Id"
+            }
+          }
+        ],
+        "requestBody": {
+          "required": true,
+          "content": {
+            "application/json": {
+              "schema": {
+                "$ref": "#/components/schemas/ConfigureAuthMaterial"
+              }
+            }
+          }
+        },
+        "responses": {
+          "204": {
+            "description": "Successful Response"
+          },
+          "422": {
+            "description": "Validation Error",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/HTTPValidationError"
+                }
+              }
+            }
+          }
+        }
+      }
+    },
     "/api/v1/workspaces": {
       "get": {
         "tags": [
@@ -1201,6 +1249,30 @@
         ],
         "title": "CodeSnippet"
       },
+      "ConfigureAuthMaterial": {
+        "properties": {
+          "auth_type": {
+            "$ref": "#/components/schemas/ProviderAuthType"
+          },
+          "api_key": {
+            "anyOf": [
+              {
+                "type": "string"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "title": "Api Key"
+          }
+        },
+        "type": "object",
+        "required": [
+          "auth_type"
+        ],
+        "title": "ConfigureAuthMaterial",
+        "description": "Represents a request to configure auth material for a provider."
+      },
       "Conversation": {
         "properties": {
           "question_answers": {
@@ -1373,9 +1445,9 @@
       },
       "MuxRule": {
         "properties": {
-          "provider": {
+          "provider_id": {
             "type": "string",
-            "title": "Provider"
+            "title": "Provider Id"
           },
           "model": {
             "type": "string",
@@ -1398,10 +1470,9 @@
         },
         "type": "object",
         "required": [
-          "provider",
+          "provider_id",
           "model",
-          "matcher_type",
-          "matcher"
+          "matcher_type"
         ],
         "title": "MuxRule",
         "description": "Represents a mux rule for a provider."
@@ -1447,15 +1518,22 @@
             "title": "Endpoint"
           },
           "auth_type": {
-            "$ref": "#/components/schemas/ProviderAuthType"
+            "anyOf": [
+              {
+                "$ref": "#/components/schemas/ProviderAuthType"
+              },
+              {
+                "type": "null"
+              }
+            ],
+            "default": "none"
           }
         },
         "type": "object",
         "required": [
           "name",
           "provider_type",
-          "endpoint",
-          "auth_type"
+          "endpoint"
         ],
         "title": "ProviderEndpoint",
         "description": "Represents a provider's endpoint configuration. This\nallows us to persist the configuration for each provider,\nso we can use this for muxing messages."

src/features/workspace/components/workspace-preferred-model.tsx

@@ -26,7 +26,7 @@ export function WorkspacePreferredModel({
   const { preferredModel, setPreferredModel } = usePreferredModelWorkspace();
   const { mutateAsync } = useMutationPreferredModelWorkspace();
   const { data: providerModels = [] } = useModelsData();
-  const { model, provider } = preferredModel;
+  const { model, provider_id } = preferredModel;
 
   const handleSubmit = (event: FormEvent) => {
     event.preventDefault();
@@ -35,7 +35,7 @@ export function WorkspacePreferredModel({
       body: [
         {
           matcher: "",
-          provider,
+          provider_id,
           model,
           matcher_type: MuxMatcherType.CATCH_ALL,
         },
@@ -69,7 +69,7 @@ export function WorkspacePreferredModel({
                   if (preferredModelProvider) {
                     setPreferredModel({
                       model: preferredModelProvider.name,
-                      provider: preferredModelProvider.provider_id,
+                      provider_id: preferredModelProvider.provider_id,
                     });
                   }
                 }}

src/features/workspace/hooks/use-preferred-preferred-model.ts

@@ -10,10 +10,10 @@ type State = {
 
 export const usePreferredModelWorkspace = create<State>((set) => ({
   preferredModel: {
-    provider: "",
+    provider_id: "",
     model: "",
   },
-  setPreferredModel: ({ model, provider }: ModelRule) => {
-    set({ preferredModel: { provider, model } });
+  setPreferredModel: ({ model, provider_id }: ModelRule) => {
+    set({ preferredModel: { provider_id, model } });
   },
 }));

src/mocks/msw/handlers.ts

@@ -96,13 +96,13 @@ export const handlers = [
   http.get("*/api/v1/workspaces/:workspace_name/muxes", () =>
     HttpResponse.json([
       {
-        provider: "openai",
+        provider_id: "openai",
         model: "gpt-3.5-turbo",
         matcher_type: "file_regex",
         matcher: ".*\\.txt",
       },
       {
-        provider: "anthropic",
+        provider_id: "anthropic",
         model: "davinci",
         matcher_type: "catch_all",
       },