File tree Expand file tree Collapse file tree
Core/Sources/HostApp/AccountSettings/EmbeddingModelManagement
Tool/Sources/OpenAIService/APIs Expand file tree Collapse file tree Original file line number Diff line number Diff 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] )
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments