feat: тайминг-лог запросов ([llm-proxy] chat model=... status=... в Nms)
Build LLM Proxy / Build and push (release) Successful in 22s

This commit is contained in:
Hermes Agent
2026-08-20 17:39:18 +03:00
parent 8fc743860c
commit f45ed0d11c
+24 -10
View File
@@ -103,6 +103,10 @@ private suspend fun handleChat(
return return
} }
val start = System.currentTimeMillis()
val model = try {
json.parseToJsonElement(patched).jsonObject["model"]?.jsonPrimitive?.content ?: "?"
} catch (e: Exception) { "?" }
val req = HttpRequest.newBuilder() val req = HttpRequest.newBuilder()
.uri(URI.create(upstream.trimEnd('/') + "/chat/completions")) .uri(URI.create(upstream.trimEnd('/') + "/chat/completions"))
.header("Authorization", "Bearer $apiKey") .header("Authorization", "Bearer $apiKey")
@@ -111,17 +115,27 @@ private suspend fun handleChat(
.build() .build()
val stream = json.parseToJsonElement(raw).jsonObject["stream"]?.jsonPrimitive?.content == "true" val stream = json.parseToJsonElement(raw).jsonObject["stream"]?.jsonPrimitive?.content == "true"
if (stream) { try {
// SSE-стрим: транслируем как есть, чанк за чанком. if (stream) {
val resp = http.send(req, HttpResponse.BodyHandlers.ofInputStream()) // SSE-стрим: транслируем как есть, чанк за чанком.
val ct = resp.headers().firstValue("content-type").orElse("text/event-stream") val resp = http.send(req, HttpResponse.BodyHandlers.ofInputStream())
call.respondOutputStream(ContentType.parse(ct), HttpStatusCode.fromValue(resp.statusCode())) { val ct = resp.headers().firstValue("content-type").orElse("text/event-stream")
resp.body().use { input -> input.copyTo(this, 8192) } call.respondOutputStream(ContentType.parse(ct), HttpStatusCode.fromValue(resp.statusCode())) {
resp.body().use { input -> input.copyTo(this, 8192) }
}
println("[llm-proxy] chat model=$model status=${resp.statusCode()} в ${System.currentTimeMillis() - start}ms stream=true")
} else {
val resp = http.send(req, HttpResponse.BodyHandlers.ofByteArray())
val ct = resp.headers().firstValue("content-type").orElse("application/json")
call.respondBytes(resp.body(), ContentType.parse(ct), HttpStatusCode.fromValue(resp.statusCode()))
println("[llm-proxy] chat model=$model status=${resp.statusCode()} в ${System.currentTimeMillis() - start}ms stream=false")
} }
} else { } catch (e: Exception) {
val resp = http.send(req, HttpResponse.BodyHandlers.ofByteArray()) println("[llm-proxy] chat model=$model ОШИБКА: ${e.message} в ${System.currentTimeMillis() - start}ms")
val ct = resp.headers().firstValue("content-type").orElse("application/json") call.respondBytes(
call.respondBytes(resp.body(), ContentType.parse(ct), HttpStatusCode.fromValue(resp.statusCode())) """{"error":{"message":"upstream: ${e.message}"}}""".toByteArray(),
ContentType.Application.Json, HttpStatusCode.BadGateway,
)
} }
} }