Index

_

__copy__() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) __deepcopy__() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) __get_pydantic_json_schema__() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) __init__() (causal_agent.components.decision_tree.DecisionTreeEngine method) (causal_agent.components.decision_tree_llm.DecisionTreeLLMEngine method) (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) (causal_agent.synthetic.DiDGenerator method) (causal_agent.synthetic.generator.DataGenerator method) (causal_agent.synthetic.generator.DiDGenerator method) (causal_agent.synthetic.generator.FrontDoorGenerator method) (causal_agent.synthetic.generator.IVGenerator method) (causal_agent.synthetic.generator.MultiTreatRCTGenerator method) (causal_agent.synthetic.generator.ObservationalDataGenerator method) (causal_agent.synthetic.generator.PSMGenerator method) (causal_agent.synthetic.generator.PSWGenerator method) (causal_agent.synthetic.generator.RCTGenerator method) (causal_agent.synthetic.generator.RDDGenerator method) (causal_agent.synthetic.IVGenerator method) (causal_agent.synthetic.MultiTreatRCTGenerator method) (causal_agent.synthetic.PSMGenerator method) (causal_agent.synthetic.PSWGenerator method) (causal_agent.synthetic.RCTGenerator method) (causal_agent.synthetic.RDDGenerator method) (causal_agent.tools.data_analyzer.DataAnalyzer method)

__iter__() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) __pretty__() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) __pydantic_init_subclass__() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) __pydantic_on_complete__() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) __repr_name__() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) __repr_recursion__() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) __rich_repr__() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method)

A

analyze_dataset() (causal_agent.tools.data_analyzer.DataAnalyzer method) (in module causal_agent) (in module causal_agent.components) (in module causal_agent.components.dataset_analyzer) analyze_dataset_for_method() (in module causal_agent.utils.llm_helpers)

assess_balance() (in module causal_agent.methods.propensity_score) (in module causal_agent.methods.propensity_score.diagnostics) assess_gps_balance() (in module causal_agent.methods.generalized_propensity_score.diagnostics) assess_trends_visually() (in module causal_agent.methods.difference_in_differences.diagnostics) assess_variable_relationships() (in module causal_agent.components.dataset_analyzer) assess_weight_distribution() (in module causal_agent.methods.propensity_score.diagnostics)

B

ba_estimate_effect() (in module causal_agent.methods)

build_iv_graph_gml() (in module causal_agent.methods.instrumental_variable.estimator)

C

calculate_first_stage_f_statistic() (in module causal_agent.methods.instrumental_variable.diagnostics) calculate_standardized_differences() (in module causal_agent.methods.utils) call_llm_with_json_output() (in module causal_agent.utils.llm_helpers) canonical_did_model() (causal_agent.synthetic.DiDGenerator method) (causal_agent.synthetic.generator.DiDGenerator method) causal_agent module causal_agent.components module causal_agent.components.dataset_analyzer module causal_agent.components.decision_tree module causal_agent.components.decision_tree_llm module causal_agent.components.explanation_generator module causal_agent.components.input_parser module causal_agent.components.method_validator module causal_agent.components.output_formatter module causal_agent.components.query_interpreter module causal_agent.components.state_manager module causal_agent.methods module causal_agent.methods.backdoor_adjustment module causal_agent.methods.backdoor_adjustment.diagnostics module causal_agent.methods.backdoor_adjustment.estimator module causal_agent.methods.backdoor_adjustment.llm_assist module causal_agent.methods.causal_method module causal_agent.methods.diff_in_means module causal_agent.methods.diff_in_means.diagnostics module causal_agent.methods.diff_in_means.estimator module causal_agent.methods.diff_in_means.llm_assist module causal_agent.methods.difference_in_differences module causal_agent.methods.difference_in_differences.diagnostics module causal_agent.methods.difference_in_differences.estimator module causal_agent.methods.difference_in_differences.llm_assist module causal_agent.methods.difference_in_differences.utils module causal_agent.methods.generalized_propensity_score module causal_agent.methods.generalized_propensity_score.diagnostics module causal_agent.methods.generalized_propensity_score.estimator module causal_agent.methods.generalized_propensity_score.llm_assist module causal_agent.methods.instrumental_variable module causal_agent.methods.instrumental_variable.diagnostics module causal_agent.methods.instrumental_variable.estimator module causal_agent.methods.instrumental_variable.llm_assist module causal_agent.methods.linear_regression module causal_agent.methods.linear_regression.diagnostics module causal_agent.methods.linear_regression.estimator module causal_agent.methods.linear_regression.llm_assist module causal_agent.methods.propensity_score module causal_agent.methods.propensity_score.base module causal_agent.methods.propensity_score.diagnostics module

