Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
1 change: 1 addition & 0 deletions src/shared/model-registry/provider-mapping.ts
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ export const PROVIDER_ID_MAP: Record<string, string> = {
'mistral-ai': 'mistral',
perplexity: 'perplexity',
openrouter: 'openrouter',
trustedrouter: 'trustedrouter',
minimax: 'minimax',
'minimax-cn': 'minimax-cn',
moonshot: 'moonshotai',
Expand Down
6 changes: 6 additions & 0 deletions src/shared/models/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@ export const aiProviderNameHash: Record<ModelProviderEnum, string> = {
[ModelProviderEnum.Perplexity]: 'Perplexity API',
[ModelProviderEnum.XAI]: 'xAI API',
[ModelProviderEnum.OpenRouter]: 'OpenRouter API',
[ModelProviderEnum.TrustedRouter]: 'TrustedRouter API',
[ModelProviderEnum.Bedrock]: 'AWS Bedrock',
[ModelProviderEnum.VercelAIGateway]: 'Vercel AI Gateway',
[ModelProviderEnum.Custom]: 'Custom Provider',
Expand Down Expand Up @@ -115,6 +116,11 @@ export const AIModelProviderMenuOptionList = [
label: aiProviderNameHash[ModelProviderEnum.OpenRouter],
disabled: false,
},
{
value: ModelProviderEnum.TrustedRouter,
label: aiProviderNameHash[ModelProviderEnum.TrustedRouter],
disabled: false,
},
{
value: ModelProviderEnum.MistralAI,
label: aiProviderNameHash[ModelProviderEnum.MistralAI],
Expand Down
28 changes: 28 additions & 0 deletions src/shared/providers/definitions/models/trustedrouter.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
import OpenAICompatible, { type OpenAICompatibleSettings } from '../../../models/openai-compatible'
import type { ModelDependencies } from '../../../types/adapters'

interface Options extends OpenAICompatibleSettings {}

export default class TrustedRouter extends OpenAICompatible {
public name = 'TrustedRouter'
public options: Options
constructor(options: Omit<Options, 'apiHost'>, dependencies: ModelDependencies) {
const apiHost = 'https://api.trustedrouter.com/v1'
super(
{
apiKey: options.apiKey,
apiHost,
model: options.model,
temperature: options.temperature,
topP: options.topP,
maxOutputTokens: options.maxOutputTokens,
stream: options.stream,
},
dependencies
)
this.options = {
...options,
apiHost,
}
}
}
59 changes: 59 additions & 0 deletions src/shared/providers/definitions/trustedrouter.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
import { ModelProviderEnum, ModelProviderType } from '../../types'
import { defineProvider } from '../registry'
import TrustedRouter from './models/trustedrouter'

export const trustedRouterProvider = defineProvider({
id: ModelProviderEnum.TrustedRouter,
name: 'TrustedRouter',
type: ModelProviderType.OpenAI,
modelsDevProviderId: 'trustedrouter',
curatedModelIds: ['trustedrouter/auto', 'trustedrouter/zdr', 'trustedrouter/e2e', 'trustedrouter/cheap'],
urls: {
website: 'https://trustedrouter.com/',
},
defaultSettings: {
apiHost: 'https://api.trustedrouter.com/v1',
models: [
{
modelId: 'trustedrouter/auto',
nickname: 'Auto (routes across healthy providers)',
capabilities: ['tool_use', 'vision'],
contextWindow: 128_000,
},
{
modelId: 'trustedrouter/zdr',
nickname: 'ZDR (zero-data-retention routes)',
capabilities: ['tool_use'],
contextWindow: 128_000,
},
{
modelId: 'trustedrouter/e2e',
nickname: 'E2E (end-to-end encrypted routes)',
capabilities: ['tool_use'],
contextWindow: 128_000,
},
{
modelId: 'trustedrouter/cheap',
nickname: 'Cheap (cost-optimized routes)',
capabilities: ['tool_use'],
contextWindow: 128_000,
},
],
},
createModel: (config) => {
return new TrustedRouter(
{
apiKey: config.effectiveApiKey,
model: config.model,
temperature: config.settings.temperature,
topP: config.settings.topP,
maxOutputTokens: config.settings.maxTokens,
stream: config.settings.stream,
},
config.dependencies
)
},
getDisplayName: (modelId, providerSettings) => {
return `TrustedRouter API (${providerSettings?.models?.find((m) => m.modelId === modelId)?.nickname || modelId})`
},
})
1 change: 1 addition & 0 deletions src/shared/providers/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ import './definitions/minimax'
import './definitions/moonshot'
import './definitions/siliconflow'
import './definitions/openrouter'
import './definitions/trustedrouter'
import './definitions/ollama'
import './definitions/lmstudio'
import './definitions/azure'
Expand Down
1 change: 1 addition & 0 deletions src/shared/types/provider.ts
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ export enum ModelProviderEnum {
Perplexity = 'perplexity',
XAI = 'xAI',
OpenRouter = 'openrouter',
TrustedRouter = 'trustedrouter',
Bedrock = 'bedrock',
VercelAIGateway = 'vercel-ai-gateway',
Custom = 'custom',
Expand Down