Skip to content

Commit 507efb0

Browse files
committed
Fix embedding configuration
1 parent cd153f5 commit 507efb0

3 files changed

Lines changed: 3 additions & 5 deletions

File tree

Tool/Sources/OpenAIService/Configuration/EmbeddingConfiguration.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ extension EmbeddingConfiguration {
1717
case .openAI:
1818
let baseURL = UserDefaults.shared.value(for: \.openAIBaseURL)
1919
if baseURL.isEmpty { return "https://api.openai.com/v1/embeddings" }
20-
return "\(baseURL)/v1/chat/completions"
20+
return "\(baseURL)/v1/embeddings"
2121
case .azureOpenAI:
2222
let baseURL = UserDefaults.shared.value(for: \.azureOpenAIBaseURL)
2323
let deployment = UserDefaults.shared.value(for: \.azureChatGPTDeployment)

Tool/Sources/OpenAIService/Configuration/UserPreferenceEmbeddingConfiguration.swift

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,7 @@ public struct UserPreferenceEmbeddingConfiguration: EmbeddingConfiguration {
77
}
88

99
public var model: String {
10-
let value = UserDefaults.shared.value(for: \.chatGPTModel)
11-
if value.isEmpty { return "text-embedding-ada-002" }
12-
return value
10+
return "text-embedding-ada-002"
1311
}
1412

1513
public var endpoint: String {

Tool/Sources/OpenAIService/EmbeddingService.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ import Foundation
22

33
public struct EmbeddingResponse: Decodable {
44
public struct Object: Decodable {
5-
public var embeddings: [Float]
5+
public var embedding: [Float]
66
public var index: Int
77
public var object: String
88
}

0 commit comments

Comments
 (0)