aiq.front_ends.fastapi.step_adaptor

Attributes

logger

Classes

StepAdaptor

Module Contents

logger

class StepAdaptor(config: aiq.data_models.step_adaptor.StepAdaptorConfig)

_history: list[aiq.data_models.intermediate_step.IntermediateStep] = []

config

_step_matches_filter(

step: aiq.data_models.intermediate_step.IntermediateStep,

config: aiq.data_models.step_adaptor.StepAdaptorConfig,

) → bool

Returns True if this intermediate step should be included (based on the config.mode).

_handle_llm(

step: aiq.data_models.intermediate_step.IntermediateStepPayload,

ancestry: aiq.data_models.invocation_node.InvocationNode,

) → aiq.data_models.api_server.AIQResponseSerializable | None

_handle_tool_end(

payload: aiq.data_models.intermediate_step.IntermediateStepPayload,

ancestry: aiq.data_models.invocation_node.InvocationNode,

) → aiq.data_models.api_server.AIQResponseSerializable | None

Handles the TOOL_END event

_handle_function(

step: aiq.data_models.intermediate_step.IntermediateStepPayload,

ancestry: aiq.data_models.invocation_node.InvocationNode,

) → aiq.data_models.api_server.AIQResponseSerializable | None

Handles the FUNCTION_START and FUNCTION_END events

_handle_custom(

payload: aiq.data_models.intermediate_step.IntermediateStepPayload,

ancestry: aiq.data_models.invocation_node.InvocationNode,

) → aiq.data_models.api_server.AIQResponseSerializable | None

Handles the CUSTOM event

process(

step: aiq.data_models.intermediate_step.IntermediateStep,

) → aiq.data_models.api_server.AIQResponseSerializable | None