causal_agent.methods.propensity_score.llm_assist module causal_agent.methods.propensity_score.matching module causal_agent.methods.propensity_score.weighting module causal_agent.methods.regression_discontinuity module causal_agent.methods.regression_discontinuity.diagnostics module causal_agent.methods.regression_discontinuity.estimator module causal_agent.methods.regression_discontinuity.llm_assist module causal_agent.methods.utils module causal_agent.prompts module causal_agent.prompts.dataset_analysis_prompts module causal_agent.prompts.method_identification_prompts module causal_agent.prompts.prompts module causal_agent.prompts.regression_prompts module causal_agent.prompts.result_interpretation_prompts module causal_agent.synthetic module causal_agent.synthetic.generator module causal_agent.synthetic.prompts module causal_agent.synthetic.util module causal_agent.tools module causal_agent.tools.data_analyzer module causal_agent.tools.dataset_analyzer_tool module causal_agent.tools.explanation_generator_tool module causal_agent.tools.input_parser_tool module causal_agent.tools.method_executor_tool module causal_agent.tools.method_selector_tool module causal_agent.tools.method_validator_tool module causal_agent.tools.output_formatter_tool module causal_agent.tools.query_interpreter_tool module causal_agent.utils module causal_agent.utils.llm_helpers module CausalMethod (class in causal_agent.methods) (class in causal_agent.methods.causal_method) check_binary_treatment() (in module causal_agent.methods.utils) check_collinearity() (in module causal_agent.methods.utils) check_for_discontinuities() (in module causal_agent.methods.utils) check_overlap() (in module causal_agent.methods.utils) check_temporal_structure() (in module causal_agent.methods.utils) compute_smd() (in module causal_agent.components.query_interpreter) constraints (causal_agent.components.input_parser.ParsedQueryInfo attribute) construct() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) copy() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) covar (causal_agent.synthetic.generator.DataGenerator attribute) covariates_mentioned (causal_agent.components.input_parser.ParsedVariables attribute) create_post_indicator() (in module causal_agent.methods.difference_in_differences.utils) create_prompt() (in module causal_agent.synthetic.prompts) create_workflow_state_update() (in module causal_agent) (in module causal_agent.components) (in module causal_agent.components.state_manager)

D

data (causal_agent.synthetic.generator.DataGenerator attribute) DataAnalyzer (class in causal_agent.tools.data_analyzer) DataGenerator (class in causal_agent.synthetic.generator) dataset_analyzer_tool() (in module causal_agent.tools) (in module causal_agent.tools.dataset_analyzer_tool) dataset_path (causal_agent.components.input_parser.ExtractedPath attribute) dataset_path_mentioned (causal_agent.components.input_parser.ParsedQueryInfo attribute) DecisionTreeEngine (class in causal_agent.components.decision_tree) DecisionTreeLLMEngine (class in causal_agent.components.decision_tree_llm) detect_discontinuities() (in module causal_agent.components.dataset_analyzer) detect_temporal_structure() (in module causal_agent.components.dataset_analyzer) determine_optimal_caliper() (in module causal_agent.methods.propensity_score.llm_assist) determine_optimal_trim_threshold() (in module causal_agent.methods.propensity_score.llm_assist)

determine_optimal_weight_type() (in module causal_agent.methods.propensity_score.llm_assist) determine_treatment_period() (in module causal_agent.methods.difference_in_differences.llm_assist) determine_treatment_reference_level() (in module causal_agent.components.query_interpreter) dict() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) did_estimate_effect() (in module causal_agent.methods) DiDGenerator (class in causal_agent.synthetic) (class in causal_agent.synthetic.generator) dim_estimate_effect() (in module causal_agent.methods)

