Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| c140d0b758 |
@@ -41,6 +41,7 @@ kotlin {
|
|||||||
implementation(libs.kotlinx.cli)
|
implementation(libs.kotlinx.cli)
|
||||||
|
|
||||||
implementation(libs.kotlinx.coroutines.core)
|
implementation(libs.kotlinx.coroutines.core)
|
||||||
|
implementation(libs.ktor.client.cio)
|
||||||
}
|
}
|
||||||
// :agentik-cli — commonMain-only (нет jvmMain/nativeMain разделения):
|
// :agentik-cli — commonMain-only (нет jvmMain/nativeMain разделения):
|
||||||
// весь код, включая platformEnv, лежит в commonMain.
|
// весь код, включая platformEnv, лежит в commonMain.
|
||||||
|
|||||||
@@ -0,0 +1,23 @@
|
|||||||
|
package pw.binom.agentik.cli
|
||||||
|
|
||||||
|
import io.ktor.client.HttpClient
|
||||||
|
import io.ktor.client.engine.cio.CIO
|
||||||
|
import pw.binom.agentik.client.agentikHttpClient
|
||||||
|
|
||||||
|
/**
|
||||||
|
* HTTP-клиент CLI: движок CIO + конфигурация agentik.
|
||||||
|
*
|
||||||
|
* Движок живёт здесь, а не в `:client`: библиотека не выбирает транспорт за
|
||||||
|
* потребителя. Таргеты `:agentik-cli` (jvm + linuxX64/macosX64/macosArm64/mingwX64)
|
||||||
|
* покрываются CIO.
|
||||||
|
*
|
||||||
|
* `requestTimeout = 0` — отключение встроенного request-таймаута CIO;
|
||||||
|
* defense-in-depth против обрыва долгих SSE-idle (основная защита —
|
||||||
|
* `noSseReadTimeout` в `:client`).
|
||||||
|
*
|
||||||
|
* [token] = `null` — авторизация выключена.
|
||||||
|
*/
|
||||||
|
internal fun defaultCliHttpClient(token: String? = null): HttpClient =
|
||||||
|
agentikHttpClient(engineFactory = CIO, token = token) {
|
||||||
|
engine { requestTimeout = 0 }
|
||||||
|
}
|
||||||
+2
-1
@@ -2,13 +2,14 @@ package pw.binom.agentik.cli.commands
|
|||||||
|
|
||||||
import kotlinx.cli.ArgType
|
import kotlinx.cli.ArgType
|
||||||
import pw.binom.agentik.cli.AgentikSubcommand
|
import pw.binom.agentik.cli.AgentikSubcommand
|
||||||
|
import pw.binom.agentik.cli.defaultCliHttpClient
|
||||||
import pw.binom.agentik.client.AgentikAgent
|
import pw.binom.agentik.client.AgentikAgent
|
||||||
|
|
||||||
class ConvDeleteSubcommand : ConvSubcommand("delete", "Удалить диалог") {
|
class ConvDeleteSubcommand : ConvSubcommand("delete", "Удалить диалог") {
|
||||||
val id by argument(ArgType.String, description = "ID диалога")
|
val id by argument(ArgType.String, description = "ID диалога")
|
||||||
|
|
||||||
override fun execute() = kotlinx.coroutines.runBlocking {
|
override fun execute() = kotlinx.coroutines.runBlocking {
|
||||||
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl)
|
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl, httpClient = defaultCliHttpClient())
|
||||||
val ok = agent.deleteConversation(id)
|
val ok = agent.deleteConversation(id)
|
||||||
if (ok) println("deleted: $id") else println("conversation not found: $id")
|
if (ok) println("deleted: $id") else println("conversation not found: $id")
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ package pw.binom.agentik.cli.commands
|
|||||||
import kotlinx.cli.ArgType
|
import kotlinx.cli.ArgType
|
||||||
import kotlinx.cli.default
|
import kotlinx.cli.default
|
||||||
import pw.binom.agentik.cli.AgentikSubcommand
|
import pw.binom.agentik.cli.AgentikSubcommand
|
||||||
|
import pw.binom.agentik.cli.defaultCliHttpClient
|
||||||
import pw.binom.agentik.client.AgentikAgent
|
import pw.binom.agentik.client.AgentikAgent
|
||||||
import pw.binom.agentik.proto.Agent
|
import pw.binom.agentik.proto.Agent
|
||||||
|
|
||||||
@@ -10,7 +11,7 @@ class ConvLsSubcommand : ConvSubcommand("ls", "Список диалогов а
|
|||||||
val limit by option(ArgType.Int, fullName = "limit", description = "Максимум диалогов").default(Agent.PAGE_SIZE)
|
val limit by option(ArgType.Int, fullName = "limit", description = "Максимум диалогов").default(Agent.PAGE_SIZE)
|
||||||
|
|
||||||
override fun execute() = kotlinx.coroutines.runBlocking {
|
override fun execute() = kotlinx.coroutines.runBlocking {
|
||||||
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl)
|
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl, httpClient = defaultCliHttpClient())
|
||||||
val convs = agent.getConversations(offset = 0, limit = limit.coerceAtMost(Agent.PAGE_SIZE))
|
val convs = agent.getConversations(offset = 0, limit = limit.coerceAtMost(Agent.PAGE_SIZE))
|
||||||
if (convs.isEmpty()) {
|
if (convs.isEmpty()) {
|
||||||
println("(no conversations)")
|
println("(no conversations)")
|
||||||
|
|||||||
+2
-1
@@ -3,13 +3,14 @@ package pw.binom.agentik.cli.commands
|
|||||||
import kotlinx.cli.ArgType
|
import kotlinx.cli.ArgType
|
||||||
import kotlinx.cli.default
|
import kotlinx.cli.default
|
||||||
import pw.binom.agentik.cli.AgentikSubcommand
|
import pw.binom.agentik.cli.AgentikSubcommand
|
||||||
|
import pw.binom.agentik.cli.defaultCliHttpClient
|
||||||
import pw.binom.agentik.client.AgentikAgent
|
import pw.binom.agentik.client.AgentikAgent
|
||||||
|
|
||||||
class ConvNewSubcommand : ConvSubcommand("new", "Создать диалог; печатает id") {
|
class ConvNewSubcommand : ConvSubcommand("new", "Создать диалог; печатает id") {
|
||||||
val temp by option(ArgType.Boolean, fullName = "temp", description = "Временный диалог").default(false)
|
val temp by option(ArgType.Boolean, fullName = "temp", description = "Временный диалог").default(false)
|
||||||
|
|
||||||
override fun execute() = kotlinx.coroutines.runBlocking {
|
override fun execute() = kotlinx.coroutines.runBlocking {
|
||||||
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl)
|
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl, httpClient = defaultCliHttpClient())
|
||||||
val conv = agent.createConversation(temp = temp)
|
val conv = agent.createConversation(temp = temp)
|
||||||
println(conv.id)
|
println(conv.id)
|
||||||
}
|
}
|
||||||
|
|||||||
+2
-1
@@ -2,6 +2,7 @@ package pw.binom.agentik.cli.commands
|
|||||||
|
|
||||||
import kotlinx.cli.ArgType
|
import kotlinx.cli.ArgType
|
||||||
import pw.binom.agentik.cli.AgentikSubcommand
|
import pw.binom.agentik.cli.AgentikSubcommand
|
||||||
|
import pw.binom.agentik.cli.defaultCliHttpClient
|
||||||
import pw.binom.agentik.client.AgentikAgent
|
import pw.binom.agentik.client.AgentikAgent
|
||||||
|
|
||||||
class ConvRenameSubcommand : ConvSubcommand("rename", "Переименовать диалог") {
|
class ConvRenameSubcommand : ConvSubcommand("rename", "Переименовать диалог") {
|
||||||
@@ -9,7 +10,7 @@ class ConvRenameSubcommand : ConvSubcommand("rename", "Переименоват
|
|||||||
val title by argument(ArgType.String, description = "Новое название")
|
val title by argument(ArgType.String, description = "Новое название")
|
||||||
|
|
||||||
override fun execute() = kotlinx.coroutines.runBlocking {
|
override fun execute() = kotlinx.coroutines.runBlocking {
|
||||||
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl)
|
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl, httpClient = defaultCliHttpClient())
|
||||||
val conv = agent.getConversation(id) ?: run {
|
val conv = agent.getConversation(id) ?: run {
|
||||||
println("conversation not found: $id")
|
println("conversation not found: $id")
|
||||||
return@runBlocking
|
return@runBlocking
|
||||||
|
|||||||
+2
-1
@@ -2,13 +2,14 @@ package pw.binom.agentik.cli.commands
|
|||||||
|
|
||||||
import kotlinx.cli.ArgType
|
import kotlinx.cli.ArgType
|
||||||
import pw.binom.agentik.cli.AgentikSubcommand
|
import pw.binom.agentik.cli.AgentikSubcommand
|
||||||
|
import pw.binom.agentik.cli.defaultCliHttpClient
|
||||||
import pw.binom.agentik.client.AgentikAgent
|
import pw.binom.agentik.client.AgentikAgent
|
||||||
|
|
||||||
class ConvShowSubcommand : ConvSubcommand("show", "Метаданные диалога") {
|
class ConvShowSubcommand : ConvSubcommand("show", "Метаданные диалога") {
|
||||||
val id by argument(ArgType.String, description = "ID диалога")
|
val id by argument(ArgType.String, description = "ID диалога")
|
||||||
|
|
||||||
override fun execute() = kotlinx.coroutines.runBlocking {
|
override fun execute() = kotlinx.coroutines.runBlocking {
|
||||||
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl)
|
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl, httpClient = defaultCliHttpClient())
|
||||||
val conv = agent.getConversation(id) ?: run {
|
val conv = agent.getConversation(id) ?: run {
|
||||||
println("conversation not found: $id")
|
println("conversation not found: $id")
|
||||||
return@runBlocking
|
return@runBlocking
|
||||||
|
|||||||
+2
-1
@@ -2,13 +2,14 @@ package pw.binom.agentik.cli.commands
|
|||||||
|
|
||||||
import kotlinx.cli.ArgType
|
import kotlinx.cli.ArgType
|
||||||
import pw.binom.agentik.cli.AgentikSubcommand
|
import pw.binom.agentik.cli.AgentikSubcommand
|
||||||
|
import pw.binom.agentik.cli.defaultCliHttpClient
|
||||||
import pw.binom.agentik.client.AgentikAgent
|
import pw.binom.agentik.client.AgentikAgent
|
||||||
|
|
||||||
class InterruptSubcommand : AgentikSubcommand("interrupt", "Прервать текущий ход диалога") {
|
class InterruptSubcommand : AgentikSubcommand("interrupt", "Прервать текущий ход диалога") {
|
||||||
val id by argument(ArgType.String, description = "ID диалога")
|
val id by argument(ArgType.String, description = "ID диалога")
|
||||||
|
|
||||||
override fun execute() = kotlinx.coroutines.runBlocking {
|
override fun execute() = kotlinx.coroutines.runBlocking {
|
||||||
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl)
|
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl, httpClient = defaultCliHttpClient())
|
||||||
val conv = agent.getConversation(id) ?: run {
|
val conv = agent.getConversation(id) ?: run {
|
||||||
println("conversation not found: $id")
|
println("conversation not found: $id")
|
||||||
return@runBlocking
|
return@runBlocking
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ package pw.binom.agentik.cli.commands
|
|||||||
import kotlinx.cli.ArgType
|
import kotlinx.cli.ArgType
|
||||||
import kotlinx.cli.default
|
import kotlinx.cli.default
|
||||||
import pw.binom.agentik.cli.AgentikSubcommand
|
import pw.binom.agentik.cli.AgentikSubcommand
|
||||||
|
import pw.binom.agentik.cli.defaultCliHttpClient
|
||||||
import pw.binom.agentik.client.AgentikAgent
|
import pw.binom.agentik.client.AgentikAgent
|
||||||
import pw.binom.agentik.proto.Content
|
import pw.binom.agentik.proto.Content
|
||||||
import pw.binom.agentik.proto.Message
|
import pw.binom.agentik.proto.Message
|
||||||
@@ -13,7 +14,7 @@ class MsgsSubcommand : AgentikSubcommand("msgs", "Показать сообще
|
|||||||
val limit by option(ArgType.Int, fullName = "limit", description = "Максимум сообщений").default(100)
|
val limit by option(ArgType.Int, fullName = "limit", description = "Максимум сообщений").default(100)
|
||||||
|
|
||||||
override fun execute() = kotlinx.coroutines.runBlocking {
|
override fun execute() = kotlinx.coroutines.runBlocking {
|
||||||
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl)
|
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl, httpClient = defaultCliHttpClient())
|
||||||
val conv = agent.getConversation(id) ?: run {
|
val conv = agent.getConversation(id) ?: run {
|
||||||
println("conversation not found: $id")
|
println("conversation not found: $id")
|
||||||
return@runBlocking
|
return@runBlocking
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ import kotlinx.coroutines.flow.onEach
|
|||||||
import kotlinx.coroutines.flow.takeWhile
|
import kotlinx.coroutines.flow.takeWhile
|
||||||
import kotlinx.coroutines.launch
|
import kotlinx.coroutines.launch
|
||||||
import pw.binom.agentik.cli.AgentikSubcommand
|
import pw.binom.agentik.cli.AgentikSubcommand
|
||||||
|
import pw.binom.agentik.cli.defaultCliHttpClient
|
||||||
import pw.binom.agentik.client.AgentikAgent
|
import pw.binom.agentik.client.AgentikAgent
|
||||||
import pw.binom.agentik.proto.Content
|
import pw.binom.agentik.proto.Content
|
||||||
import pw.binom.agentik.proto.Event
|
import pw.binom.agentik.proto.Event
|
||||||
@@ -17,7 +18,7 @@ class SendSubcommand : AgentikSubcommand("send", "Отправить user-ход
|
|||||||
val text by argument(ArgType.String, description = "Текст хода (все позиционные после <id> склеиваются пробелом)").vararg()
|
val text by argument(ArgType.String, description = "Текст хода (все позиционные после <id> склеиваются пробелом)").vararg()
|
||||||
|
|
||||||
override fun execute() = kotlinx.coroutines.runBlocking {
|
override fun execute() = kotlinx.coroutines.runBlocking {
|
||||||
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl)
|
val agent = AgentikAgent(id = agentId, baseUrl = serverUrl, httpClient = defaultCliHttpClient())
|
||||||
val conv = agent.getConversation(id) ?: run {
|
val conv = agent.getConversation(id) ?: run {
|
||||||
println("conversation not found: $id")
|
println("conversation not found: $id")
|
||||||
return@runBlocking
|
return@runBlocking
|
||||||
|
|||||||
@@ -22,8 +22,7 @@ kotlin {
|
|||||||
commonMain.dependencies {
|
commonMain.dependencies {
|
||||||
api(project(":proto"))
|
api(project(":proto"))
|
||||||
|
|
||||||
implementation(libs.ktor.client.core)
|
api(libs.ktor.client.core)
|
||||||
implementation(libs.ktor.client.cio)
|
|
||||||
implementation(libs.ktor.client.content.negotiation)
|
implementation(libs.ktor.client.content.negotiation)
|
||||||
implementation(libs.ktor.serialization.kotlinx.json)
|
implementation(libs.ktor.serialization.kotlinx.json)
|
||||||
|
|
||||||
@@ -37,6 +36,7 @@ kotlin {
|
|||||||
implementation(libs.ktor.server.core)
|
implementation(libs.ktor.server.core)
|
||||||
implementation(libs.ktor.server.test.host)
|
implementation(libs.ktor.server.test.host)
|
||||||
implementation(libs.ktor.client.content.negotiation)
|
implementation(libs.ktor.client.content.negotiation)
|
||||||
|
implementation(libs.ktor.client.cio)
|
||||||
implementation(libs.ktor.server.cio)
|
implementation(libs.ktor.server.cio)
|
||||||
implementation(libs.ktor.server.sse)
|
implementation(libs.ktor.server.sse)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,9 +8,11 @@ import pw.binom.agentik.proto.Agent
|
|||||||
* (модуль `:server`).
|
* (модуль `:server`).
|
||||||
*
|
*
|
||||||
* ```
|
* ```
|
||||||
|
* val http = HttpClient(CIO) { applyAgentikDefaults(token = "s3cret") }
|
||||||
* val client = AgentikAgent(
|
* val client = AgentikAgent(
|
||||||
* id = "my-agent",
|
* id = "my-agent",
|
||||||
* baseUrl = "http://localhost:8080/agentik",
|
* baseUrl = "http://localhost:8080/agentik",
|
||||||
|
* httpClient = http,
|
||||||
* )
|
* )
|
||||||
* val conv = client.createConversation(temp = false)
|
* val conv = client.createConversation(temp = false)
|
||||||
* conv.send(listOf(Content.Text("hi")))
|
* conv.send(listOf(Content.Text("hi")))
|
||||||
@@ -21,27 +23,13 @@ import pw.binom.agentik.proto.Agent
|
|||||||
* агента не знает, поэтому клиент должен её знать сам (или взять из
|
* агента не знает, поэтому клиент должен её знать сам (или взять из
|
||||||
* конфига).
|
* конфига).
|
||||||
*
|
*
|
||||||
* [httpClient] по умолчанию — [defaultAgentikHttpClient] (платформо-зависимый
|
* Клиент приходит снаружи: `:client` не выбирает движок. Собрать [HttpClient]
|
||||||
* движок: CIO на JVM, libcurl на desktop-native). Можно передать свой.
|
* можно через [agentikHttpClient] (фабрика движка + опциональные движковые
|
||||||
|
* настройки) или вручную, применив к блоку конфигурации [applyAgentikDefaults]
|
||||||
|
* (JSON + опциональный Bearer-токен).
|
||||||
*/
|
*/
|
||||||
fun AgentikAgent(
|
fun AgentikAgent(
|
||||||
id: String,
|
id: String,
|
||||||
baseUrl: String,
|
baseUrl: String,
|
||||||
token: String? = null,
|
httpClient: HttpClient,
|
||||||
httpClient: HttpClient = defaultAgentikHttpClient(token),
|
|
||||||
): Agent = AgentClient(httpClient = httpClient, baseUrl = baseUrl, id = id)
|
): Agent = AgentClient(httpClient = httpClient, baseUrl = baseUrl, id = id)
|
||||||
|
|
||||||
/**
|
|
||||||
* Дефолтный [HttpClient] для общения с `agentikAgent`. SSE-парсер ([readSse])
|
|
||||||
* живёт в общем коде и плагина `SSEClientContent` не требует.
|
|
||||||
*
|
|
||||||
* **Платформы:**
|
|
||||||
* - JVM: движок CIO. `engine { requestTimeout = 0 }` отключает встроенный
|
|
||||||
* 15-секундный request-таймаут движка (наш кастомный SSE-ридер не маркирует
|
|
||||||
* для долгих idle-стримов). Defense-in-depth: SSE-запросы в
|
|
||||||
* `ConversationClient.events`/`AgentClient.events` уже ставят
|
|
||||||
* `HttpTimeoutCapability` = INFINITE (см. [noSseReadTimeout]).
|
|
||||||
*
|
|
||||||
* Один движок CIO работает и на JVM, и на всех desktop-native (linux/macos/mingw).
|
|
||||||
* Реализация — в [HttpClientFactory.kt].
|
|
||||||
*/
|
|
||||||
|
|||||||
@@ -1,7 +1,9 @@
|
|||||||
package pw.binom.agentik.client
|
package pw.binom.agentik.client
|
||||||
|
|
||||||
import io.ktor.client.HttpClient
|
import io.ktor.client.HttpClient
|
||||||
import io.ktor.client.engine.cio.CIO
|
import io.ktor.client.HttpClientConfig
|
||||||
|
import io.ktor.client.engine.HttpClientEngineConfig
|
||||||
|
import io.ktor.client.engine.HttpClientEngineFactory
|
||||||
import io.ktor.client.plugins.DefaultRequest
|
import io.ktor.client.plugins.DefaultRequest
|
||||||
import io.ktor.client.plugins.contentnegotiation.ContentNegotiation
|
import io.ktor.client.plugins.contentnegotiation.ContentNegotiation
|
||||||
import io.ktor.client.request.header
|
import io.ktor.client.request.header
|
||||||
@@ -9,19 +11,26 @@ import io.ktor.http.HttpHeaders
|
|||||||
import io.ktor.serialization.kotlinx.json.json
|
import io.ktor.serialization.kotlinx.json.json
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Единый HTTP-клиент для JVM и всех 5 native-таргетов (:agentik-cli).
|
* Общая конфигурация HTTP-клиента agentik — платформо-независимая часть.
|
||||||
* CIO в ktor 3.x — KMP, поддерживает linuxX64/Arm64, macosX64/Arm64, mingwX64.
|
|
||||||
*
|
*
|
||||||
* `requestTimeout = 0` — defense-in-depth против read-таймаута на SSE:
|
* `:client` НЕ выбирает движок: его приносит потребитель. Здесь живёт только то,
|
||||||
* основная защита в `HttpRequestBuilder.noSseReadTimeout()` ([SseTimeout]).
|
* без чего клиент несовместим с `/agentik`:
|
||||||
|
* - JSON-конфиг [agentikJson] (обязан совпадать с серверным);
|
||||||
|
* - при заданном [token] — `Authorization: Bearer <token>` на ВСЕ запросы
|
||||||
|
* через [DefaultRequest] (накрывает 10 REST-вызовов и оба SSE-потока;
|
||||||
|
* заголовок живёт на клиенте, а не в отдельных запросах).
|
||||||
*
|
*
|
||||||
* При заданном [token] на ВСЕ запросы клиента навешивается
|
* `null` — авторизация выключена, заголовок не отправляется.
|
||||||
* `Authorization: Bearer <token>` через плагин [DefaultRequest]. Это накрывает
|
*
|
||||||
* все 10 REST-вызовов и оба SSE-потока сразу — заголовок живёт на HTTP-клиенте,
|
* Потребитель, знающий свой движок, добавляет к этому движковые настройки, напр.:
|
||||||
* а не в отдельных запросах.
|
* ```
|
||||||
|
* val http = HttpClient(CIO) {
|
||||||
|
* engine { requestTimeout = 0 } // CIO-специфика, живёт у потребителя
|
||||||
|
* applyAgentikDefaults(token)
|
||||||
|
* }
|
||||||
|
* ```
|
||||||
*/
|
*/
|
||||||
fun defaultAgentikHttpClient(token: String? = null): HttpClient = HttpClient(CIO) {
|
fun HttpClientConfig<*>.applyAgentikDefaults(token: String? = null) {
|
||||||
engine { requestTimeout = 0 }
|
|
||||||
install(ContentNegotiation) { json(agentikJson) }
|
install(ContentNegotiation) { json(agentikJson) }
|
||||||
if (token != null) {
|
if (token != null) {
|
||||||
install(DefaultRequest) {
|
install(DefaultRequest) {
|
||||||
@@ -29,3 +38,23 @@ fun defaultAgentikHttpClient(token: String? = null): HttpClient = HttpClient(CIO
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Создаёт [HttpClient] из фабрики движка потребителя и сразу применяет к нему
|
||||||
|
* конфигурацию agentik ([applyAgentikDefaults]).
|
||||||
|
*
|
||||||
|
* Это точка, где `:client` НЕ привязан к реализации транспорта: [engineFactory]
|
||||||
|
* выбирает потребитель (CIO, OkHttp, Darwin, …), а `:client` только конфигурирует
|
||||||
|
* созданный клиент.
|
||||||
|
*
|
||||||
|
* [configure] — опциональный последний штрих потребителя (движковые настройки:
|
||||||
|
* таймауты, прокси, логирование). Вызывается ПОСЛЕ [applyAgentikDefaults].
|
||||||
|
*/
|
||||||
|
fun <T : HttpClientEngineConfig> agentikHttpClient(
|
||||||
|
engineFactory: HttpClientEngineFactory<T>,
|
||||||
|
token: String? = null,
|
||||||
|
configure: (HttpClientConfig<T>.() -> Unit)? = null,
|
||||||
|
): HttpClient = HttpClient(engineFactory) {
|
||||||
|
applyAgentikDefaults(token)
|
||||||
|
configure?.invoke(this)
|
||||||
|
}
|
||||||
@@ -1,5 +1,7 @@
|
|||||||
package pw.binom.agentik.client
|
package pw.binom.agentik.client
|
||||||
|
|
||||||
|
import io.ktor.client.HttpClient
|
||||||
|
import io.ktor.client.engine.cio.CIO
|
||||||
import io.ktor.client.request.get
|
import io.ktor.client.request.get
|
||||||
import io.ktor.client.statement.bodyAsText
|
import io.ktor.client.statement.bodyAsText
|
||||||
import io.ktor.http.ContentType
|
import io.ktor.http.ContentType
|
||||||
@@ -19,7 +21,7 @@ import kotlin.test.Test
|
|||||||
import kotlin.test.assertEquals
|
import kotlin.test.assertEquals
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Тесты клиентской части: [defaultAgentikHttpClient] с заданным `token` прикладывает
|
* Тесты клиентской части: [applyAgentikDefaults] с заданным `token` прикладывает
|
||||||
* `Authorization: Bearer <token>` ко всем запросам через плагин `DefaultRequest`,
|
* `Authorization: Bearer <token>` ко всем запросам через плагин `DefaultRequest`,
|
||||||
* без токена — заголовок не отправляется.
|
* без токена — заголовок не отправляется.
|
||||||
*
|
*
|
||||||
@@ -47,6 +49,9 @@ class BearerHeaderTest {
|
|||||||
var token: String? = null
|
var token: String? = null
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private fun clientWith(token: String?): HttpClient =
|
||||||
|
HttpClient(CIO) { applyAgentikDefaults(token) }
|
||||||
|
|
||||||
private suspend fun startServer(): Pair<EmbeddedServer<*, *>, Int> {
|
private suspend fun startServer(): Pair<EmbeddedServer<*, *>, Int> {
|
||||||
val server = embeddedServer(ServerCIO, port = 0) {
|
val server = embeddedServer(ServerCIO, port = 0) {
|
||||||
routing {
|
routing {
|
||||||
@@ -66,7 +71,7 @@ class BearerHeaderTest {
|
|||||||
fun clientWithTokenAttachesBearerHeader() = runBlocking {
|
fun clientWithTokenAttachesBearerHeader() = runBlocking {
|
||||||
val (server, port) = startServer()
|
val (server, port) = startServer()
|
||||||
try {
|
try {
|
||||||
val client = defaultAgentikHttpClient("secret")
|
val client = clientWith("secret")
|
||||||
val resp = client.get("http://127.0.0.1:$port/agentik/conversations")
|
val resp = client.get("http://127.0.0.1:$port/agentik/conversations")
|
||||||
assertEquals(HttpStatusCode.OK, resp.status)
|
assertEquals(HttpStatusCode.OK, resp.status)
|
||||||
assertEquals("[]", resp.bodyAsText())
|
assertEquals("[]", resp.bodyAsText())
|
||||||
@@ -79,7 +84,7 @@ class BearerHeaderTest {
|
|||||||
fun clientWithoutTokenGets401(): Unit = runBlocking {
|
fun clientWithoutTokenGets401(): Unit = runBlocking {
|
||||||
val (server, port) = startServer()
|
val (server, port) = startServer()
|
||||||
try {
|
try {
|
||||||
val client = defaultAgentikHttpClient(null)
|
val client = clientWith(null)
|
||||||
val resp = client.get("http://127.0.0.1:$port/agentik/conversations")
|
val resp = client.get("http://127.0.0.1:$port/agentik/conversations")
|
||||||
assertEquals(HttpStatusCode.Unauthorized, resp.status)
|
assertEquals(HttpStatusCode.Unauthorized, resp.status)
|
||||||
} finally {
|
} finally {
|
||||||
@@ -91,7 +96,7 @@ class BearerHeaderTest {
|
|||||||
fun clientWithWrongTokenGets401(): Unit = runBlocking {
|
fun clientWithWrongTokenGets401(): Unit = runBlocking {
|
||||||
val (server, port) = startServer()
|
val (server, port) = startServer()
|
||||||
try {
|
try {
|
||||||
val client = defaultAgentikHttpClient("wrong")
|
val client = clientWith("wrong")
|
||||||
val resp = client.get("http://127.0.0.1:$port/agentik/conversations")
|
val resp = client.get("http://127.0.0.1:$port/agentik/conversations")
|
||||||
assertEquals(HttpStatusCode.Unauthorized, resp.status)
|
assertEquals(HttpStatusCode.Unauthorized, resp.status)
|
||||||
} finally {
|
} finally {
|
||||||
|
|||||||
Reference in New Issue
Block a user