Skip to content

Commit 88caa97

Browse files
committed
clean-up
1 parent b8c9cb4 commit 88caa97

1 file changed

Lines changed: 4 additions & 10 deletions

File tree

core/src/main/kotlin/org/evomaster/core/languagemodel/LanguageModelConnector.kt

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ import org.evomaster.core.languagemodel.data.OllamaRequest
88
import org.evomaster.core.languagemodel.data.OllamaResponse
99
import org.evomaster.core.languagemodel.data.Prompt
1010
import org.evomaster.core.logging.LoggingUtil
11-
import org.slf4j.LoggerFactory
1211
import java.io.DataOutputStream
1312
import java.net.HttpURLConnection
1413
import java.net.URL
@@ -27,13 +26,9 @@ import kotlin.math.min
2726
*/
2827
class 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

Comments
 (0)