feat: add StudioByAI21Labs integration

packages/corsair/core/constants.ts

@@ -59,8 +59,8 @@ export const BaseProviders = [
 	'bugsnag',
 	'cal',
 	'calendly',
-	'canvas',
 	'canva',
+	'canvas',
 	'circleci',
 	'cloudflare',
 	'cloudinary',
@@ -137,6 +137,7 @@ export const BaseProviders = [
 	'spotify',
 	'strava',
 	'stripe',
+	'studiobyai21labs',
 	'supabase',
 	'tally',
 	'tavily',
@@ -211,8 +212,8 @@ export const ProviderDisplayNames = {
 	bugsnag: 'BugSnag',
 	cal: 'Cal',
 	calendly: 'Calendly',
-	canvas: 'Canvas LMS',
 	canva: 'Canva',
+	canvas: 'Canvas LMS',
 	circleci: 'CircleCI',
 	cloudflare: 'Cloudflare',
 	cloudinary: 'Cloudinary',
@@ -289,6 +290,7 @@ export const ProviderDisplayNames = {
 	spotify: 'Spotify',
 	strava: 'Strava',
 	stripe: 'Stripe',
+	studiobyai21labs: 'StudioByAI21Labs',
 	supabase: 'Supabase',
 	tally: 'Tally',
 	tavily: 'Tavily',
@@ -448,6 +450,7 @@ export type AllProviders =
 	| 'spotify'
 	| 'strava'
 	| 'stripe'
+	| 'studiobyai21labs'
 	| 'supabase'
 	| 'tally'
 	| 'tavily'

packages/studiobyai21labs/client.ts

@@ -0,0 +1,122 @@
+import type { ApiRequestOptions, OpenAPIConfig } from 'corsair/http';
+import { request } from 'corsair/http';
+
+export class StudioByAI21LabsAPIError extends Error {
+	constructor(
+		message: string,
+		public readonly code?: string,
+	) {
+		super(message);
+		this.name = 'StudioByAI21LabsAPIError';
+	}
+}
+
+// TODO: Update with your API base URL
+const STUDIOBYAI21LABS_API_BASE = 'https://api.ai21.com/studio/v1';
+
+export async function makeStudioByAI21LabsRequest<T>(
+	endpoint: string,
+	apiKey: string,
+	options: {
+		method?: 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH';
+		body?: Record<string, unknown>;
+		query?: Record<string, string | number | boolean | undefined>;
+	} = {},
+): Promise<T> {
+	const { method = 'GET', body, query } = options;
+
+	const config: OpenAPIConfig = {
+		BASE: STUDIOBYAI21LABS_API_BASE,
+		VERSION: '1.0.0',
+		WITH_CREDENTIALS: false,
+		CREDENTIALS: 'omit',
+		TOKEN: apiKey,
+		HEADERS: {
+			'Content-Type': 'application/json',
+			Authorization: `Bearer ${apiKey}`,
+		},
+	};
+
+	const requestOptions: ApiRequestOptions = {
+		method,
+		url: endpoint,
+		body:
+			method === 'POST' || method === 'PUT' || method === 'PATCH'
+				? body
+				: undefined,
+		mediaType: 'application/json; charset=utf-8',
+		query: method === 'GET' ? query : undefined,
+	};
+
+	try {
+		return await request<T>(config, requestOptions);
+	} catch (error) {
+		if (error instanceof Error) {
+			throw new StudioByAI21LabsAPIError(error.message);
+		}
+		throw new StudioByAI21LabsAPIError('Unknown error');
+	}
+}
+
+const buildUrl = (endpoint: string): string => {
+	const baseUrl = STUDIOBYAI21LABS_API_BASE.endsWith('/')
+		? STUDIOBYAI21LABS_API_BASE.slice(0, -1)
+		: STUDIOBYAI21LABS_API_BASE;
+	const path = endpoint.startsWith('/') ? endpoint.slice(1) : endpoint;
+	return `${baseUrl}/${path}`;
+};
+
+export async function uploadStudioByAI21LabsFile<T>(
+	endpoint: string,
+	apiKey: string,
+	options: {
+		file: Blob | string;
+		fileName: string;
+		fields?: Record<string, string | undefined>;
+	},
+): Promise<T> {
+	const { file, fileName, fields = {} } = options;
+	const blob = typeof file === 'string' ? new Blob([file]) : file;
+
+	const formData = new FormData();
+	formData.append('file', blob, fileName);
+	for (const [key, value] of Object.entries(fields)) {
+		if (value !== undefined) formData.append(key, value);
+	}
+
+	const response = await fetch(buildUrl(endpoint), {
+		method: 'POST',
+		headers: { Authorization: `Bearer ${apiKey}` },
+		body: formData,
+	});
+
+	if (!response.ok) {
+		const text = await response.text();
+		throw new StudioByAI21LabsAPIError(
+			`Upload failed: status ${response.status}; body: ${text}`,
+			undefined,
+		);
+	}
+
+	return response.json() as Promise<T>;
+}
+
+export async function downloadStudioByAI21LabsFile(
+	endpoint: string,
+	apiKey: string,
+): Promise<string> {
+	const response = await fetch(buildUrl(endpoint), {
+		method: 'GET',
+		headers: { Authorization: `Bearer ${apiKey}` },
+	});
+
+	if (!response.ok) {
+		const text = await response.text();
+		throw new StudioByAI21LabsAPIError(
+			`Download failed: status ${response.status}; body: ${text}`,
+			undefined,
+		);
+	}
+
+	return response.text();
+}

packages/studiobyai21labs/endpoints/assistants.ts

@@ -0,0 +1,189 @@
+import { makeStudioByAI21LabsRequest } from '../client';
+import type { StudioByAI21LabsEndpoints } from '../index';
+
+export const createAssistant: StudioByAI21LabsEndpoints['createAssistant'] =
+	async (ctx, input) => {
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'POST',
+			'assistants',
+			input,
+		);
+		return response as any;
+	};
+
+export const createAssistantPlan: StudioByAI21LabsEndpoints['createAssistantPlan'] =
+	async (ctx, input) => {
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'POST',
+			'assistant-plans',
+			input,
+		);
+		return response as any;
+	};
+
+export const createAssistantRoute: StudioByAI21LabsEndpoints['createAssistantRoute'] =
+	async (ctx, input) => {
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'POST',
+			'assistant-routes',
+			input,
+		);
+		return response as any;
+	};
+
+export const deleteAssistant: StudioByAI21LabsEndpoints['deleteAssistant'] =
+	async (ctx, input) => {
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'DELETE',
+			`assistants/${input.id}`,
+			input,
+		);
+		return response as any;
+	};
+
+export const deleteAssistantRoute: StudioByAI21LabsEndpoints['deleteAssistantRoute'] =
+	async (ctx, input) => {
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'DELETE',
+			`assistant-routes/${input.id}`,
+			input,
+		);
+		return response as any;
+	};
+
+export const getAssistant: StudioByAI21LabsEndpoints['getAssistant'] = async (
+	ctx,
+	input,
+) => {
+	const response = await makeStudioByAI21LabsRequest(
+		ctx,
+		'GET',
+		`assistants/${input.id}`,
+	);
+	return response as any;
+};
+
+export const getAssistantRoute: StudioByAI21LabsEndpoints['getAssistantRoute'] =
+	async (ctx, input) => {
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'GET',
+			`assistant-routes/${input.id}`,
+		);
+		return response as any;
+	};
+
+export const getAssistantsByMcp: StudioByAI21LabsEndpoints['getAssistantsByMcp'] =
+	async (ctx, input) => {
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'GET',
+			'assistants/mcp',
+			{ mcpId: input.mcpId },
+		);
+		return response as any;
+	};
+
+export const getPlan: StudioByAI21LabsEndpoints['getPlan'] = async (
+	ctx,
+	input,
+) => {
+	const response = await makeStudioByAI21LabsRequest(
+		ctx,
+		'GET',
+		`assistant-plans/${input.id}`,
+	);
+	return response as any;
+};
+
+export const listAssistants: StudioByAI21LabsEndpoints['listAssistants'] =
+	async (ctx, input) => {
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'GET',
+			'assistants',
+			input,
+		);
+		return response as any;
+	};
+
+export const listPlans: StudioByAI21LabsEndpoints['listPlans'] = async (
+	ctx,
+	input,
+) => {
+	const response = await makeStudioByAI21LabsRequest(
+		ctx,
+		'GET',
+		'assistant-plans',
+		input,
+	);
+	return response as any;
+};
+
+export const modifyAssistant: StudioByAI21LabsEndpoints['modifyAssistant'] =
+	async (ctx, input) => {
+		const { id, ...body } = input;
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'PUT',
+			`assistants/${id}`,
+			body,
+		);
+		return response as any;
+	};
+
+export const modifyAssistantPlan: StudioByAI21LabsEndpoints['modifyAssistantPlan'] =
+	async (ctx, input) => {
+		const { id, ...body } = input;
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'PUT',
+			`assistant-plans/${id}`,
+			body,
+		);
+		return response as any;
+	};
+
+export const modifyAssistantRoute: StudioByAI21LabsEndpoints['modifyAssistantRoute'] =
+	async (ctx, input) => {
+		const { id, ...body } = input;
+		const response = await makeStudioByAI21LabsRequest(
+			ctx,
+			'PUT',
+			`assistant-routes/${id}`,
+			body,
+		);
+		return response as any;
+	};
+
+export const runAssistant: StudioByAI21LabsEndpoints['runAssistant'] = async (
+	ctx,
+	input,
+) => {
+	const { id, ...body } = input;
+	const response = await makeStudioByAI21LabsRequest(
+		ctx,
+		'POST',
+		`assistants/${id}/run`,
+		body,
+	);
+	return response as any;
+};
+
+export const validatePlan: StudioByAI21LabsEndpoints['validatePlan'] = async (
+	ctx,
+	input,
+) => {
+	const response = await makeStudioByAI21LabsRequest(
+		ctx,
+		'POST',
+		'assistant-plans/validate',
+		input,
+	);
+	return response as any;
+};

packages/studiobyai21labs/endpoints/chat.ts

@@ -0,0 +1,21 @@
+import { logEventFromContext } from 'corsair/core';
+import type { StudioByAI21LabsEndpoints } from '..';
+import { makeStudioByAI21LabsRequest } from '../client';
+import type { StudioByAI21LabsEndpointOutputs } from './types';
+
+export const completions: StudioByAI21LabsEndpoints['chatCompletions'] = async (
+	ctx,
+	input,
+) => {
+	const response = await makeStudioByAI21LabsRequest<
+		StudioByAI21LabsEndpointOutputs['chatCompletions']
+	>('chat/completions', ctx.key, { method: 'POST', body: input });
+
+	await logEventFromContext(
+		ctx,
+		'studiobyai21labs.chat.completions',
+		{ ...input },
+		'completed',
+	);
+	return response;
+};