E

effect_estimate_rdd() (in module causal_agent.methods.regression_discontinuity.estimator) estimate_effect() (causal_agent.methods.causal_method.CausalMethod method) (causal_agent.methods.CausalMethod method) (in module causal_agent.methods.backdoor_adjustment.estimator) (in module causal_agent.methods.diff_in_means.estimator) (in module causal_agent.methods.difference_in_differences.estimator) (in module causal_agent.methods.instrumental_variable.estimator) (in module causal_agent.methods.linear_regression.estimator) (in module causal_agent.methods.propensity_score.matching) (in module causal_agent.methods.propensity_score.weighting) (in module causal_agent.methods.regression_discontinuity.estimator) estimate_effect_dowhy() (in module causal_agent.methods.regression_discontinuity.estimator) estimate_effect_fallback() (in module causal_agent.methods.regression_discontinuity.estimator) estimate_effect_gps() (in module causal_agent.methods) (in module causal_agent.methods.generalized_propensity_score.estimator)

estimate_matching_effect() (in module causal_agent.methods.propensity_score) estimate_propensity_scores() (in module causal_agent.methods.propensity_score) (in module causal_agent.methods.propensity_score.base) estimate_weighting_effect() (in module causal_agent.methods.propensity_score) explain() (causal_agent.methods.causal_method.CausalMethod method) (causal_agent.methods.CausalMethod method) explain_application() (in module causal_agent.components.explanation_generator) explain_assumptions() (in module causal_agent.components.explanation_generator) explain_limitations() (in module causal_agent.components.explanation_generator) explanation_generator_tool() (in module causal_agent.tools) (in module causal_agent.tools.explanation_generator_tool) export_info() (in module causal_agent.synthetic.util) extract_dataset_path() (in module causal_agent.components.input_parser) extract_dataset_path_regex() (in module causal_agent.components.input_parser) extract_properties_from_inputs() (in module causal_agent.tools.method_validator_tool) ExtractedPath (class in causal_agent.components.input_parser)

F

filter_question() (in module causal_agent.synthetic.prompts) find_potential_instruments() (in module causal_agent.components.dataset_analyzer) (in module causal_agent.methods.utils) format_did_results() (in module causal_agent.methods.difference_in_differences.estimator) format_iv_results() (in module causal_agent.methods.instrumental_variable.estimator) format_output() (in module causal_agent) (in module causal_agent.components) (in module causal_agent.components.output_formatter)

format_ps_results() (in module causal_agent.methods.propensity_score.base) from_orm() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) FrontDoorGenerator (class in causal_agent.synthetic.generator)

G

generate_code() (causal_agent.methods.causal_method.CausalMethod method) (causal_agent.methods.CausalMethod method) generate_covariates() (causal_agent.synthetic.DiDGenerator method) (causal_agent.synthetic.generator.DataGenerator method) (causal_agent.synthetic.generator.DiDGenerator method) (causal_agent.synthetic.generator.FrontDoorGenerator method) (causal_agent.synthetic.generator.IVGenerator method) (causal_agent.synthetic.generator.MultiTreatRCTGenerator method) (causal_agent.synthetic.generator.ObservationalDataGenerator method) (causal_agent.synthetic.generator.PSMGenerator method) (causal_agent.synthetic.generator.PSWGenerator method) (causal_agent.synthetic.generator.RCTGenerator method) (causal_agent.synthetic.generator.RDDGenerator method) (causal_agent.synthetic.IVGenerator method) (causal_agent.synthetic.MultiTreatRCTGenerator method) (causal_agent.synthetic.PSMGenerator method) (causal_agent.synthetic.PSWGenerator method) (causal_agent.synthetic.RCTGenerator method) (causal_agent.synthetic.RDDGenerator method) generate_data() (causal_agent.synthetic.DiDGenerator method) (causal_agent.synthetic.generator.DataGenerator method) (causal_agent.synthetic.generator.DiDGenerator method) (causal_agent.synthetic.generator.FrontDoorGenerator method) (causal_agent.synthetic.generator.IVGenerator method) (causal_agent.synthetic.generator.MultiTreatRCTGenerator method) (causal_agent.synthetic.generator.ObservationalDataGenerator method) (causal_agent.synthetic.generator.PSMGenerator method) (causal_agent.synthetic.generator.PSWGenerator method) (causal_agent.synthetic.generator.RCTGenerator method) (causal_agent.synthetic.generator.RDDGenerator method) (causal_agent.synthetic.IVGenerator method) (causal_agent.synthetic.MultiTreatRCTGenerator method) (causal_agent.synthetic.PSMGenerator method) (causal_agent.synthetic.PSWGenerator method) (causal_agent.synthetic.RCTGenerator method) (causal_agent.synthetic.RDDGenerator method)

