-
-
-
-
-
-
-
setLocation(event.target.value)}
- placeholder={VERTEX_DEFAULT_LOCATION}
- />
-
- Region where your Google Vertex AI models are hosted.
-
-
- {authMethod === VERTEX_AUTH_SERVICE_ACCOUNT ? (
-
-
-
handleCredentialsFile(event.target.files?.[0])}
- />
-
-
- {credentials
- ? "Credentials file loaded."
- : "Attach your service account key JSON from Google Cloud."}
-
-
- ) : (
-
-
-
setProject(event.target.value)}
- placeholder="my-vertex-project"
- />
-
- The GCP project where Vertex AI is enabled.
-
-
- )}
+
+
+
+
+
+
+
+
setLocation(event.target.value)}
+ placeholder={VERTEX_DEFAULT_LOCATION}
+ />
- Add Vertex AI model IDs from the provider's settings after connecting.
+ Region where your Google Vertex AI models are hosted.
-
-
+ {authMethod === VERTEX_AUTH_SERVICE_ACCOUNT ? (
+
+
+
handleCredentialsFile(event.target.files?.[0])}
+ />
+
+
+ {credentials
+ ? "Credentials file loaded."
+ : "Attach your service account key JSON from Google Cloud."}
+
+
+ ) : (
+
+
+
setProject(event.target.value)}
+ placeholder="my-vertex-project"
+ />
+
+ The GCP project where Vertex AI is enabled.
+
+
+ )}
+
+ Add Vertex AI model IDs from the provider's settings after connecting.
+
+
);
}
diff --git a/surfsense_web/contracts/types/model-connections.types.ts b/surfsense_web/contracts/types/model-connections.types.ts
index c75f4c90a..134c740b2 100644
--- a/surfsense_web/contracts/types/model-connections.types.ts
+++ b/surfsense_web/contracts/types/model-connections.types.ts
@@ -40,6 +40,21 @@ export const connectionRead = z.object({
created_at: z.string().nullable().optional(),
});
+export const modelSelection = z.object({
+ model_id: z.string().min(1),
+ display_name: z.string().nullable().optional(),
+ source: z.union([modelSourceEnum, z.string()]).default("DISCOVERED"),
+ supports_chat: z.boolean().nullable().optional(),
+ max_input_tokens: z.number().nullable().optional(),
+ supports_image_input: z.boolean().nullable().optional(),
+ supports_tools: z.boolean().nullable().optional(),
+ supports_image_generation: z.boolean().nullable().optional(),
+ enabled: z.boolean().default(false),
+ metadata: z.record(z.string(), z.any()).default({}),
+});
+
+export const modelPreviewRead = modelSelection;
+
export const connectionCreateRequest = z.object({
provider: z.string().min(1),
base_url: z.string().nullable().optional(),
@@ -48,6 +63,7 @@ export const connectionCreateRequest = z.object({
scope: connectionScopeEnum.default("SEARCH_SPACE"),
search_space_id: z.number().nullable().optional(),
enabled: z.boolean().default(true),
+ models: z.array(modelSelection).default([]),
});
export const connectionUpdateRequest = z.object({
@@ -105,9 +121,12 @@ export const modelProviderListResponse = z.array(modelProviderRead);
export const connectionListResponse = z.array(connectionRead);
export const modelListResponse = z.array(modelRead);
+export const modelPreviewListResponse = z.array(modelPreviewRead);
export type ConnectionScope = z.infer
;
export type ModelRead = z.infer;
+export type ModelPreviewRead = z.infer;
+export type ModelSelection = z.infer;
export type ConnectionRead = z.infer;
export type ConnectionCreateRequest = z.infer;
export type ConnectionUpdateRequest = z.infer;
diff --git a/surfsense_web/lib/apis/model-connections-api.service.ts b/surfsense_web/lib/apis/model-connections-api.service.ts
index bd5aa1309..f463a27e7 100644
--- a/surfsense_web/lib/apis/model-connections-api.service.ts
+++ b/surfsense_web/lib/apis/model-connections-api.service.ts
@@ -7,6 +7,7 @@ import {
connectionRead,
connectionUpdateRequest,
type ModelCreateRequest,
+ type ModelPreviewRead,
type ModelProviderRead,
type ModelRead,
type ModelRoles,
@@ -14,6 +15,7 @@ import {
type ModelUpdateRequest,
modelCreateRequest,
modelListResponse,
+ modelPreviewListResponse,
modelProviderListResponse,
modelRead,
modelRoles,
@@ -76,6 +78,20 @@ class ModelConnectionsApiService {
return baseApiService.post(`/api/v1/model-connections/${id}/discover`, modelListResponse);
};
+ previewModels = async (request: ConnectionCreateRequest): Promise => {
+ const parsed = connectionCreateRequest.safeParse(request);
+ if (!parsed.success) {
+ throw new ValidationError(parsed.error.issues.map((issue) => issue.message).join(", "));
+ }
+ return baseApiService.post(
+ `/api/v1/model-connections/discover-preview`,
+ modelPreviewListResponse,
+ {
+ body: parsed.data,
+ }
+ );
+ };
+
addManualModel = async (
connectionId: number,
request: ModelCreateRequest