Skip to content

Commit f2a5710

Browse files
committed
update after review
1 parent 55687e1 commit f2a5710

3 files changed

Lines changed: 12 additions & 6 deletions

File tree

core/src/main/kotlin/org/evomaster/core/BaseModule.kt

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@ package org.evomaster.core
33
import com.google.inject.AbstractModule
44
import com.google.inject.Provides
55
import com.google.inject.Singleton
6+
import org.evomaster.core.llm.service.DictionaryService
7+
import org.evomaster.core.llm.service.LlmService
68
import org.evomaster.core.output.service.PartialOracles
79
import org.evomaster.core.search.service.mutator.genemutation.ArchiveImpactSelector
810
import org.evomaster.core.search.service.*
@@ -95,6 +97,12 @@ class BaseModule(val args: Array<String>, val noTests: Boolean = false) : Abstra
9597
bind(WarningsAggregator::class.java)
9698
.asEagerSingleton()
9799

100+
bind(LlmService::class.java)
101+
.asEagerSingleton()
102+
103+
bind(DictionaryService::class.java)
104+
.asEagerSingleton()
105+
98106
//no longer needed if TestSuiteWriter is moved out?
99107
// if(noTests){
100108
// bind(TestCaseWriter::class.java)

core/src/main/kotlin/org/evomaster/core/llm/mock/MockChatModel.kt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,10 @@ class MockChatModel : ChatModel {
5555
val text = chatRequest.messages().joinToString { it.toString() }
5656

5757
val res = responses.find { mock -> mock.matcher(text) }?.responseProducer(text)
58+
/*
59+
TODO we could also have a boolen parameter to decide what to do in case of no match
60+
(eg response as now or throw exception), to make the mock more configurable
61+
*/
5862
?: "Hei! You forgot to setup the mock for me"
5963

6064
return ChatResponse.builder().aiMessage(AiMessage.from(res)).build()

core/src/main/kotlin/org/evomaster/core/problem/enterprise/service/EnterpriseModule.kt

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,6 @@ abstract class EnterpriseModule : AbstractModule() {
1414
bind(WFCReportWriter::class.java)
1515
.asEagerSingleton()
1616

17-
bind(LlmService::class.java)
18-
.asEagerSingleton()
19-
20-
bind(DictionaryService::class.java)
21-
.asEagerSingleton()
22-
2317
bind(SSRFAnalyser::class.java)
2418
.asEagerSingleton()
2519

0 commit comments

Comments
 (0)