generate_data_summary() (in module causal_agent.synthetic.prompts) generate_explanation() (in module causal_agent) (in module causal_agent.components) (in module causal_agent.components.explanation_generator) generate_interpretation_guide() (in module causal_agent.components.explanation_generator) get_columns_info() (in module causal_agent.utils.llm_helpers) get_llm_parameters() (in module causal_agent.methods.propensity_score.llm_assist) get_method_explanation() (in module causal_agent.components.explanation_generator) group_variable_name (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar attribute) grouping_vars (causal_agent.components.input_parser.ParsedVariables attribute)

H

heterogeneity (causal_agent.synthetic.generator.DataGenerator attribute)

I

identify_backdoor_set() (in module causal_agent.methods.backdoor_adjustment.llm_assist) identify_instrument_variable() (in module causal_agent.methods.instrumental_variable.llm_assist) identify_interaction_term() (in module causal_agent.components.query_interpreter) identify_time_variable() (in module causal_agent.methods.difference_in_differences.llm_assist) identify_treatment_group() (in module causal_agent.methods.difference_in_differences.llm_assist) infer_treatment_variable_type() (in module causal_agent.components.query_interpreter) info (causal_agent.synthetic.generator.DataGenerator attribute) input_parser_tool() (in module causal_agent.tools) (in module causal_agent.tools.input_parser_tool) instruments_mentioned (causal_agent.components.input_parser.ParsedVariables attribute) interpret_backdoor_results() (in module causal_agent.methods.backdoor_adjustment.llm_assist)

interpret_did_results() (in module causal_agent.methods.difference_in_differences.llm_assist) interpret_dim_results() (in module causal_agent.methods.diff_in_means.llm_assist) interpret_iv_results() (in module causal_agent.methods.instrumental_variable.llm_assist) interpret_lr_results() (in module causal_agent.methods.linear_regression.llm_assist) interpret_query() (in module causal_agent) (in module causal_agent.components) (in module causal_agent.components.query_interpreter) interpret_rdd_results() (in module causal_agent.methods.regression_discontinuity.llm_assist) iv_estimate_effect() (in module causal_agent.methods) IVGenerator (class in causal_agent.synthetic) (class in causal_agent.synthetic.generator)

J

json() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method)

L

llm_identify_temporal_and_unit_vars() (in module causal_agent.utils.llm_helpers) LLMGroupVar (class in causal_agent.methods.difference_in_differences.llm_assist)

LLMTimeVar (class in causal_agent.methods.difference_in_differences.llm_assist) LLMTreatmentPeriod (class in causal_agent.methods.difference_in_differences.llm_assist) lr_estimate_effect() (in module causal_agent.methods)

M

