@agentuity/server 1.0.1 → 1.0.3

package/src/api/apikey/get.ts

@@ -2,7 +2,7 @@ import { z } from 'zod';
 import { APIClient, APIResponseSchema } from '../api';
 import { APIKeyResponseError } from './util';
 
-const APIKeyDetailSchema = z.object({
+export const APIKeyDetailSchema = z.object({
 	id: z.string().describe('the API key id'),
 	name: z.string().describe('the API key name'),
 	orgId: z.string().describe('the organization id'),
@@ -20,7 +20,7 @@ const APIKeyDetailSchema = z.object({
 		.describe('the associated project'),
 });
 
-const APIKeyGetResponseSchema = APIResponseSchema(APIKeyDetailSchema);
+export const APIKeyGetResponseSchema = APIResponseSchema(APIKeyDetailSchema);
 
 type APIKeyGetResponse = z.infer<typeof APIKeyGetResponseSchema>;
 

package/src/api/apikey/index.ts

@@ -2,3 +2,4 @@ export * from './create';
 export * from './delete';
 export * from './get';
 export * from './list';
+export * from './util';

package/src/api/apikey/list.ts

@@ -22,7 +22,7 @@ export const APIKeySchema = z.object({
 
 export const APIKeyListResponseArray = z.array(APIKeySchema);
 
-const APIKeyListResponseSchema = APIResponseSchema(APIKeyListResponseArray);
+export const APIKeyListResponseSchema = APIResponseSchema(APIKeyListResponseArray);
 
 export type APIKeyListResponse = z.infer<typeof APIKeyListResponseSchema>;
 export type APIKeyList = z.infer<typeof APIKeyListResponseArray>;

package/src/api/db/index.ts

@@ -1,10 +1,27 @@
-export { dbLogs, DbQueryLogSchema, type DbQueryLog, type DbQueryLogs } from './logs';
+export {
+	DbLogsAPIResponseSchema,
+	DbLogsRequestSchema,
+	DbLogsResponseSchema,
+	type DbQueryLog,
+	DbQueryLogSchema,
+	type DbQueryLogs,
+	dbLogs,
+} from './logs';
+export {
+	dbQuery,
+	type QueryColumn,
+	QueryColumnSchema,
+	QueryResponseSchema,
+	type QueryResult,
+	QueryResultSchema,
+} from './query';
 export {
 	dbTables,
 	generateCreateTableSQL,
-	TableSchemaSchema,
 	type TableColumn,
+	TableColumnSchema,
 	type TableSchema,
+	TableSchemaSchema,
+	TablesResponseSchema,
 } from './tables';
-export { dbQuery, type QueryColumn, type QueryResult } from './query';
-export { DbResponseError, DbInvalidArgumentError } from './util';
+export { DbInvalidArgumentError, DbResponseError } from './util';

package/src/api/db/logs.ts

@@ -1,8 +1,8 @@
 import { z } from 'zod';
-import { APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api';
 import { DbInvalidArgumentError, DbResponseError } from './util';
 
-const _DbLogsRequestSchema = z.object({
+export const DbLogsRequestSchema = z.object({
 	database: z.string().describe('the database name'),
 	orgId: z.string().describe('the organization ID'),
 	region: z.string().describe('the region'),
@@ -28,12 +28,12 @@ export const DbQueryLogSchema = z.object({
 	sessionId: z.string().optional().describe('session ID with sess_ prefix'),
 });
 
-const DbLogsResponse = z.array(DbQueryLogSchema);
+export const DbLogsResponseSchema = z.array(DbQueryLogSchema);
 
-const DbLogsResponseSchema = APIResponseSchema(DbLogsResponse);
+export const DbLogsAPIResponseSchema = APIResponseSchema(DbLogsResponseSchema);
 
-type DbLogsRequest = z.infer<typeof _DbLogsRequestSchema>;
-type DbLogsResponse = z.infer<typeof DbLogsResponseSchema>;
+type DbLogsRequest = z.infer<typeof DbLogsRequestSchema>;
+type DbLogsAPIResponse = z.infer<typeof DbLogsAPIResponseSchema>;
 
 export type DbQueryLog = z.infer<typeof DbQueryLogSchema>;
 export type DbQueryLogs = DbQueryLog[];
@@ -62,7 +62,7 @@ export async function dbLogs(client: APIClient, request: DbLogsRequest): Promise
 	const queryString = params.toString();
 	const url = `/resource/2025-03-17/${orgId}/${region}/${database}/logs${queryString ? `?${queryString}` : ''}`;
 
-	const resp = await client.get<DbLogsResponse>(url, DbLogsResponseSchema);
+	const resp = await client.get<DbLogsAPIResponse>(url, DbLogsAPIResponseSchema);
 
 	if (resp.success) {
 		return resp.data;

package/src/api/db/query.ts

@@ -2,7 +2,7 @@ import { z } from 'zod';
 import { APIClient, APIError } from '../api';
 import { DbInvalidArgumentError, DbResponseError } from './util';
 
-const QueryColumnSchema = z.object({
+export const QueryColumnSchema = z.object({
 	name: z.string().describe('column name'),
 	type: z.string().describe('PostgreSQL data type OID'),
 });
@@ -14,7 +14,7 @@ export const QueryResultSchema = z.object({
 	truncated: z.boolean().describe('whether results were truncated (max 1000 rows)'),
 });
 
-const QueryResponseSchema = z.object({
+export const QueryResponseSchema = z.object({
 	success: z.boolean(),
 	message: z.string().optional(),
 	data: QueryResultSchema.optional(),

package/src/api/db/tables.ts

@@ -1,8 +1,8 @@
 import { z } from 'zod';
-import { APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api';
 import { DbInvalidArgumentError, DbResponseError } from './util';
 
-const TableColumnSchema = z.object({
+export const TableColumnSchema = z.object({
 	name: z.string().describe('column name'),
 	data_type: z.string().describe('PostgreSQL data type'),
 	is_nullable: z.boolean().describe('whether the column is nullable'),
@@ -15,7 +15,7 @@ export const TableSchemaSchema = z.object({
 	columns: z.array(TableColumnSchema).describe('table columns'),
 });
 
-const TablesResponseSchema = APIResponseSchema(
+export const TablesResponseSchema = APIResponseSchema(
 	z.object({
 		tables: z.array(TableSchemaSchema),
 	})

package/src/api/eval/get.ts

@@ -1,7 +1,7 @@
 import { z } from 'zod';
-import { APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api';
 
-const EvaluationDetailSchema = z.object({
+export const EvaluationDetailSchema = z.object({
 	id: z.string().describe('Evaluation ID'),
 	name: z.string().describe('Evaluation name'),
 	description: z.string().nullable().describe('Evaluation description'),
@@ -14,7 +14,7 @@ const EvaluationDetailSchema = z.object({
 	updatedAt: z.string().describe('Last updated timestamp'),
 });
 
-const EvalGetResponseSchema = APIResponseSchema(EvaluationDetailSchema);
+export const EvalGetResponseSchema = APIResponseSchema(EvaluationDetailSchema);
 
 export type EvaluationDetail = z.infer<typeof EvaluationDetailSchema>;
 

package/src/api/eval/index.ts

@@ -1,4 +1,28 @@
-export { evalList, type Evaluation, type EvaluationListRequest } from './list';
-export { evalGet, type EvaluationDetail } from './get';
-export { evalRunList, type EvalRunListItem, type EvalRunListRequest } from './run-list';
-export { evalRunGet, type EvalRunDetail } from './run-get';
+export {
+	EvalGetResponseSchema,
+	type EvaluationDetail,
+	EvaluationDetailSchema,
+	evalGet,
+} from './get';
+export {
+	EvalListResponseData,
+	EvalListResponseSchema,
+	type Evaluation,
+	type EvaluationListRequest,
+	EvaluationSchema,
+	evalList,
+} from './list';
+export {
+	type EvalRunDetail,
+	EvalRunDetailSchema,
+	EvalRunGetResponseSchema,
+	evalRunGet,
+} from './run-get';
+export {
+	type EvalRunListItem,
+	type EvalRunListRequest,
+	EvalRunListResponseData,
+	EvalRunListResponseSchema,
+	EvalRunSchema,
+	evalRunList,
+} from './run-list';

package/src/api/eval/list.ts

@@ -1,7 +1,7 @@
 import { z } from 'zod';
-import { APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api';
 
-const EvaluationSchema = z.object({
+export const EvaluationSchema = z.object({
 	id: z.string().describe('Evaluation ID'),
 	name: z.string().describe('Evaluation name'),
 	description: z.string().nullable().describe('Evaluation description'),
@@ -13,8 +13,8 @@ const EvaluationSchema = z.object({
 	updatedAt: z.string().describe('Last updated timestamp'),
 });
 
-const EvalListResponseData = z.array(EvaluationSchema);
-const EvalListResponseSchema = APIResponseSchema(EvalListResponseData);
+export const EvalListResponseData = z.array(EvaluationSchema);
+export const EvalListResponseSchema = APIResponseSchema(EvalListResponseData);
 
 export type Evaluation = z.infer<typeof EvaluationSchema>;
 export type EvaluationListRequest = {

package/src/api/eval/run-get.ts

@@ -1,7 +1,7 @@
 import { z } from 'zod';
-import { APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api';
 
-const EvalRunDetailSchema = z.object({
+export const EvalRunDetailSchema = z.object({
 	id: z.string().describe('Eval run ID'),
 	sessionId: z.string().describe('Session ID'),
 	evalId: z.string().describe('Evaluation record ID'),
@@ -20,7 +20,7 @@ const EvalRunDetailSchema = z.object({
 	updatedAt: z.string().describe('Last updated timestamp'),
 });
 
-const EvalRunGetResponseSchema = APIResponseSchema(EvalRunDetailSchema);
+export const EvalRunGetResponseSchema = APIResponseSchema(EvalRunDetailSchema);
 
 export type EvalRunDetail = z.infer<typeof EvalRunDetailSchema>;
 

package/src/api/eval/run-list.ts

@@ -1,7 +1,7 @@
 import { z } from 'zod';
-import { APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api';
 
-const EvalRunSchema = z.object({
+export const EvalRunSchema = z.object({
 	id: z.string().describe('Eval run ID'),
 	sessionId: z.string().describe('Session ID'),
 	evalId: z.string().describe('Evaluation record ID'),
@@ -19,8 +19,8 @@ const EvalRunSchema = z.object({
 	updatedAt: z.string().describe('Last updated timestamp'),
 });
 
-const EvalRunListResponseData = z.array(EvalRunSchema);
-const EvalRunListResponseSchema = APIResponseSchema(EvalRunListResponseData);
+export const EvalRunListResponseData = z.array(EvalRunSchema);
+export const EvalRunListResponseSchema = APIResponseSchema(EvalRunListResponseData);
 
 export type EvalRunListItem = z.infer<typeof EvalRunSchema>;
 export type EvalRunListRequest = {

package/src/api/index.ts

@@ -9,5 +9,6 @@ export * from './queue';
 export * from './region';
 export * from './sandbox';
 export * from './session';
+export * from './stream';
 export * from './thread';
 export * from './user';

package/src/api/machine/machine.ts

@@ -5,7 +5,7 @@ import { MachineResponseError } from './util';
 // TODO: The old /cli/auth/machine/* endpoints should be aliased to redirect
 // to /cli/auth/org/* in the backend (app repo). Remove aliases in follow-up PR.
 
-const MachineSchema = z.object({
+export const MachineSchema = z.object({
 	id: z.string(),
 	instanceId: z.string().nullable().optional(),
 	privateIPv4: z.string().nullable().optional(),
@@ -28,9 +28,9 @@ const MachineSchema = z.object({
 	metadata: z.record(z.string(), z.unknown()).nullable().optional(),
 });
 
-const MachineListResponseSchema = APIResponseSchema(z.array(MachineSchema));
-const MachineGetResponseSchema = APIResponseSchema(MachineSchema);
-const MachineDeleteResponseSchema = APIResponseSchemaNoData();
+export const MachineListResponseSchema = APIResponseSchema(z.array(MachineSchema));
+export const MachineGetResponseSchema = APIResponseSchema(MachineSchema);
+export const MachineDeleteResponseSchema = APIResponseSchemaNoData();
 
 export type Machine = z.infer<typeof MachineSchema>;
 
@@ -57,20 +57,24 @@ export async function machineDelete(client: APIClient, machineId: string): Promi
 	}
 }
 
-const MachineDeploymentProjectSchema = z.object({
+export const MachineDeploymentProjectSchema = z.object({
 	id: z.string(),
 	name: z.string(),
 	identifier: z.string(),
 	domains: z.array(z.string()),
 });
 
-const MachineDeploymentResourcesSchema = z.object({
+export type MachineDeploymentProject = z.infer<typeof MachineDeploymentProjectSchema>;
+
+export const MachineDeploymentResourcesSchema = z.object({
 	cpuUnits: z.number(),
 	memoryUnits: z.number(),
 	diskUnits: z.number(),
 });
 
-const MachineDeploymentSchema = z.object({
+export type MachineDeploymentResources = z.infer<typeof MachineDeploymentResourcesSchema>;
+
+export const MachineDeploymentSchema = z.object({
 	id: z.string(),
 	identifier: z.string().optional(),
 	state: z.string().optional(),
@@ -82,7 +86,7 @@ const MachineDeploymentSchema = z.object({
 	domainSuffix: z.string(),
 });
 
-const MachineDeploymentsResponseSchema = APIResponseSchema(z.array(MachineDeploymentSchema));
+export const MachineDeploymentsResponseSchema = APIResponseSchema(z.array(MachineDeploymentSchema));
 
 export type MachineDeployment = z.infer<typeof MachineDeploymentSchema>;
 
@@ -100,7 +104,7 @@ export async function machineDeployments(
 	throw new MachineResponseError({ message: resp.message });
 }
 
-const OrgAuthEnrollResponseSchema = APIResponseSchema(
+export const OrgAuthEnrollResponseSchema = APIResponseSchema(
 	z.object({
 		orgId: z.string(),
 	})
@@ -122,7 +126,7 @@ export async function orgAuthEnroll(
 	throw new MachineResponseError({ message: resp.message });
 }
 
-const OrgAuthStatusResponseSchema = APIResponseSchema(
+export const OrgAuthStatusResponseSchema = APIResponseSchema(
 	z.object({
 		publicKey: z.string().nullable(),
 	})
@@ -139,7 +143,7 @@ export async function orgAuthStatus(
 	throw new MachineResponseError({ message: resp.message });
 }
 
-const OrgAuthUnenrollResponseSchema = APIResponseSchemaNoData();
+export const OrgAuthUnenrollResponseSchema = APIResponseSchemaNoData();
 
 export async function orgAuthUnenroll(client: APIClient, orgId: string): Promise<void> {
 	const resp = await client.delete(

package/src/api/org/env-delete.ts

@@ -2,16 +2,16 @@ import { z } from 'zod';
 import { APIClient, APIResponseSchemaNoData } from '../api';
 import { OrgResponseError } from './util';
 
-const _OrgEnvDeleteRequestSchema = z.object({
+export const OrgEnvDeleteRequestSchema = z.object({
 	id: z.string().describe('the organization id'),
 	env: z.array(z.string()).optional().describe('environment variable keys to delete'),
 	secrets: z.array(z.string()).optional().describe('secret keys to delete'),
 });
 
-const OrgEnvDeleteResponseSchema = APIResponseSchemaNoData();
+export const OrgEnvDeleteResponseSchema = APIResponseSchemaNoData();
 
-type OrgEnvDeleteRequest = z.infer<typeof _OrgEnvDeleteRequestSchema>;
-type OrgEnvDeleteResponse = z.infer<typeof OrgEnvDeleteResponseSchema>;
+export type OrgEnvDeleteRequest = z.infer<typeof OrgEnvDeleteRequestSchema>;
+export type OrgEnvDeleteResponse = z.infer<typeof OrgEnvDeleteResponseSchema>;
 
 /**
  * Delete environment variables and/or secrets from an organization.

package/src/api/org/env-get.ts

@@ -2,19 +2,19 @@ import { z } from 'zod';
 import { APIClient, APIResponseSchema } from '../api';
 import { OrgResponseError } from './util';
 
-const OrgEnvDataSchema = z.object({
+export const OrgEnvDataSchema = z.object({
 	id: z.string().describe('the organization id'),
 	env: z.record(z.string(), z.string()).optional().describe('environment variables'),
 	secrets: z.record(z.string(), z.string()).optional().describe('secrets (may be masked)'),
 });
 
-const OrgEnvGetResponseSchema = APIResponseSchema(OrgEnvDataSchema);
+export const OrgEnvGetResponseSchema = APIResponseSchema(OrgEnvDataSchema);
 
 interface OrgEnvGetRequest {
 	id: string;
 	mask?: boolean;
 }
-type OrgEnvGetResponse = z.infer<typeof OrgEnvGetResponseSchema>;
+export type OrgEnvGetResponse = z.infer<typeof OrgEnvGetResponseSchema>;
 
 export type OrgEnv = z.infer<typeof OrgEnvDataSchema>;
 

package/src/api/org/env-update.ts

@@ -2,16 +2,16 @@ import { z } from 'zod';
 import { APIClient, APIResponseSchemaNoData } from '../api';
 import { OrgResponseError } from './util';
 
-const _OrgEnvUpdateRequestSchema = z.object({
+export const OrgEnvUpdateRequestSchema = z.object({
 	id: z.string().describe('the organization id'),
 	env: z.record(z.string(), z.string()).optional().describe('environment variables to set/update'),
 	secrets: z.record(z.string(), z.string()).optional().describe('secrets to set/update'),
 });
 
-const OrgEnvUpdateResponseSchema = APIResponseSchemaNoData();
+export const OrgEnvUpdateResponseSchema = APIResponseSchemaNoData();
 
-type OrgEnvUpdateRequest = z.infer<typeof _OrgEnvUpdateRequestSchema>;
-type OrgEnvUpdateResponse = z.infer<typeof OrgEnvUpdateResponseSchema>;
+export type OrgEnvUpdateRequest = z.infer<typeof OrgEnvUpdateRequestSchema>;
+export type OrgEnvUpdateResponse = z.infer<typeof OrgEnvUpdateResponseSchema>;
 
 /**
  * Update environment variables and/or secrets for an organization.

package/src/api/org/list.ts

@@ -2,13 +2,13 @@ import { z } from 'zod';
 import { APIResponseSchema, APIClient } from '../api';
 import { OrgResponseError } from './util';
 
-const ListOrganizationsResponse = z.array(
+export const ListOrganizationsResponse = z.array(
 	z.object({
 		id: z.string().describe('the unique id for the organization'),
 		name: z.string().describe('the name of the organization'),
 	})
 );
-const ListOrganizationsResponseSchema = APIResponseSchema(ListOrganizationsResponse);
+export const ListOrganizationsResponseSchema = APIResponseSchema(ListOrganizationsResponse);
 
 export type ListOrganizationsResponse = z.infer<typeof ListOrganizationsResponseSchema>;
 export type OrganizationList = z.infer<typeof ListOrganizationsResponse>;

package/src/api/org/resources.ts

@@ -2,7 +2,7 @@ import { z } from 'zod';
 import { APIResponseSchema, APIClient } from '../api';
 import { OrgResourceResponseError } from './util';
 
-const OrgS3Resource = z.object({
+export const OrgS3Resource = z.object({
 	bucket_name: z.string().describe('the S3 bucket name'),
 	access_key: z.string().nullable().optional().describe('the S3 access key'),
 	secret_key: z.string().nullable().optional().describe('the S3 secret key'),
@@ -16,7 +16,7 @@ const OrgS3Resource = z.object({
 	description: z.string().nullable().optional().describe('optional description of the bucket'),
 });
 
-const OrgDBResource = z.object({
+export const OrgDBResource = z.object({
 	name: z.string().describe('the database name'),
 	description: z.string().nullable().optional().describe('optional description of the database'),
 	username: z.string().nullable().optional().describe('the database username'),
@@ -28,12 +28,12 @@ const OrgDBResource = z.object({
 	internal: z.boolean().describe('whether this is a system-managed database (KV/Vector/Queue)'),
 });
 
-const OrgResourceListResponse = z.object({
+export const OrgResourceListResponse = z.object({
 	s3: z.array(OrgS3Resource),
 	db: z.array(OrgDBResource),
 });
 
-const OrgResourceListResponseSchema = APIResponseSchema(OrgResourceListResponse);
+export const OrgResourceListResponseSchema = APIResponseSchema(OrgResourceListResponse);
 
 export type OrgResourceListResponse = z.infer<typeof OrgResourceListResponseSchema>;
 export type OrgResourceList = z.infer<typeof OrgResourceListResponse>;

package/src/api/project/agent.ts

@@ -3,7 +3,7 @@ import type { APIClient } from '../api';
 import { APIResponseSchema } from '../api';
 import { AgentNotFoundError, ProjectResponseError } from './util';
 
-const AgentSchema = z.object({
+export const AgentSchema = z.object({
 	id: z.string().describe('Agent ID (same as identifier)'),
 	name: z.string().describe('Agent name'),
 	description: z.string().nullable().describe('Agent description'),
@@ -28,8 +28,8 @@ const AgentSchema = z.object({
 		.describe('Associated evaluations'),
 });
 
-const AgentListResponseSchema = APIResponseSchema(z.array(AgentSchema));
-const AgentGetResponseSchema = APIResponseSchema(z.array(AgentSchema));
+export const AgentListResponseSchema = APIResponseSchema(z.array(AgentSchema));
+export const AgentGetResponseSchema = APIResponseSchema(z.array(AgentSchema));
 
 export type Agent = z.infer<typeof AgentSchema>;
 

package/src/api/project/create.ts

@@ -1,8 +1,8 @@
 import { z } from 'zod';
-import { APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api';
 import { ProjectResponseError } from './util';
 
-const CreateProjectRequestSchema = z.object({
+export const CreateProjectRequestSchema = z.object({
 	name: z.string().max(255).min(1).describe('the name of the new project'),
 	description: z.string().max(255).min(0).optional().describe('the description of the project'),
 	tags: z.array(z.string()).optional().describe('tags for the project'),
@@ -11,16 +11,16 @@ const CreateProjectRequestSchema = z.object({
 	domains: z.array(z.string()).optional().describe('the custom domains for this project'),
 });
 
-const CreateProjectResponse = z.object({
+export const NewProjectSchema = z.object({
 	id: z.string().describe('the unique id for the project'),
 	sdkKey: z.string().describe('the SDK key for the project'),
 });
 
-const CreateProjectResponseSchema = APIResponseSchema(CreateProjectResponse);
+export const CreateProjectResponseSchema = APIResponseSchema(NewProjectSchema);
 
 export type CreateProjectRequest = z.infer<typeof CreateProjectRequestSchema>;
 export type CreateProjectResponse = z.infer<typeof CreateProjectResponseSchema>;
-export type NewProject = z.infer<typeof CreateProjectResponse>;
+export type NewProject = z.infer<typeof NewProjectSchema>;
 
 /**
  * Create a new Project

package/src/api/project/delete.ts

@@ -1,9 +1,9 @@
 import { z } from 'zod';
-import { APIClient, APIResponseSchema } from '../api';
+import { type APIClient, APIResponseSchema } from '../api';
 import { ProjectResponseError } from './util';
 
-const ProjectDeleteRequestSchema = z.object({ ids: z.array(z.string()) });
-const ProjectDeleteResponseSchema = APIResponseSchema(z.array(z.string()));
+export const ProjectDeleteRequestSchema = z.object({ ids: z.array(z.string()) });
+export const ProjectDeleteResponseSchema = APIResponseSchema(z.array(z.string()));
 
 type ProjectDeleteRequest = z.infer<typeof ProjectDeleteRequestSchema>;
 type ProjectDeleteResponse = z.infer<typeof ProjectDeleteResponseSchema>;