Module io.helidon.integrations.langchain4j.providers.mock

Package io.helidon.integrations.langchain4j.providers.mock

Class MockChatModel

java.lang.Object

io.helidon.integrations.langchain4j.providers.mock.MockChatModel

All Implemented Interfaces:

dev.langchain4j.model.chat.ChatModel, RuntimeType.Api<MockChatModelConfig>

public class MockChatModel
extends Object
implements RuntimeType.Api<MockChatModelConfig>, dev.langchain4j.model.chat.ChatModel

Mock implementation of ChatModel for testing.

This model evaluates the list of MockChatRule defined in configuration or injected as singleton service. The first rule that matches the incoming ChatRequest is used to generate a mock ChatResponse. If no rule matches, a default rule is applied.

Method Summary

Modifier and Type	Method	Description
List<MockChatRule>	activeRules()	Returns the list of MockChatRules currently used by this model.
static MockChatModelConfig.Builder	builder()	Returns a new MockChatModelConfig.Builder instance for building a configuration.
static MockChatModel	create(MockChatModelConfig config)	Creates a new MockChatModel instance using the provided configuration.
static MockChatModel	create(Consumer<MockChatModelConfig.Builder> consumer)	Creates a new MockChatModel instance using a builder consumer to configure the model.
dev.langchain4j.model.chat.response.ChatResponse	doChat(dev.langchain4j.model.chat.request.ChatRequest chatRequest)
MockChatModelConfig	prototype()	Returns the configuration prototype of this mock chat model.
void	resetRules()	Resets the mock chat rules to the original set defined in the configuration.

Methods inherited from class java.lang.Object

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

Methods inherited from interface dev.langchain4j.model.chat.ChatModel

chat, chat, chat, chat, defaultRequestParameters, listeners, provider, supportedCapabilities

Method Details

create

public static MockChatModel create(MockChatModelConfig config)

Creates a new MockChatModel instance using the provided configuration.

Parameters:

config - the mock chat model configuration

Returns:

a new MockChatModel

create

public static MockChatModel create(Consumer<MockChatModelConfig.Builder> consumer)

Creates a new MockChatModel instance using a builder consumer to configure the model.

Parameters:

consumer - a consumer that modifies the MockChatModelConfig.Builder

Returns:

a new MockChatModel

builder

public static MockChatModelConfig.Builder builder()

Returns a new MockChatModelConfig.Builder instance for building a configuration.

Returns:

a builder for MockChatModelConfig

doChat

public dev.langchain4j.model.chat.response.ChatResponse doChat(dev.langchain4j.model.chat.request.ChatRequest chatRequest)

Specified by:

doChat in interface dev.langchain4j.model.chat.ChatModel

activeRules

public List<MockChatRule> activeRules()

Returns the list of MockChatRules currently used by this model.

The returned list is the live internal list; modifications to it will affect the model's behavior. Use resetRules() to restore the original configuration.

Returns:

mutable list of mock chat rules

resetRules

public void resetRules()

Resets the mock chat rules to the original set defined in the configuration.

This method clears any modifications made to the activeRules() list and restores the rules from the initial MockChatModelConfig.

prototype

public MockChatModelConfig prototype()

Returns the configuration prototype of this mock chat model.

Specified by:

prototype in interface RuntimeType.Api<MockChatModelConfig>

Returns:

the MockChatModelConfig used by this model