Pi-Multi-Agent/models.config.ts at main · jwangkun/Pi-Multi-Agent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
// models.config.ts — Multi-model provider configuration for Pi-Multi-Agent
// Copy this file and fill in your API keys, then run with:
//   DEEPSEEK_API_KEY=sk-... npx tsx server/index.ts

import type { ModelProvidersConfig } from './src/models/config.js';

export const exampleModelProvidersConfig: ModelProvidersConfig = {
  providers: [
    {
      id: 'deepseek',
      displayName: 'DeepSeek',
      baseURL: 'https://api.deepseek.com',
      apiKey: process.env['DEEPSEEK_API_KEY'] ?? '',
      isDefault: true,
    },
    // Uncomment and configure to add more providers:
    // {
    //   id: 'openai',
    //   displayName: 'OpenAI',
    //   baseURL: 'https://api.openai.com',
    //   apiKey: process.env['OPENAI_API_KEY'] ?? '',
    // },
    // {
    //   id: 'anthropic',
    //   displayName: 'Anthropic',
    //   baseURL: 'https://api.anthropic.com',
    //   apiKey: process.env['ANTHROPIC_API_KEY'] ?? '',
    // },
  ],
  models: [
    // DeepSeek models (small/light model — good for simple tasks, chat, planning)
    {
      id: 'deepseek-chat',
      provider: 'deepseek',
      displayName: 'DeepSeek Chat',
      complexity: 'light',
      specialties: ['chat', 'general', 'planning'],
      tags: ['tools'],
      contextWindow: 64000,
      maxOutputTokens: 4096,
      pricingPer1kInput: 0.0014,
      pricingPer1kOutput: 0.0028,
    },
    // DeepSeek Reasoner (heavy model — good for reasoning, evaluation, complex analysis)
    {
      id: 'deepseek-reasoner',
      provider: 'deepseek',
      displayName: 'DeepSeek Reasoner',
      complexity: 'heavy',
      specialties: ['reasoning', 'analysis'],
      contextWindow: 64000,
      maxOutputTokens: 4096,
      pricingPer1kInput: 0.004,
      pricingPer1kOutput: 0.016,
    },
    // Uncomment to add more models:
    // {
    //   id: 'gpt-4o',
    //   provider: 'openai',
    //   displayName: 'GPT-4o',
    //   complexity: 'heavy',
    //   specialties: ['chat', 'reasoning', 'vision', 'writing'],
    //   tags: ['tools'],
    //   contextWindow: 128000,
    //   maxOutputTokens: 16384,
    //   pricingPer1kInput: 0.005,
    //   pricingPer1kOutput: 0.015,
    // },
    // {
    //   id: 'gpt-4o-mini',
    //   provider: 'openai',
    //   displayName: 'GPT-4o Mini',
    //   complexity: 'light',
    //   specialties: ['chat'],
    //   tags: ['tools'],
    //   contextWindow: 128000,
    //   maxOutputTokens: 16384,
    //   pricingPer1kInput: 0.00015,
    //   pricingPer1kOutput: 0.0006,
    // },
  ],
};