http4k-connect-ai-ollama-fake/org.http4k.connect.ollama

Package-level declarations

Types

ChatCompletionGenerator

fun interface ChatCompletionGenerator : Function2<List<Message>, ResponseFormat?, List<String>>

class FakeOllama(val models: Storage<Model> = DEFAULT_OLLAMA_MODELS, val completionGenerators: Map<ModelName, ChatCompletionGenerator> = emptyMap(), clock: Clock = systemUTC()) : ChaoticHttpHandler

Properties

DEFAULT_OLLAMA_MODELS

val DEFAULT_OLLAMA_MODELS: Storage<Model>

val gemma: ModelName

val mistral: ModelName

Functions

fun chatCompletion(clock: Clock, completionGenerators: Map<ModelName, ChatCompletionGenerator>): RoutingHttpHandler

fun completion(clock: Clock, completionGenerators: Map<ModelName, ChatCompletionGenerator>): RoutingHttpHandler

createEmbeddings

fun createEmbeddings(models: Storage<Model>): RoutingHttpHandler

fun getModels(models: Storage<Model>): RoutingHttpHandler

fun ChatCompletionGenerator.Companion.LoremIpsum(random: Random = Random(0)): ChatCompletionGenerator

fun main()

fun pullModel(): RoutingHttpHandler