Skip to content

Commit 8c58fb6

Browse files
committed
[Fix] #174 - Reflect promptGenerator
1 parent 0dbe105 commit 8c58fb6

File tree

2 files changed

+4
-22
lines changed

2 files changed

+4
-22
lines changed

Heim/Domain/Domain/UseCase/GenerativeEmotionPromptUseCase.swift

+2-11
Original file line numberDiff line numberDiff line change
@@ -9,29 +9,20 @@ public protocol GenerativeEmotionPromptUseCase {
99
func generate(_ input: String) async throws -> String?
1010
}
1111

12-
public struct GeminiGenerativeEmotionPromptUseCase: GenerativeEmotionPromptUseCase, UserUseCase {
13-
public var userRepository: UserRepository
12+
public struct GeminiGenerativeEmotionPromptUseCase: GenerativeEmotionPromptUseCase {
1413
var generativeRepository: GenerativeAIRepository
1514
var generator: PromptGenerator
1615

1716
public init(
18-
userRepository: UserRepository,
1917
generativeRepository: GenerativeAIRepository,
2018
generator: PromptGenerator
2119
) {
22-
self.userRepository = userRepository
2320
self.generativeRepository = generativeRepository
2421
self.generator = generator
2522
}
2623

2724
public func generate(_ input: String) async throws -> String? {
28-
var username: String
29-
do {
30-
username = try await userRepository.fetchUserName()
31-
} catch {
32-
username = "User"
33-
}
34-
let prompt = try generator.generatePrompt(for: input, username: username)
25+
let prompt = try generator.generatePrompt(for: input)
3526
return try await generativeRepository.generateContent(for: prompt)
3627
}
3728
}

Heim/Domain/Domain/UseCase/GenerativeSummaryUseCase.swift

+2-11
Original file line numberDiff line numberDiff line change
@@ -9,29 +9,20 @@ public protocol GenerativeSummaryPromptUseCase {
99
func generate(_ input: String) async throws -> String?
1010
}
1111

12-
public struct GeminiGenerativeSummaryPromptUseCase: GenerativeSummaryPromptUseCase, UserUseCase {
13-
public var userRepository: UserRepository
12+
public struct GeminiGenerativeSummaryPromptUseCase: GenerativeSummaryPromptUseCase {
1413
var generativeRepository: GenerativeAIRepository
1514
var generator: PromptGenerator
1615

1716
public init(
18-
userRepository: UserRepository,
1917
generativeRepository: GenerativeAIRepository,
2018
generator: PromptGenerator
2119
) {
22-
self.userRepository = userRepository
2320
self.generativeRepository = generativeRepository
2421
self.generator = generator
2522
}
2623

2724
public func generate(_ input: String) async throws -> String? {
28-
var username: String
29-
do {
30-
username = try await userRepository.fetchUserName()
31-
} catch {
32-
username = "User"
33-
}
34-
let prompt = try generator.generatePrompt(for: input, username: username)
25+
let prompt = try generator.generatePrompt(for: input)
3526
return try await generativeRepository.generateContent(for: prompt)
3627
}
3728
}

0 commit comments

Comments
 (0)