@shipfox/api-agent 5.0.0 → 7.1.0

package/src/presentation/inter-module.ts

@@ -0,0 +1,78 @@
+import {agentInterModuleContract} from '@shipfox/api-agent-dto/inter-module';
+import {secretsInterModuleContract} from '@shipfox/api-secrets-dto/inter-module';
+import {
+  createInterModuleKnownError,
+  defineInterModulePresentation,
+  type InterModulePresentation,
+  isInterModuleKnownError,
+} from '@shipfox/inter-module';
+import {
+  InvalidAgentModelError,
+  ModelProviderConfigNotFoundError,
+  UnsupportedHarnessProviderError,
+  UnsupportedHarnessThinkingError,
+  UnsupportedModelProviderError,
+} from '#core/errors.js';
+import {resolveAgentConfig} from '#core/resolve-agent-config.js';
+import {resolveRuntimeCredentials} from '#core/resolve-runtime-credentials.js';
+import type {AgentSecretsClient} from '#core/secrets-client.js';
+import {createWorkspaceAgentDefaultsResolver} from '#core/workspace-agent-defaults-resolver.js';
+
+export function createAgentInterModulePresentation(params: {
+  secrets: AgentSecretsClient;
+}): InterModulePresentation<typeof agentInterModuleContract> {
+  return defineInterModulePresentation(agentInterModuleContract, {
+    resolveAgentConfig: async ({workspaceId, config}) => {
+      try {
+        const resolve =
+          workspaceId === null
+            ? resolveAgentConfig
+            : await createWorkspaceAgentDefaultsResolver(workspaceId);
+        return await resolve(config);
+      } catch (error) {
+        throw toResolveAgentConfigKnownError(error);
+      }
+    },
+    resolveRuntimeCredentials: async (input) => {
+      try {
+        return await resolveRuntimeCredentials(input, {secrets: params.secrets});
+      } catch (error) {
+        throw toResolveRuntimeCredentialsKnownError(error);
+      }
+    },
+  });
+}
+
+function toResolveAgentConfigKnownError(error: unknown): unknown {
+  if (
+    error instanceof InvalidAgentModelError ||
+    error instanceof UnsupportedHarnessProviderError ||
+    error instanceof UnsupportedHarnessThinkingError ||
+    error instanceof UnsupportedModelProviderError
+  ) {
+    return createInterModuleKnownError(
+      agentInterModuleContract.methods.resolveAgentConfig,
+      'agent-config-invalid',
+      {},
+    );
+  }
+  return error;
+}
+
+function toResolveRuntimeCredentialsKnownError(error: unknown): unknown {
+  if (error instanceof ModelProviderConfigNotFoundError) {
+    return createInterModuleKnownError(
+      agentInterModuleContract.methods.resolveRuntimeCredentials,
+      'model-provider-not-configured',
+      {},
+    );
+  }
+  if (isInterModuleKnownError(secretsInterModuleContract.methods.getSecretsByNamespace, error)) {
+    return createInterModuleKnownError(
+      agentInterModuleContract.methods.resolveRuntimeCredentials,
+      'model-provider-credentials-invalid',
+      {},
+    );
+  }
+  return error;
+}

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

@@ -6,40 +6,46 @@ import {requireWorkspaceAccess} from '@shipfox/api-auth-context';
 import {defineRoute} from '@shipfox/node-fastify';
 import {z} from 'zod';
 import {createCustomModelProviderConfig} from '#core/index.js';
+import type {AgentSecretsClient} from '#core/secrets-client.js';
 import {toCustomModelProviderConfigDto} from '#presentation/dto/index.js';
 import {translateModelProviderRouteError} from './errors.js';
 
