dialog2graph.pipelines.d2g_extender.pipeline

The module contains pipeline to extend dialog graphs.

Classes

D2GExtenderPipeline

LLM graph extender pipeline

Module Contents

class dialog2graph.pipelines.d2g_extender.pipeline.D2GExtenderPipeline(name: str, model_storage: dialog2graph.pipelines.model_storage.ModelStorage, grouping_llm: str = 'extender_grouping_llm:v1', extending_llm: str = 'extender_extending_llm:v1', filling_llm: str = 'extender_filling_llm:v1', formatting_llm: str = 'extender_formatting_llm:v1', dialog_llm: str = 'extender_dialog_llm:v1', sim_model: str = 'extender_sim_model:v1', step1_evals: list[Callable] = metrics.PreDGEvalBase, extender_evals: list[Callable] = metrics.PreDGEvalBase, step2_evals: list[Callable] = metrics.DGEvalBase, end_evals: list[Callable] = metrics.DGEvalBase, step: int = 2)

Bases: dialog2graph.pipelines.core.pipeline.BasePipeline

LLM graph extender pipeline

Parameters:

• name (str)

• model_storage (dialog2graph.pipelines.model_storage.ModelStorage)

• grouping_llm (str)

• extending_llm (str)

• filling_llm (str)

• formatting_llm (str)

• dialog_llm (str)

• sim_model (str)

• step1_evals (list[Callable])

• extender_evals (list[Callable])

• step2_evals (list[Callable])

• end_evals (list[Callable])

• step (int)