feat: тайминг-лог запросов ([llm-proxy] chat model=... status=... в Nms)
Build LLM Proxy / Build and push (release) Successful in 22s
Build LLM Proxy / Build and push (release) Successful in 22s
This commit is contained in:
@@ -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,
|
||||||
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user