-export const createCustomModelProviderRoute = defineRoute({
-  method: 'POST',
-  path: '/custom-model-providers',
-  description: 'Create a custom model provider configuration for a workspace',
-  schema: {
-    params: z.object({workspaceId: z.string().uuid()}),
-    body: createCustomModelProviderBodySchema,
-    response: {
-      200: customModelProviderConfigDtoSchema,
+export function createCustomModelProviderRoute(secrets: AgentSecretsClient) {
+  return defineRoute({
+    method: 'POST',
+    path: '/custom-model-providers',
+    description: 'Create a custom model provider configuration for a workspace',
+    schema: {
+      params: z.object({workspaceId: z.string().uuid()}),
+      body: createCustomModelProviderBodySchema,
+      response: {
+        200: customModelProviderConfigDtoSchema,
+      },
     },
-  },
-  errorHandler: translateModelProviderRouteError,
-  handler: async (request, reply) => {
-    const {workspaceId} = request.params;
-    const abortController = new AbortController();
-    let responseFinished = false;
-    reply.raw.on('finish', () => {
-      responseFinished = true;
-    });
-    reply.raw.on('close', () => {
-      if (!responseFinished) abortController.abort();
-    });
+    errorHandler: translateModelProviderRouteError,
+    handler: async (request, reply) => {
+      const {workspaceId} = 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});
+      requireWorkspaceAccess({request, workspaceId});
 
-    const config = await createCustomModelProviderConfig({
-      workspaceId,
-      body: request.body,
-      signal: abortController.signal,
-    });
+      const config = await createCustomModelProviderConfig(
+        {
+          workspaceId,
+          body: request.body,
+          signal: abortController.signal,
+        },
+        {secrets},
+      );
 
-    return toCustomModelProviderConfigDto(config);
-  },
-});
+      return toCustomModelProviderConfigDto(config);
+    },
+  });
+}

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

@@ -3,29 +3,32 @@ import {requireWorkspaceAccess} from '@shipfox/api-auth-context';
 import {ClientError, defineRoute} from '@shipfox/node-fastify';
 import {z} from 'zod';
 import {deleteModelProviderConfig} from '#core/index.js';
+import type {AgentSecretsClient} from '#core/secrets-client.js';
 
