Upstage.ai

src/globals.ts

@@ -69,6 +69,7 @@ export const SILICONFLOW: string = 'siliconflow';
 export const CEREBRAS: string = 'cerebras';
 export const INFERENCENET: string = 'inference-net';
 export const SAMBANOVA: string = 'sambanova';
+export const UPSTAGE: string = 'upstage';
 
 export const VALID_PROVIDERS = [
   ANTHROPIC,
@@ -112,6 +113,7 @@ export const VALID_PROVIDERS = [
   CEREBRAS,
   INFERENCENET,
   SAMBANOVA,
+  UPSTAGE,
 ];
 
 export const CONTENT_TYPES = {

src/providers/index.ts

@@ -42,6 +42,7 @@ import HuggingfaceConfig from './huggingface';
 import { cerebrasProviderAPIConfig } from './cerebras';
 import { InferenceNetProviderConfigs } from './inference-net';
 import SambaNovaConfig from './sambanova';
+import { UpstageProviderConfig } from './upstage';
 
 const Providers: { [key: string]: ProviderConfigs } = {
   openai: OpenAIConfig,
@@ -85,6 +86,7 @@ const Providers: { [key: string]: ProviderConfigs } = {
   cerebras: cerebrasProviderAPIConfig,
   'inference-net': InferenceNetProviderConfigs,
   sambanova: SambaNovaConfig,
+  upstage: UpstageProviderConfig,
 };
 
 export default Providers;

src/providers/upstage/api.ts

@@ -0,0 +1,19 @@
+import { ProviderAPIConfig } from '../types';
+
+export const upstageAPIConfig: ProviderAPIConfig = {
+  getBaseURL: () => 'https://api.upstage.ai/v1/solar',
+  headers({ providerOptions }) {
+    const { apiKey } = providerOptions;
+    return { Authorization: `Bearer ${apiKey}` };
+  },
+  getEndpoint({ fn }) {
+    switch (fn) {
+      case 'chatComplete':
+        return `/chat/completions`;
+      case 'embed':
+        return `/embeddings`;
+      default:
+        return '';
+    }
+  },
+};

src/providers/upstage/index.ts

@@ -0,0 +1,18 @@
+import { UPSTAGE } from '../../globals';
+import {
+  chatCompleteParams,
+  embedParams,
+  responseTransformers,
+} from '../open-ai-base';
+import { ProviderConfigs } from '../types';
+import { upstageAPIConfig } from './api';
+
+export const UpstageProviderConfig: ProviderConfigs = {
+  chatComplete: chatCompleteParams([], { model: 'solar-pro' }),
+  embed: embedParams([], { model: 'solar-embedding-1-large-query' }),
+  api: upstageAPIConfig,
+  responseTransforms: responseTransformers(UPSTAGE, {
+    chatComplete: true,
+    embed: true,
+  }),
+};