Skip to content

Commit 3ee97af

Browse files
feat!: drop unused getEmbeddingContextSize function
1 parent 9433142 commit 3ee97af

File tree

1 file changed

+4
-14
lines changed

1 file changed

+4
-14
lines changed

src/index.ts

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -20,18 +20,15 @@ const modelContextSizeMap = new Map<ModelName, number>([
2020
* Resolve a model name to a canonical model name.
2121
*/
2222
export 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-
5343
const WHITESPACE_RE = /^\s+$/
5444
const 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]/
5545
const NUMERIC_SEQUENCE_RE = /^\d+(?:[.,]\d+)?$/

0 commit comments

Comments
 (0)