provider: meshy.ai provider

Author: ballpointcarrot

src/globals.ts

@@ -102,6 +102,7 @@ export const FEATHERLESS_AI: string = 'featherless-ai';
 export const KRUTRIM: string = 'krutrim';
 export const QDRANT: string = 'qdrant';
 export const THREE_ZERO_TWO_AI: string = '302ai';
+export const MESHY: string = 'meshy';
 
 export const VALID_PROVIDERS = [
   ANTHROPIC,
@@ -167,6 +168,7 @@ export const VALID_PROVIDERS = [
   KRUTRIM,
   QDRANT,
   THREE_ZERO_TWO_AI,
+  MESHY,
 ];
 
 export const CONTENT_TYPES = {

src/providers/index.ts

@@ -63,6 +63,7 @@ import HyperbolicConfig from './hyperbolic';
 import { FeatherlessAIConfig } from './featherless-ai';
 import KrutrimConfig from './krutrim';
 import AI302Config from './302ai';
+import MeshyConfig from './meshy';
 
 const Providers: { [key: string]: ProviderConfigs } = {
   openai: OpenAIConfig,
@@ -126,6 +127,7 @@ const Providers: { [key: string]: ProviderConfigs } = {
   'featherless-ai': FeatherlessAIConfig,
   krutrim: KrutrimConfig,
   '302ai': AI302Config,
+  meshy: MeshyConfig,
 };
 
 export default Providers;

src/providers/meshy/api.ts

@@ -0,0 +1,26 @@
+import { ProviderAPIConfig } from '../types';
+
+const MeshyAPIConfig: ProviderAPIConfig = {
+  getBaseURL: ({ gatewayRequestURL }) => {
+    const version = gatewayRequestURL.includes('text-to-3d') ? 'v2' : 'v1';
+    return `https://api.meshy.ai/openapi/${version}`;
+  },
+  headers: ({ providerOptions }) => {
+    return {
+      Authorization: `Bearer ${providerOptions.apiKey}`,
+      'Content-Type': 'application/json',
+    };
+  },
+  getEndpoint: ({ fn, gatewayRequestURL }) => {
+    const basePath = gatewayRequestURL.split('/v1')?.[1];
+
+    switch (fn) {
+      case 'modelGenerate':
+        return '/text-to-3d';
+      default:
+        return basePath || '';
+    }
+  },
+};
+
+export default MeshyAPIConfig;

src/providers/meshy/index.ts

@@ -0,0 +1,16 @@
+import { ProviderConfigs } from '../types';
+import MeshyAPIConfig from './api';
+import {
+  MeshyModelGenerateConfig,
+  MeshyModelGenerateResponseTransform,
+} from './modelGenerate';
+
+const MeshyConfig: ProviderConfigs = {
+  modelGenerate: MeshyModelGenerateConfig,
+  api: MeshyAPIConfig,
+  responseTransforms: {
+    modelGenerate: MeshyModelGenerateResponseTransform,
+  },
+};
+
+export default MeshyConfig;

src/providers/meshy/modelGenerate.ts

@@ -0,0 +1,55 @@
+import { MESHY } from '../../globals';
+import { ErrorResponse, ProviderConfig } from '../types';
+import { generateInvalidProviderResponseError } from '../utils';
+
+export const MeshyModelGenerateConfig: ProviderConfig = {
+  prompt: {
+    param: 'prompt',
+    required: true,
+  },
+  negative_prompt: {
+    param: 'negative_prompt',
+  },
+  art_style: {
+    param: 'art_style',
+  },
+  mode: {
+    param: 'mode',
+    default: 'preview',
+  },
+  seed: {
+    param: 'seed',
+  },
+};
+
+interface MeshyModelGenerateResponse {
+  result: string;
+  id?: string;
+  status?: string;
+  created_at?: string;
+  expires_at?: string;
+}
+
+export const MeshyModelGenerateResponseTransform: (
+  response: MeshyModelGenerateResponse | ErrorResponse,
+  responseStatus: number
+) => MeshyModelGenerateResponse | ErrorResponse = (
+  response,
+  responseStatus
+) => {
+  if (responseStatus !== 200) {
+    return generateInvalidProviderResponseError(response, MESHY);
+  }
+
+  if ('result' in response) {
+    return {
+      result: response.result,
+      id: response.id,
+      status: response.status,
+      created_at: response.created_at,
+      expires_at: response.expires_at,
+    };
+  }
+
+  return generateInvalidProviderResponseError(response, MESHY);
+};

src/providers/types.ts

@@ -105,7 +105,8 @@ export type endpointStrings =
   | 'getModelResponse'
   | 'deleteModelResponse'
   | 'listResponseInputItems'
-  | 'messages';
+  | 'messages'
+  | 'modelGenerate';
 
 /**
  * A collection of API configurations for multiple AI providers.