mean (causal_agent.synthetic.generator.DataGenerator attribute) method (causal_agent.synthetic.generator.DataGenerator attribute) method_executor_tool() (in module causal_agent.tools) (in module causal_agent.tools.method_executor_tool) method_selector_tool() (in module causal_agent.tools) (in module causal_agent.tools.method_selector_tool) method_validator_tool() (in module causal_agent.tools) (in module causal_agent.tools.method_validator_tool) model_computed_fields (causal_agent.components.input_parser.ExtractedPath attribute) (causal_agent.components.input_parser.ParsedQueryInfo attribute) (causal_agent.components.input_parser.ParsedVariables attribute) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar attribute) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar attribute) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod attribute) model_config (causal_agent.components.input_parser.ExtractedPath attribute) (causal_agent.components.input_parser.ParsedQueryInfo attribute) (causal_agent.components.input_parser.ParsedVariables attribute) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar attribute) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar attribute) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod attribute) model_construct() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) model_copy() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) model_dump() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) model_dump_json() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) model_extra (causal_agent.components.input_parser.ExtractedPath property) (causal_agent.components.input_parser.ParsedQueryInfo property) (causal_agent.components.input_parser.ParsedVariables property) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar property) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar property) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod property) model_fields (causal_agent.components.input_parser.ExtractedPath attribute) (causal_agent.components.input_parser.ParsedQueryInfo attribute) (causal_agent.components.input_parser.ParsedVariables attribute) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar attribute) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar attribute) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod attribute) model_fields_set (causal_agent.components.input_parser.ExtractedPath property) (causal_agent.components.input_parser.ParsedQueryInfo property) (causal_agent.components.input_parser.ParsedVariables property) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar property) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar property) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod property) model_json_schema() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) model_parametrized_name() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) model_post_init() (causal_agent.components.input_parser.ExtractedPath method) (causal_agent.components.input_parser.ParsedQueryInfo method) (causal_agent.components.input_parser.ParsedVariables method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod method) model_rebuild() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) model_validate() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method)

model_validate_json() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) model_validate_strings() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) module causal_agent causal_agent.components causal_agent.components.dataset_analyzer causal_agent.components.decision_tree causal_agent.components.decision_tree_llm causal_agent.components.explanation_generator causal_agent.components.input_parser causal_agent.components.method_validator causal_agent.components.output_formatter causal_agent.components.query_interpreter causal_agent.components.state_manager causal_agent.methods causal_agent.methods.backdoor_adjustment causal_agent.methods.backdoor_adjustment.diagnostics causal_agent.methods.backdoor_adjustment.estimator causal_agent.methods.backdoor_adjustment.llm_assist causal_agent.methods.causal_method causal_agent.methods.diff_in_means causal_agent.methods.diff_in_means.diagnostics causal_agent.methods.diff_in_means.estimator causal_agent.methods.diff_in_means.llm_assist causal_agent.methods.difference_in_differences causal_agent.methods.difference_in_differences.diagnostics causal_agent.methods.difference_in_differences.estimator causal_agent.methods.difference_in_differences.llm_assist causal_agent.methods.difference_in_differences.utils causal_agent.methods.generalized_propensity_score causal_agent.methods.generalized_propensity_score.diagnostics causal_agent.methods.generalized_propensity_score.estimator causal_agent.methods.generalized_propensity_score.llm_assist causal_agent.methods.instrumental_variable causal_agent.methods.instrumental_variable.diagnostics causal_agent.methods.instrumental_variable.estimator causal_agent.methods.instrumental_variable.llm_assist causal_agent.methods.linear_regression causal_agent.methods.linear_regression.diagnostics causal_agent.methods.linear_regression.estimator causal_agent.methods.linear_regression.llm_assist causal_agent.methods.propensity_score causal_agent.methods.propensity_score.base causal_agent.methods.propensity_score.diagnostics causal_agent.methods.propensity_score.llm_assist causal_agent.methods.propensity_score.matching causal_agent.methods.propensity_score.weighting causal_agent.methods.regression_discontinuity causal_agent.methods.regression_discontinuity.diagnostics causal_agent.methods.regression_discontinuity.estimator causal_agent.methods.regression_discontinuity.llm_assist causal_agent.methods.utils causal_agent.prompts causal_agent.prompts.dataset_analysis_prompts causal_agent.prompts.method_identification_prompts causal_agent.prompts.prompts causal_agent.prompts.regression_prompts causal_agent.prompts.result_interpretation_prompts causal_agent.synthetic causal_agent.synthetic.generator causal_agent.synthetic.prompts causal_agent.synthetic.util causal_agent.tools causal_agent.tools.data_analyzer causal_agent.tools.dataset_analyzer_tool causal_agent.tools.explanation_generator_tool causal_agent.tools.input_parser_tool causal_agent.tools.method_executor_tool causal_agent.tools.method_selector_tool causal_agent.tools.method_validator_tool causal_agent.tools.output_formatter_tool causal_agent.tools.query_interpreter_tool causal_agent.utils causal_agent.utils.llm_helpers MultiTreatRCTGenerator (class in causal_agent.synthetic) (class in causal_agent.synthetic.generator)

