nat.test.functions

Classes

EchoFunctionConfig
StreamingEchoFunctionConfig
ConstantFunctionConfig
StreamingConstantFunctionConfig

Functions

echo_function(config, builder)
streaming_function(config, builder)
constant_function(config, builder)
streaming_constant_function(config, builder)

Module Contents

class EchoFunctionConfig

Bases: nat.data_models.function.FunctionBaseConfig

use_openai_api: bool = False

async echo_function(

config: EchoFunctionConfig,

builder: nat.builder.builder.Builder,

)

class StreamingEchoFunctionConfig

Bases: nat.data_models.function.FunctionBaseConfig

use_openai_api: bool = False

async streaming_function(

config: StreamingEchoFunctionConfig,

builder: nat.builder.builder.Builder,

)

class ConstantFunctionConfig

Bases: nat.data_models.function.FunctionBaseConfig

response: str

async constant_function(

config: ConstantFunctionConfig,

builder: nat.builder.builder.Builder,

)

class StreamingConstantFunctionConfig

Bases: nat.data_models.function.FunctionBaseConfig

responses: list[str]

async streaming_constant_function(

config: StreamingConstantFunctionConfig,

builder: nat.builder.builder.Builder,

)