Skip to content

Commit 802fa30

Browse files
committed
Add OpenAI embedding models
1 parent 626aa5e commit 802fa30

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

Tool/Sources/Preferences/Types/OpenAIEmbeddingModel.swift

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
11
public enum OpenAIEmbeddingModel: String, CaseIterable {
2+
case textEmbedding3Small = "text-embedding-3-small"
3+
case textEmbedding3Large = "text-embedding-3-large"
24
case textEmbeddingAda002 = "text-embedding-ada-002"
35
}
46

@@ -7,6 +9,10 @@ public extension OpenAIEmbeddingModel {
79
switch self {
810
case .textEmbeddingAda002:
911
return 8191
12+
case .textEmbedding3Small:
13+
return 8191
14+
case .textEmbedding3Large:
15+
return 8191
1016
}
1117
}
1218
}

0 commit comments

Comments
 (0)