N

n_continuous_covars (causal_agent.synthetic.generator.DataGenerator attribute) n_covars (causal_agent.synthetic.generator.DataGenerator attribute)

n_observations (causal_agent.synthetic.generator.DataGenerator attribute) n_treatments (causal_agent.synthetic.generator.DataGenerator attribute)

O

ObservationalDataGenerator (class in causal_agent.synthetic.generator) outcome (causal_agent.components.input_parser.ParsedVariables attribute)

output_formatter_tool() (in module causal_agent.tools) (in module causal_agent.tools.output_formatter_tool)

P

parse_file() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) parse_input() (in module causal_agent) (in module causal_agent.components) (in module causal_agent.components.input_parser) parse_obj() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) parse_raw() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method)

ParsedQueryInfo (class in causal_agent.components.input_parser) ParsedVariables (class in causal_agent.components.input_parser) plot_balance() (in module causal_agent.methods.propensity_score) (in module causal_agent.methods.propensity_score.diagnostics) plot_covariate_balance() (in module causal_agent.methods.utils) plot_overlap() (in module causal_agent.methods.propensity_score) (in module causal_agent.methods.propensity_score.diagnostics) plot_propensity_overlap() (in module causal_agent.methods.utils) preprocess_data() (in module causal_agent.methods.utils) process_llm_response() (in module causal_agent.utils.llm_helpers) psm_estimate_effect() (in module causal_agent.methods) PSMGenerator (class in causal_agent.synthetic) (class in causal_agent.synthetic.generator) psw_estimate_effect() (in module causal_agent.methods) PSWGenerator (class in causal_agent.synthetic) (class in causal_agent.synthetic.generator)

Q

query_interpreter_tool() (in module causal_agent.tools) (in module causal_agent.tools.query_interpreter_tool)

query_type (causal_agent.components.input_parser.ParsedQueryInfo attribute)

R

RCTGenerator (class in causal_agent.synthetic) (class in causal_agent.synthetic.generator) rdd_estimate_effect() (in module causal_agent.methods) RDDGenerator (class in causal_agent.synthetic) (class in causal_agent.synthetic.generator) recommend_alternative() (in module causal_agent.components.method_validator) rule_based_select_method() (in module causal_agent.components.decision_tree)

run_backdoor_diagnostics() (in module causal_agent.methods.backdoor_adjustment.diagnostics) run_causal_analysis() (in module causal_agent), [1] run_dim_diagnostics() (in module causal_agent.methods.diff_in_means.diagnostics) run_iv_diagnostics() (in module causal_agent.methods.instrumental_variable.diagnostics) run_lr_diagnostics() (in module causal_agent.methods.linear_regression.diagnostics) run_overidentification_test() (in module causal_agent.methods.instrumental_variable.diagnostics) run_placebo_test() (in module causal_agent.methods.difference_in_differences.diagnostics) run_rdd_diagnostics() (in module causal_agent.methods.regression_discontinuity.diagnostics)

S

save_data() (causal_agent.synthetic.DiDGenerator method) (causal_agent.synthetic.generator.DataGenerator method) (causal_agent.synthetic.generator.DiDGenerator method) (causal_agent.synthetic.generator.FrontDoorGenerator method) (causal_agent.synthetic.generator.IVGenerator method) (causal_agent.synthetic.generator.MultiTreatRCTGenerator method) (causal_agent.synthetic.generator.ObservationalDataGenerator method) (causal_agent.synthetic.generator.PSMGenerator method) (causal_agent.synthetic.generator.PSWGenerator method) (causal_agent.synthetic.generator.RCTGenerator method) (causal_agent.synthetic.generator.RDDGenerator method) (causal_agent.synthetic.IVGenerator method) (causal_agent.synthetic.MultiTreatRCTGenerator method) (causal_agent.synthetic.PSMGenerator method) (causal_agent.synthetic.PSWGenerator method) (causal_agent.synthetic.RCTGenerator method) (causal_agent.synthetic.RDDGenerator method) schema() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method)