-export const deleteModelProviderConfigRoute = defineRoute({
-  method: 'DELETE',
-  path: '/model-providers/:providerId',
-  description: 'Delete a model provider configuration for a workspace',
-  schema: {
-    params: z.object({
-      workspaceId: z.string().uuid(),
-      providerId: modelProviderRefSchema,
-    }),
-    response: {
-      204: z.void(),
+export function createDeleteModelProviderConfigRoute(secrets: AgentSecretsClient) {
+  return defineRoute({
+    method: 'DELETE',
+    path: '/model-providers/:providerId',
+    description: 'Delete a model provider configuration for a workspace',
+    schema: {
+      params: z.object({
+        workspaceId: z.string().uuid(),
+        providerId: modelProviderRefSchema,
+      }),
+      response: {
+        204: z.void(),
+      },
     },
-  },
-  handler: async (request, reply) => {
-    const {workspaceId, providerId} = request.params;
-    requireWorkspaceAccess({request, workspaceId});
+    handler: async (request, reply) => {
+      const {workspaceId, providerId} = request.params;
+      requireWorkspaceAccess({request, workspaceId});
 
-    const deleted = await deleteModelProviderConfig({workspaceId, providerId});
-    if (!deleted) {
-      throw new ClientError('Model provider configuration not found', 'not-found', {status: 404});
-    }
+      const deleted = await deleteModelProviderConfig({workspaceId, providerId}, {secrets});
+      if (!deleted) {
+        throw new ClientError('Model provider configuration not found', 'not-found', {status: 404});
+      }
 
-    reply.code(204);
-  },
-});
+      reply.code(204);
+    },
+  });
+}

package/src/presentation/routes/discover-custom-model-provider-models-by-slug.ts

@@ -10,33 +10,39 @@ import {
   discoverCustomModelProviderModels,
   resolveCustomModelProviderDiscoveryParams,
 } from '#core/index.js';
+import type {AgentSecretsClient} from '#core/secrets-client.js';
 import {translateModelProviderRouteError} from './errors.js';
 
-export const discoverCustomModelProviderModelsBySlugRoute = defineRoute({
-  method: 'POST',
-  path: '/custom-model-providers/:slug/discover-models',
-  description: 'Discover models for a stored custom model provider configuration',
-  schema: {
-    params: z.object({
-      workspaceId: z.string().uuid(),
-      slug: modelProviderRefSchema,
-    }),
-    body: discoverCustomModelProviderModelsBySlugBodySchema,
-    response: {
-      200: discoverCustomModelProviderModelsResponseSchema,
+export function createDiscoverCustomModelProviderModelsBySlugRoute(secrets: AgentSecretsClient) {
+  return defineRoute({
+    method: 'POST',
+    path: '/custom-model-providers/:slug/discover-models',
+    description: 'Discover models for a stored custom model provider configuration',
+    schema: {
+      params: z.object({
+        workspaceId: z.string().uuid(),
+        slug: modelProviderRefSchema,
+      }),
+      body: discoverCustomModelProviderModelsBySlugBodySchema,
+      response: {
+        200: discoverCustomModelProviderModelsResponseSchema,
+      },
     },
-  },
-  errorHandler: translateModelProviderRouteError,
-  handler: async (request) => {
-    const {workspaceId, slug} = request.params;
-    requireWorkspaceAccess({request, workspaceId});
+    errorHandler: translateModelProviderRouteError,
+    handler: async (request) => {
+      const {workspaceId, slug} = request.params;
+      requireWorkspaceAccess({request, workspaceId});
 
-    const discoveryParams = await resolveCustomModelProviderDiscoveryParams({
-      workspaceId,
-      providerId: slug,
-      body: request.body ?? {},
-    });
+      const discoveryParams = await resolveCustomModelProviderDiscoveryParams(
+        {
+          workspaceId,
+          providerId: slug,
+          body: request.body ?? {},
+        },
+        {secrets},
+      );
 
-    return await discoverCustomModelProviderModels(discoveryParams);
-  },
-});
+      return await discoverCustomModelProviderModels(discoveryParams);
+    },
+  });
+}

package/src/presentation/routes/errors.ts

@@ -1,5 +1,6 @@
 import {getModelProviderCredentialKeys} from '@shipfox/api-agent-dto';
-import {SecretValueTooLargeError, WorkspaceSecretCapExceededError} from '@shipfox/api-secrets';
+import {secretsInterModuleContract} from '@shipfox/api-secrets-dto/inter-module';
+import {isInterModuleKnownError} from '@shipfox/inter-module';
 import {EgressDeniedError} from '@shipfox/node-egress-guard';
 import {ClientError} from '@shipfox/node-fastify';
 import {
@@ -104,17 +105,16 @@ export function translateModelProviderRouteError(error: unknown): never {
     });
   }
 
-  if (error instanceof WorkspaceSecretCapExceededError) {
-    throw new ClientError('Workspace secret cap exceeded', 'workspace-secret-cap-exceeded', {
-      status: 409,
-      details: {cap: error.cap},
-    });
-  }
-
-  if (error instanceof SecretValueTooLargeError) {
-    throw new ClientError(error.message, 'value-too-large', {
+  if (isInterModuleKnownError(secretsInterModuleContract.methods.setSecrets, error)) {
+    if (error.code === 'workspace-secret-cap-exceeded') {
+      throw new ClientError('Workspace secret cap exceeded', 'workspace-secret-cap-exceeded', {
+        status: 409,
+        details: {cap: error.details.cap},
+      });
+    }
+    throw new ClientError('Secret value is too large', 'value-too-large', {
       status: 400,
-      details: {max_bytes: error.maxBytes},
+      details: {max_bytes: error.details.maxBytes},
     });
   }
 

package/src/presentation/routes/index.ts

@@ -1,37 +1,42 @@
 import {AUTH_USER} from '@shipfox/api-auth-context';
 import type {RouteGroup} from '@shipfox/node-fastify';
+import type {AgentSecretsClient} from '#core/secrets-client.js';
 import {createCustomModelProviderRoute} from './create-custom-model-provider.js';
-import {deleteModelProviderConfigRoute} from './delete-model-provider-config.js';
+import {createDeleteModelProviderConfigRoute} from './delete-model-provider-config.js';
 import {discoverCustomModelProviderModelsRoute} from './discover-custom-model-provider-models.js';
-import {discoverCustomModelProviderModelsBySlugRoute} from './discover-custom-model-provider-models-by-slug.js';
+import {createDiscoverCustomModelProviderModelsBySlugRoute} from './discover-custom-model-provider-models-by-slug.js';
 import {listModelProviderCatalogRoute} from './list-model-provider-catalog.js';
 import {listModelProviderConfigsRoute} from './list-model-provider-configs.js';
 import {setDefaultHarnessRoute} from './set-default-harness.js';
 import {setDefaultModelProviderRoute} from './set-default-model-provider.js';
-import {updateCustomModelProviderRoute} from './update-custom-model-provider.js';
+import {createUpdateCustomModelProviderRoute} from './update-custom-model-provider.js';
 import {updateModelProviderDefaultModelRoute} from './update-model-provider-default-model.js';
-import {upsertModelProviderConfigRoute} from './upsert-model-provider-config.js';
+import {createUpsertModelProviderConfigRoute} from './upsert-model-provider-config.js';
 
-export const agentRoutes: RouteGroup[] = [
-  {
-    prefix: '/workspaces/:workspaceId/agent',
-    auth: AUTH_USER,
-    routes: [
-      listModelProviderConfigsRoute,
-      createCustomModelProviderRoute,
-      discoverCustomModelProviderModelsRoute,
-      discoverCustomModelProviderModelsBySlugRoute,
-      updateCustomModelProviderRoute,
-      upsertModelProviderConfigRoute,
-      updateModelProviderDefaultModelRoute,
-      deleteModelProviderConfigRoute,
-      setDefaultHarnessRoute,
-      setDefaultModelProviderRoute,
-    ],
-  },
-  {
-    prefix: '/agent',
-    auth: AUTH_USER,
-    routes: [listModelProviderCatalogRoute],
-  },
-];
+export function createAgentRoutes(secrets: AgentSecretsClient): RouteGroup[] {
+  return [
+    {
+      prefix: '/workspaces/:workspaceId/agent',
+      auth: AUTH_USER,
+      routes: [
+        listModelProviderConfigsRoute,
+        createCustomModelProviderRoute(secrets),
+        discoverCustomModelProviderModelsRoute,
+        createDiscoverCustomModelProviderModelsBySlugRoute(secrets),
+        createUpdateCustomModelProviderRoute(secrets),
+        createUpsertModelProviderConfigRoute(secrets),
+        updateModelProviderDefaultModelRoute,
+        createDeleteModelProviderConfigRoute(secrets),
+        setDefaultHarnessRoute,
+        setDefaultModelProviderRoute,
+      ],
+    },
+    {
+      prefix: '/agent',
+      auth: AUTH_USER,
+      routes: [listModelProviderCatalogRoute],
+    },
+  ];
+}
+
+export const agentRoutes = createAgentRoutes(undefined as unknown as AgentSecretsClient);

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

@@ -6,7 +6,6 @@ import {
   setUserContext,
   type UserContextMembership,
 } from '@shipfox/api-auth-context';
-import {getSecretsByNamespace, setSecrets} from '@shipfox/api-secrets';
 import type {AuthMethod, FastifyRequest} from '@shipfox/node-fastify';
 import {ClientError, closeApp, createApp} from '@shipfox/node-fastify';
 import {eq} from 'drizzle-orm';
@@ -20,6 +19,7 @@ import {
   upsertModelProviderConfig,
 } from '#db/index.js';
 import {modelProviderConfigs} from '#db/schema/model-provider-configs.js';
+import {getSecretsByNamespace, setSecrets} from '#test/fixtures/secrets-client.js';
 import {agentRoutes} from './index.js';
 
 vi.mock('@earendil-works/pi-ai', async (importOriginal) => {

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

@@ -7,44 +7,50 @@ import {requireWorkspaceAccess} from '@shipfox/api-auth-context';
 import {defineRoute} from '@shipfox/node-fastify';
 import {z} from 'zod';
 import {updateCustomModelProviderConfig} from '#core/index.js';
+import type {AgentSecretsClient} from '#core/secrets-client.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,
+export function createUpdateCustomModelProviderRoute(secrets: AgentSecretsClient) {
+  return 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();
-    });
+    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});
+      requireWorkspaceAccess({request, workspaceId});
 
-    const config = await updateCustomModelProviderConfig({
-      workspaceId,
-      providerId: slug,
-      body: request.body,
-      signal: abortController.signal,
-    });
+      const config = await updateCustomModelProviderConfig(
+        {
+          workspaceId,
+          providerId: slug,
+          body: request.body,
+          signal: abortController.signal,
+        },
+        {secrets},
+      );
 
-    return toCustomModelProviderConfigDto(config);
-  },
-});
+      return toCustomModelProviderConfigDto(config);
+    },
+  });
+}

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

@@ -7,47 +7,53 @@ import {requireWorkspaceAccess} from '@shipfox/api-auth-context';
 import {defineRoute} from '@shipfox/node-fastify';
 import {z} from 'zod';
 import {testAndSaveModelProviderConfig} from '#core/index.js';
+import type {AgentSecretsClient} from '#core/secrets-client.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,
+export function createUpsertModelProviderConfigRoute(secrets: AgentSecretsClient) {
+  return 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();
-    });
+    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 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,
-    });
+      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,
+        },
+        {secrets},
+      );
 
-    return toModelProviderConfigDto(config);
-  },
-});
+      return toModelProviderConfigDto(config);
+    },
+  });
+}