Skip to content

Commit 6282aa1

Browse files
committed
Update test success message
1 parent 9f137ee commit 6282aa1

2 files changed

Lines changed: 11 additions & 12 deletions

File tree

Core/Sources/HostApp/AccountSettings/EmbeddingModelManagement/EmbeddingModelEdit.swift

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -84,14 +84,13 @@ struct EmbeddingModelEdit: ReducerProtocol {
8484
)
8585
return .run { send in
8686
do {
87-
let tokenUsage =
88-
try await EmbeddingService(
89-
configuration: UserPreferenceEmbeddingConfiguration()
90-
.overriding {
91-
$0.model = model
92-
}
93-
).embed(text: "Hello").usage.total_tokens
94-
await send(.testSucceeded("Used \(tokenUsage) tokens."))
87+
_ = try await EmbeddingService(
88+
configuration: UserPreferenceEmbeddingConfiguration()
89+
.overriding {
90+
$0.model = model
91+
}
92+
).embed(text: "Hello")
93+
await send(.testSucceeded("Succeeded!"))
9594
} catch {
9695
await send(.testFailed(error.localizedDescription))
9796
}
@@ -156,7 +155,7 @@ extension EmbeddingModelEdit.State {
156155
format: model.format,
157156
maxTokens: model.info.maxTokens,
158157
modelName: model.info.modelName,
159-
ollamaKeepAlive: model.info.ollamaKeepAlive,
158+
ollamaKeepAlive: model.info.ollamaKeepAlive,
160159
apiKeySelection: .init(
161160
apiKeyName: model.info.apiKeyName,
162161
apiKeyManagement: .init(availableAPIKeyNames: [model.info.apiKeyName])

Tool/Sources/OpenAIService/APIs/OllamaEmbeddingService.swift

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,9 @@ struct OllamaEmbeddingService: EmbeddingAPI {
4444
#if DEBUG
4545
Logger.service.info("""
4646
Embedding usage
47-
- number of strings: \(text.count)
47+
- number of strings: 1
4848
- prompt tokens: N/A
49-
- total tokens: \(embeddingResponse.embedding.count)
49+
- total tokens: N/A
5050
5151
""")
5252
#endif
@@ -57,7 +57,7 @@ struct OllamaEmbeddingService: EmbeddingAPI {
5757
object: model.info.modelName
5858
)],
5959
model: model.info.modelName,
60-
usage: .init(prompt_tokens: 0, total_tokens: embeddingResponse.embedding.count)
60+
usage: .init(prompt_tokens: 0, total_tokens: 0)
6161
)
6262
}
6363

0 commit comments

Comments
 (0)