schema_json() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) seed (causal_agent.synthetic.generator.DataGenerator attribute) select_method() (causal_agent.components.decision_tree.DecisionTreeEngine method) (in module causal_agent.components) (in module causal_agent.components.decision_tree) select_method_llm() (causal_agent.components.decision_tree_llm.DecisionTreeLLMEngine method) select_propensity_model() (in module causal_agent.methods.propensity_score.base) suggest_dose_response_t_values() (in module causal_agent.methods.generalized_propensity_score.llm_assist) suggest_lr_covariates() (in module causal_agent.methods.linear_regression.llm_assist) suggest_outcome_model_spec() (in module causal_agent.methods.generalized_propensity_score.llm_assist) suggest_rdd_parameters() (in module causal_agent.methods.regression_discontinuity.llm_assist) suggest_treatment_model_spec() (in module causal_agent.methods.generalized_propensity_score.llm_assist)

T

test_data() (causal_agent.synthetic.DiDGenerator method) (causal_agent.synthetic.generator.DataGenerator method) (causal_agent.synthetic.generator.DiDGenerator method) (causal_agent.synthetic.generator.FrontDoorGenerator method) (causal_agent.synthetic.generator.IVGenerator method) (causal_agent.synthetic.generator.MultiTreatRCTGenerator method) (causal_agent.synthetic.generator.ObservationalDataGenerator method) (causal_agent.synthetic.generator.PSMGenerator method) (causal_agent.synthetic.generator.PSWGenerator method) (causal_agent.synthetic.generator.RCTGenerator method) (causal_agent.synthetic.generator.RDDGenerator method) (causal_agent.synthetic.IVGenerator method) (causal_agent.synthetic.MultiTreatRCTGenerator method) (causal_agent.synthetic.PSMGenerator method) (causal_agent.synthetic.PSWGenerator method) (causal_agent.synthetic.RCTGenerator method) (causal_agent.synthetic.RDDGenerator method)

test_parallel_trends() (in module causal_agent.methods.utils) time_variable_name (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar attribute) treatment (causal_agent.components.input_parser.ParsedVariables attribute) treatment_start_period (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod attribute) true_effect (causal_agent.synthetic.generator.DataGenerator attribute) twfe_model() (causal_agent.synthetic.DiDGenerator method) (causal_agent.synthetic.generator.DiDGenerator method)

U

update_forward_refs() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method)

V

validate() (causal_agent.components.input_parser.ExtractedPath class method) (causal_agent.components.input_parser.ParsedQueryInfo class method) (causal_agent.components.input_parser.ParsedVariables class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMGroupVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTimeVar class method) (causal_agent.methods.difference_in_differences.llm_assist.LLMTreatmentPeriod class method) validate_assumptions() (causal_agent.methods.causal_method.CausalMethod method) (causal_agent.methods.CausalMethod method) validate_backdoor_adjustment() (in module causal_agent.components.method_validator) validate_difference_in_differences() (in module causal_agent.components.method_validator)

validate_instrument_assumptions_qualitative() (in module causal_agent.methods.instrumental_variable.llm_assist) validate_instrumental_variable() (in module causal_agent.components.method_validator) validate_method() (in module causal_agent) (in module causal_agent.components) (in module causal_agent.components.method_validator) validate_parallel_trends() (in module causal_agent.methods.difference_in_differences.diagnostics) validate_propensity_score_matching() (in module causal_agent.components.method_validator) validate_regression_adjustment() (in module causal_agent.components.method_validator) validate_regression_discontinuity() (in module causal_agent.components.method_validator) variables (causal_agent.components.input_parser.ParsedQueryInfo attribute)