@shipfox/api-agent 2.0.0

package/src/presentation/routes/set-default-harness.ts

@@ -0,0 +1,33 @@
+import {setDefaultHarnessBodySchema, setDefaultHarnessResponseSchema} from '@shipfox/api-agent-dto';
+import {requireWorkspaceAccess} from '@shipfox/api-auth-context';
+import {defineRoute} from '@shipfox/node-fastify';
+import {z} from 'zod';
+import {setDefaultHarness} from '#db/index.js';
+
+export const setDefaultHarnessRoute = defineRoute({
+  method: 'PUT',
+  path: '/default-harness',
+  description: 'Set the default agent harness for a workspace',
+  schema: {
+    params: z.object({workspaceId: z.string().uuid()}),
+    body: setDefaultHarnessBodySchema,
+    response: {
+      200: setDefaultHarnessResponseSchema,
+    },
+  },
+  handler: async (request) => {
+    const {workspaceId} = request.params;
+    requireWorkspaceAccess({request, workspaceId});
+
+    const settings = await setDefaultHarness({
+      workspaceId,
+      harnessId: request.body.harness_id,
+    });
+
+    if (settings.defaultHarnessId === null) {
+      throw new Error('Default harness upsert returned a null harness');
+    }
+
+    return {default_harness_id: settings.defaultHarnessId};
+  },
+});

package/src/presentation/routes/set-default-model-provider.ts

@@ -0,0 +1,47 @@
+import {
+  getModelProviderEntry,
+  setDefaultModelProviderBodySchema,
+  setDefaultModelProviderResponseSchema,
+} from '@shipfox/api-agent-dto';
+import {requireWorkspaceAccess} from '@shipfox/api-auth-context';
+import {defineRoute} from '@shipfox/node-fastify';
+import {z} from 'zod';
+import {UnsupportedModelProviderError} from '#core/index.js';
+import {getModelProviderConfig, setDefaultModelProvider} from '#db/index.js';
+import {translateModelProviderRouteError} from './errors.js';
+
+export const setDefaultModelProviderRoute = defineRoute({
+  method: 'PUT',
+  path: '/default-model-provider',
+  description: 'Set the default model provider for a workspace',
+  schema: {
+    params: z.object({workspaceId: z.string().uuid()}),
+    body: setDefaultModelProviderBodySchema,
+    response: {
+      200: setDefaultModelProviderResponseSchema,
+    },
+  },
+  errorHandler: translateModelProviderRouteError,
+  handler: async (request) => {
+    const {workspaceId} = request.params;
+    requireWorkspaceAccess({request, workspaceId});
+
+    const existingConfig = await getModelProviderConfig({
+      workspaceId,
+      providerId: request.body.provider_id,
+    });
+    if (existingConfig?.kind !== 'custom') {
+      const entry = getModelProviderEntry(request.body.provider_id);
+      if (entry === undefined || entry.support_status !== 'supported') {
+        throw new UnsupportedModelProviderError(request.body.provider_id);
+      }
+    }
+
+    const settings = await setDefaultModelProvider({
+      workspaceId,
+      providerId: request.body.provider_id,
+    });
+
+    return {default_provider_id: settings.defaultProviderId};
+  },
+});

package/src/presentation/routes/update-custom-model-provider.ts

@@ -0,0 +1,50 @@
+import {
+  customModelProviderConfigDtoSchema,
+  modelProviderRefSchema,
+  updateCustomModelProviderBodySchema,
+} from '@shipfox/api-agent-dto';
+import {requireWorkspaceAccess} from '@shipfox/api-auth-context';
+import {defineRoute} from '@shipfox/node-fastify';
+import {z} from 'zod';
+import {updateCustomModelProviderConfig} from '#core/index.js';
+import {toCustomModelProviderConfigDto} from '#presentation/dto/index.js';
+import {translateModelProviderRouteError} from './errors.js';
+
+export const updateCustomModelProviderRoute = defineRoute({
+  method: 'PUT',
+  path: '/custom-model-providers/:slug',
+  description: 'Update a custom model provider configuration for a workspace',
+  schema: {
+    params: z.object({
+      workspaceId: z.string().uuid(),
+      slug: modelProviderRefSchema,
+    }),
+    body: updateCustomModelProviderBodySchema,
+    response: {
+      200: customModelProviderConfigDtoSchema,
+    },
+  },
+  errorHandler: translateModelProviderRouteError,
+  handler: async (request, reply) => {
+    const {workspaceId, slug} = request.params;
+    const abortController = new AbortController();
+    let responseFinished = false;
+    reply.raw.on('finish', () => {
+      responseFinished = true;
+    });
+    reply.raw.on('close', () => {
+      if (!responseFinished) abortController.abort();
+    });
+
+    requireWorkspaceAccess({request, workspaceId});
+
+    const config = await updateCustomModelProviderConfig({
+      workspaceId,
+      providerId: slug,
+      body: request.body,
+      signal: abortController.signal,
+    });
+
+    return toCustomModelProviderConfigDto(config);
+  },
+});

package/src/presentation/routes/update-model-provider-default-model.ts

