Skip to content

Commit a210078

Browse files
committed
feat: add OpenAI models and logo integration
1 parent f624850 commit a210078

File tree

2 files changed

+125
-26
lines changed

2 files changed

+125
-26
lines changed

src/utilities/modelSelection.ts

Lines changed: 120 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -6,22 +6,78 @@ export namespace MockModelSelection {
66
export const fastModel: string = 'ClaudeSonnet';
77

88
export const models: LanguageModels = {
9-
// Gpt4: {
10-
// name: 'GPT-4',
11-
// contextLength: 8192,
12-
// temperature: 0.2,
13-
// provider: {
14-
// type: 'codestory',
15-
// },
16-
// },
17-
// DeepSeekCoder33BInstruct: {
18-
// name: 'DeepSeek Coder 33B Instruct',
19-
// contextLength: 16384,
20-
// temperature: 0.2,
21-
// provider: {
22-
// type: 'codestory',
23-
// },
24-
// },
9+
// 'gpt-4': {
10+
// name: 'GPT-4',
11+
// contextLength: 8192,
12+
// temperature: 0.2,
13+
// provider: {
14+
// type: 'codestory',
15+
// },
16+
// },
17+
// 'DeepSeekCoder33BInstruct': {
18+
// name: 'DeepSeek Coder 33B Instruct',
19+
// contextLength: 16384,
20+
// temperature: 0.2,
21+
// provider: {
22+
// type: 'codestory',
23+
// },
24+
//},
25+
'gpt-4': {
26+
name: 'GPT-4',
27+
contextLength: 8192,
28+
temperature: 0.2,
29+
provider: {
30+
type: 'openai',
31+
},
32+
},
33+
'gpt-4-turbo-preview': {
34+
name: 'GPT-4 Turbo',
35+
contextLength: 128000,
36+
temperature: 0.2,
37+
provider: {
38+
type: 'openai',
39+
},
40+
},
41+
'gpt-3.5-turbo': {
42+
name: 'GPT-3.5 Turbo',
43+
contextLength: 16385,
44+
temperature: 0.2,
45+
provider: {
46+
type: 'openai',
47+
},
48+
},
49+
'gpt-4o': {
50+
name: 'GPT-4o',
51+
contextLength: 128000,
52+
temperature: 0.2,
53+
provider: {
54+
type: 'openai',
55+
},
56+
},
57+
'gpt-4o-mini': {
58+
name: 'GPT-4o Mini',
59+
contextLength: 16385,
60+
temperature: 0.2,
61+
provider: {
62+
type: 'openai',
63+
},
64+
},
65+
'o1-preview': {
66+
name: 'O1 Preview',
67+
contextLength: 128000,
68+
temperature: 0.2,
69+
provider: {
70+
type: 'openai',
71+
},
72+
},
73+
'o1-mini': {
74+
name: 'O1 Mini',
75+
contextLength: 16385,
76+
temperature: 0.2,
77+
provider: {
78+
type: 'openai',
79+
},
80+
},
2581
ClaudeSonnet: {
2682
name: 'Claude Sonnet',
2783
contextLength: 200000,
@@ -38,18 +94,61 @@ export namespace MockModelSelection {
3894
type: 'anthropic',
3995
},
4096
},
97+
'anthropic.claude-3-sonnet-20240229-v1:0': {
98+
name: 'Claude 3 Sonnet (Bedrock)',
99+
contextLength: 200000,
100+
temperature: 0.2,
101+
provider: {
102+
type: 'aws-bedrock',
103+
},
104+
},
105+
'anthropic.claude-3-haiku-20240307-v1:0': {
106+
name: 'Claude 3 Haiku (Bedrock)',
107+
contextLength: 200000,
108+
temperature: 0.2,
109+
provider: {
110+
type: 'aws-bedrock',
111+
},
112+
},
113+
'anthropic.claude-v2:1': {
114+
name: 'Claude 2 (Bedrock)',
115+
contextLength: 100000,
116+
temperature: 0.2,
117+
provider: {
118+
type: 'aws-bedrock',
119+
},
120+
},
121+
'amazon.titan-text-express-v1': {
122+
name: 'Titan Text Express',
123+
contextLength: 8000,
124+
temperature: 0.2,
125+
provider: {
126+
type: 'aws-bedrock',
127+
},
128+
},
129+
'meta.llama2-70b-chat-v1': {
130+
name: 'Llama 2 70B',
131+
contextLength: 4096,
132+
temperature: 0.2,
133+
provider: {
134+
type: 'aws-bedrock',
135+
},
136+
},
41137
};
42138

43139
export const providers: ModelProviders = {
44-
//"codestory": {
45-
// name: "CodeStory"
140+
//codestory: {
141+
// name: "CodeStory"
46142
//},
47-
//"ollama": {
48-
// name: "Ollama"
49-
//}
143+
openai: {
144+
name: 'OpenAI',
145+
},
50146
anthropic: {
51147
name: 'Anthropic',
52148
},
149+
'aws-bedrock': {
150+
name: 'AWS Bedrock',
151+
},
53152
'open-router': {
54153
name: 'Open Router',
55154
},

src/webviews/components/preset.tsx

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
import { Provider, ProviderType } from '../../model';
22
import AnthropicLogo from '../assets/providers-logos/anthropic.svg';
3-
// import AWSBedrockLogo from '../assets/providers-logos/aws-bedrock.svg';
3+
import AWSBedrockLogo from '../assets/providers-logos/aws-bedrock.svg';
44
// import GeminiLogo from '../assets/providers-logos/gemini.svg';
55
// import OllamaLogo from '../assets/providers-logos/ollama.svg';
6-
// import OpenAILogo from '../assets/providers-logos/openai.svg';
6+
import OpenAILogo from '../assets/providers-logos/openai.svg';
77
import OpenRouterLogo from '../assets/providers-logos/open-router.svg';
88
import { cn } from 'utils/cn';
99

1010
const logoMap = new Map<string, React.FunctionComponent<React.SVGProps<SVGSVGElement>>>();
1111
logoMap.set(Provider.Anthropic, AnthropicLogo);
12-
// logoMap.set(Provider.OpenAI, OpenAILogo);
12+
logoMap.set(Provider.OpenAI, OpenAILogo);
1313
logoMap.set(Provider.OpenRouter, OpenRouterLogo);
1414
// logoMap.set(Provider.GoogleGemini, GeminiLogo);
15-
// logoMap.set(Provider.AWSBedrock, AWSBedrockLogo);
16-
// logoMap.set(Provider.OpenAICompatible, OpenAILogo);
15+
logoMap.set(Provider.AWSBedrock, AWSBedrockLogo);
16+
logoMap.set(Provider.OpenAICompatible, OpenAILogo);
1717
// logoMap.set(Provider.Ollama, OllamaLogo);
1818

1919
export type PresetLogoProps = React.SVGProps<SVGSVGElement> & {

0 commit comments

Comments
 (0)