File tree Expand file tree Collapse file tree 1 file changed +4
-14
lines changed Expand file tree Collapse file tree 1 file changed +4
-14
lines changed Original file line number Diff line number Diff line change @@ -20,18 +20,15 @@ const modelContextSizeMap = new Map<ModelName, number>([
2020 * Resolve a model name to a canonical model name.
2121 */
2222export function resolveModelName ( modelName : string ) : ModelName {
23- if ( modelName . startsWith ( 'gpt-3.5-turbo-16k' ) )
24- return 'gpt-3.5-turbo-16k'
25-
26- if ( modelName . startsWith ( 'gpt-3.5-turbo-' ) )
23+ if ( modelName . startsWith ( 'gpt-3.5-' ) )
2724 return 'gpt-3.5-turbo'
2825
29- if ( modelName . startsWith ( 'gpt-4-32k' ) )
30- return 'gpt-4-32k'
31-
3226 if ( modelName . startsWith ( 'gpt-4-' ) )
3327 return 'gpt-4'
3428
29+ if ( modelName . startsWith ( 'gpt-4o' ) )
30+ return 'gpt-4'
31+
3532 return modelName as ModelName
3633}
3734
@@ -43,13 +40,6 @@ export function getModelContextSize(modelName: string): number {
4340 return modelContextSizeMap . get ( modelKey ) ?? 4097
4441}
4542
46- export function getEmbeddingContextSize ( modelName ?: string ) : number {
47- if ( modelName === 'text-embedding-ada-002' )
48- return 8191
49-
50- return 2046
51- }
52-
5343const WHITESPACE_RE = / ^ \s + $ /
5444const CJK_RE = / [ \u4E00 - \u9FFF \u3400 - \u4DBF \u3000 - \u303F \uFF00 - \uFFEF \u30A0 - \u30FF \u2E80 - \u2EFF \u31C0 - \u31EF \u3200 - \u32FF \u3300 - \u33FF \uAC00 - \uD7AF \u1100 - \u11FF \u3130 - \u318F \uA960 - \uA97F \uD7B0 - \uD7FF ] /
5545const NUMERIC_SEQUENCE_RE = / ^ \d + (?: [ . , ] \d + ) ? $ /
You can’t perform that action at this time.
0 commit comments