@@ -0,0 +1,40 @@
+import {
+  modelProviderConfigResponseSchema,
+  modelProviderRefSchema,
+  updateModelProviderDefaultModelBodySchema,
+} from '@shipfox/api-agent-dto';
+import {requireWorkspaceAccess} from '@shipfox/api-auth-context';
+import {defineRoute} from '@shipfox/node-fastify';
+import {z} from 'zod';
+import {updateModelProviderConfigDefaultModel} from '#core/index.js';
+import {toModelProviderConfigResponseDto} from '#presentation/dto/index.js';
+import {translateModelProviderRouteError} from './errors.js';
+
+export const updateModelProviderDefaultModelRoute = defineRoute({
+  method: 'PUT',
+  path: '/model-providers/:providerId/default-model',
+  description: 'Update the default model for an existing model provider configuration',
+  schema: {
+    params: z.object({
+      workspaceId: z.string().uuid(),
+      providerId: modelProviderRefSchema,
+    }),
+    body: updateModelProviderDefaultModelBodySchema,
+    response: {
+      200: modelProviderConfigResponseSchema,
+    },
+  },
+  errorHandler: translateModelProviderRouteError,
+  handler: async (request) => {
+    const {workspaceId, providerId} = request.params;
+    requireWorkspaceAccess({request, workspaceId});
+
+    const config = await updateModelProviderConfigDefaultModel({
+      workspaceId,
+      providerId,
+      defaultModel: request.body.default_model,
+    });
+
+    return toModelProviderConfigResponseDto(config);
+  },
+});

package/src/presentation/routes/upsert-model-provider-config.ts

@@ -0,0 +1,53 @@
+import {
+  modelProviderConfigDtoSchema,
+  supportedModelProviderIdSchema,
+  updateModelProviderConfigBodySchema,
+} from '@shipfox/api-agent-dto';
+import {requireWorkspaceAccess} from '@shipfox/api-auth-context';
+import {defineRoute} from '@shipfox/node-fastify';
+import {z} from 'zod';
+import {testAndSaveModelProviderConfig} from '#core/index.js';
+import {toModelProviderConfigDto} from '#presentation/dto/index.js';
+import {translateModelProviderRouteError} from './errors.js';
+
+export const upsertModelProviderConfigRoute = defineRoute({
+  method: 'PUT',
+  path: '/model-providers/:providerId',
+  description: 'Test and save a model provider configuration for a workspace',
+  schema: {
+    params: z.object({
+      workspaceId: z.string().uuid(),
+      providerId: supportedModelProviderIdSchema,
+    }),
+    body: updateModelProviderConfigBodySchema,
+    response: {
+      200: modelProviderConfigDtoSchema,
+    },
+  },
+  errorHandler: translateModelProviderRouteError,
+  handler: async (request, reply) => {
+    const {workspaceId, providerId} = request.params;
+    const abortController = new AbortController();
+    let responseFinished = false;
+    reply.raw.on('finish', () => {
+      responseFinished = true;
+    });
+    reply.raw.on('close', () => {
+      if (!responseFinished) abortController.abort();
+    });
+
+    const membership = requireWorkspaceAccess({request, workspaceId});
+
+    const config = await testAndSaveModelProviderConfig({
+      workspaceId,
+      providerId,
+      ...('default_model' in request.body ? {defaultModel: request.body.default_model} : {}),
+      credentials: request.body.credentials,
+      editedBy: membership.userId,
+      setAsDefault: request.body.set_as_default,
+      signal: abortController.signal,
+    });
+
+    return toModelProviderConfigDto(config);
+  },
+});

package/test/env.ts

@@ -0,0 +1,8 @@
+process.env.POSTGRES_HOST ??= 'localhost';
+process.env.POSTGRES_PORT ??= '5432';
+process.env.POSTGRES_USERNAME ??= 'shipfox';
+process.env.POSTGRES_PASSWORD ??= 'password';
+process.env.POSTGRES_DATABASE = 'api_test';
+process.env.POSTGRES_MAX_CONNECTIONS ??= '5';
+process.env.TZ = 'UTC';
+process.env.SECRETS_ENCRYPTION_KEK = 'ZmVkY2JhOTg3NjU0MzIxMGZlZGNiYTk4NzY1NDMyMTA=';

package/test/globalSetup.ts

@@ -0,0 +1,25 @@
+import './env.js';
+import {secretsModule} from '@shipfox/api-secrets';
+import {runMigrations} from '@shipfox/node-drizzle';
+import {closePostgresClient, createPostgresClient} from '@shipfox/node-postgres';
+import {sql} from 'drizzle-orm';
+import {closeDb, db, migrationsPath} from '#db/index.js';
+
+export async function setup() {
+  createPostgresClient();
+
+  await runMigrations(db(), migrationsPath, '__drizzle_migrations_agent');
+  if (!secretsModule.database || Array.isArray(secretsModule.database)) {
+    throw new Error('Secrets module database is not configured');
+  }
+  await runMigrations(
+    secretsModule.database.db(),
+    secretsModule.database.migrationsPath,
+    '__drizzle_migrations_secrets',
+  );
+  await db().execute(sql`TRUNCATE model_provider_configs CASCADE`);
+  await db().execute(sql`TRUNCATE agent_workspace_settings CASCADE`);
+
+  closeDb();
+  await closePostgresClient();
+}

package/test/setup.ts

@@ -0,0 +1,17 @@
+import './env.js';
+import {closePostgresClient, createPostgresClient} from '@shipfox/node-postgres';
+import {afterAll, afterEach, beforeAll, vi} from '@shipfox/vitest/vi';
+import {closeDb} from '#db/index.js';
+
+beforeAll(() => {
+  createPostgresClient();
+});
+
+afterEach(() => {
+  vi.restoreAllMocks();
+});
+
+afterAll(async () => {
+  closeDb();
+  await closePostgresClient();
+});

package/tsconfig.build.json

@@ -0,0 +1,9 @@
+{
+  "extends": "@shipfox/ts-config/node",
+  "compilerOptions": {
+    "rootDir": "src",
+    "outDir": "dist"
+  },
+  "include": ["src"],
+  "exclude": ["**/*.test.tsx", "**/*.test.ts"]
+}