org.http4k.connect.openai.plugins.ServicePluginIntegration.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of http4k-connect-ai-openai-fake Show documentation

null

The newest version!

package org.http4k.connect.openai.plugins

import org.http4k.connect.openai.auth.OpenAIPluginId
import org.http4k.connect.openai.plugins.internal.ForwardCallsToPluginServer
import org.http4k.connect.openai.plugins.internal.LoadOpenApi
import org.http4k.core.Filter
import org.http4k.core.HttpHandler
import org.http4k.core.Uri
import org.http4k.routing.routes
import java.time.Clock

/**
 * Plugin implementation which plugs into the FakeOpenAI server. It uses Basic or Bearer
 * auth to auth against the server.
 */
fun ServicePluginIntegration(
    securityTokenFilter: Filter,
    pluginId: OpenAIPluginId,
    pluginUri: Uri
) = object : PluginIntegration {
    override val pluginId = pluginId

    override fun buildIntegration(openAiUrl: Uri, http: HttpHandler, clock: Clock) =
        IntegratedPlugin(
            pluginId,
            routes(
                LoadOpenApi(pluginId, openAiUrl, http, pluginUri),
                ForwardCallsToPluginServer(pluginId, http, pluginUri) {
                    securityTokenFilter
                }
            ),
            securityTokenFilter
        )
}