@@ -8,7 +8,6 @@ import org.evomaster.core.languagemodel.data.OllamaRequest
88import org.evomaster.core.languagemodel.data.OllamaResponse
99import org.evomaster.core.languagemodel.data.Prompt
1010import org.evomaster.core.logging.LoggingUtil
11- import org.slf4j.LoggerFactory
1211import java.io.DataOutputStream
1312import java.net.HttpURLConnection
1413import java.net.URL
@@ -27,13 +26,9 @@ import kotlin.math.min
2726 */
2827class LanguageModelConnector {
2928
30- companion object {
31- private val log = LoggerFactory .getLogger(LanguageModelConnector ::class .java)
32- }
33-
3429 @PostConstruct
3530 fun initialise () {
36- log.debug (" Initializing {}" , LanguageModelConnector ::class .simpleName)
31+ LoggingUtil .getInfoLogger().info (" Initializing {}" , LanguageModelConnector ::class .simpleName)
3732 }
3833
3934 @Inject
@@ -107,8 +102,7 @@ class LanguageModelConnector {
107102
108103 return makeQuery(prompt)
109104 }
110-
111-
105+
112106 private fun makeQuery (prompt : String , id : String? = null): String? {
113107 validatePrompt(prompt)
114108
@@ -165,7 +159,7 @@ class LanguageModelConnector {
165159 writer.close()
166160
167161 if (connection.responseCode != HttpURLConnection .HTTP_OK ) {
168- LoggingUtil .Companion .uniqueWarn(log, " Failed to connect to language model server" )
162+ LoggingUtil .getInfoLogger().warn( " Failed to connect to language model server. " )
169163 return null
170164 }
171165
@@ -176,7 +170,7 @@ class LanguageModelConnector {
176170
177171 return response.response
178172 } catch (e: Exception ) {
179- LoggingUtil .Companion .uniqueWarn(log, " Failed to connect to language model server: ${e.message} " )
173+ LoggingUtil .getInfoLogger().warn( " Failed to connect to language model server: ${e.message} . " )
180174
181175 return null
182176 }
0 commit comments