Package ai.vespa.llm.test

Class MockLanguageModel

java.lang.Object

ai.vespa.llm.test.MockLanguageModel

All Implemented Interfaces:

LanguageModel

public class MockLanguageModel
extends Object
implements LanguageModel

Author:

bratseth

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	MockLanguageModel.Builder

Constructor Summary

Constructors

Constructor	Description
MockLanguageModel(MockLanguageModel.Builder builder)

Method Summary

Modifier and Type	Method	Description
List<Completion>	complete(Prompt prompt, InferenceParameters options)
CompletableFuture<Completion.FinishReason>	completeAsync(Prompt prompt, InferenceParameters options, Consumer<Completion> action)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

MockLanguageModel

public MockLanguageModel(MockLanguageModel.Builder builder)

Method Details

complete

public List<Completion> complete(Prompt prompt,
 InferenceParameters options)

Specified by:

complete in interface LanguageModel

completeAsync

public CompletableFuture<Completion.FinishReason> completeAsync(Prompt prompt,
 InferenceParameters options,
 Consumer<Completion> action)

Specified by:

completeAsync in interface LanguageModel