Package org.springframework.ai.moonshot

Class MoonshotChatModel

java.lang.Object

org.springframework.ai.chat.model.AbstractToolCallSupport

org.springframework.ai.moonshot.MoonshotChatModel

All Implemented Interfaces:

ChatModel, StreamingChatModel, Model<Prompt,ChatResponse>, StreamingModel<Prompt,ChatResponse>

public class MoonshotChatModel
extends AbstractToolCallSupport
implements ChatModel, StreamingChatModel

Author:

Geng Rong

Field Summary

Fields inherited from class org.springframework.ai.chat.model.AbstractToolCallSupport

functionCallbackContext, functionCallbackRegister, IS_RUNTIME_CALL

Constructor Summary

Constructors

Constructor	Description
MoonshotChatModel(MoonshotApi moonshotApi)	Initializes a new instance of the MoonshotChatModel.
MoonshotChatModel(MoonshotApi moonshotApi, MoonshotChatOptions options)	Initializes a new instance of the MoonshotChatModel.
MoonshotChatModel(MoonshotApi moonshotApi, MoonshotChatOptions options, FunctionCallbackContext functionCallbackContext, List<FunctionCallback> toolFunctionCallbacks, org.springframework.retry.support.RetryTemplate retryTemplate, io.micrometer.observation.ObservationRegistry observationRegistry)	Initializes a new instance of the MoonshotChatModel.
MoonshotChatModel(MoonshotApi moonshotApi, MoonshotChatOptions options, FunctionCallbackContext functionCallbackContext, org.springframework.retry.support.RetryTemplate retryTemplate)	Initializes a new instance of the MoonshotChatModel.

Method Summary

Modifier and Type	Method	Description
ChatResponse	call(Prompt prompt)	Executes a method call to the AI model.
MoonshotApi.ChatCompletionRequest	createRequest(Prompt prompt, boolean stream)	Accessible for testing.
ChatOptions	getDefaultOptions()
void	setObservationConvention(ChatModelObservationConvention observationConvention)
reactor.core.publisher.Flux<ChatResponse>	stream(Prompt prompt)	Executes a method call to the AI model.

Methods inherited from class org.springframework.ai.chat.model.AbstractToolCallSupport

buildToolCallConversation, executeFunctions, getFunctionCallbackRegister, handleToolCalls, isProxyToolCalls, isToolCall, isToolCall, resolveFunctionCallbacks, runtimeFunctionCallbackConfigurations

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.springframework.ai.chat.model.ChatModel

call, call

Methods inherited from interface org.springframework.ai.chat.model.StreamingChatModel

stream, stream

Constructor Details

MoonshotChatModel

public MoonshotChatModel(MoonshotApi moonshotApi)

Initializes a new instance of the MoonshotChatModel.

Parameters:

moonshotApi - The Moonshot instance to be used for interacting with the Moonshot Chat API.

MoonshotChatModel

public MoonshotChatModel(MoonshotApi moonshotApi,
 MoonshotChatOptions options)

Initializes a new instance of the MoonshotChatModel.

Parameters:

moonshotApi - The Moonshot instance to be used for interacting with the Moonshot Chat API.

options - The MoonshotChatOptions to configure the chat client.

MoonshotChatModel

public MoonshotChatModel(MoonshotApi moonshotApi,
 MoonshotChatOptions options,
 FunctionCallbackContext functionCallbackContext,
 org.springframework.retry.support.RetryTemplate retryTemplate)

Initializes a new instance of the MoonshotChatModel.

Parameters:

moonshotApi - The Moonshot instance to be used for interacting with the Moonshot Chat API.

options - The MoonshotChatOptions to configure the chat client.

functionCallbackContext - The function callback context.

retryTemplate - The retry template.

MoonshotChatModel

public MoonshotChatModel(MoonshotApi moonshotApi,
 MoonshotChatOptions options,
 FunctionCallbackContext functionCallbackContext,
 List<FunctionCallback> toolFunctionCallbacks,
 org.springframework.retry.support.RetryTemplate retryTemplate,
 io.micrometer.observation.ObservationRegistry observationRegistry)

Initializes a new instance of the MoonshotChatModel.

Parameters:

moonshotApi - The Moonshot instance to be used for interacting with the Moonshot Chat API.

options - The MoonshotChatOptions to configure the chat client.

functionCallbackContext - The function callback context.

toolFunctionCallbacks - The tool function callbacks.

retryTemplate - The retry template.

observationRegistry - The ObservationRegistry used for instrumentation.

Method Details

call

public ChatResponse call(Prompt prompt)

Description copied from interface: Model

Executes a method call to the AI model.

Specified by:

call in interface ChatModel

Specified by:

call in interface Model<Prompt,ChatResponse>

Parameters:

prompt - the request object to be sent to the AI model

Returns:

the response from the AI model

getDefaultOptions

public ChatOptions getDefaultOptions()

Specified by:

getDefaultOptions in interface ChatModel

stream

public reactor.core.publisher.Flux<ChatResponse> stream(Prompt prompt)

Description copied from interface: StreamingModel

Executes a method call to the AI model.

Specified by:

stream in interface ChatModel

Specified by:

stream in interface StreamingChatModel

Specified by:

stream in interface StreamingModel<Prompt,ChatResponse>

Parameters:

prompt - the request object to be sent to the AI model

Returns:

the streaming response from the AI model

createRequest

public MoonshotApi.ChatCompletionRequest createRequest(Prompt prompt,
 boolean stream)

Accessible for testing.

setObservationConvention

public void setObservationConvention(ChatModelObservationConvention observationConvention)