File size: 1,219 Bytes
1721aea
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
"""
Auto Causal module for causal inference.

This module provides automated causal inference capabilities
through a pipeline that selects and applies appropriate causal methods.
"""

__version__ = "0.1.0"

# Import components
from auto_causal.components import (
    parse_input,
    analyze_dataset,
    interpret_query,
    validate_method,
    generate_explanation,
    format_output,
    create_workflow_state_update
)

# Import tools
from auto_causal.tools import (
    input_parser_tool,
    dataset_analyzer_tool,
    query_interpreter_tool,
    method_selector_tool,
    method_validator_tool,
    method_executor_tool,
    explanation_generator_tool,
    output_formatter_tool
)

# Import the main agent function
from .agent import run_causal_analysis

# Remove backward compatibility for old pipeline
# try:
#     from .pipeline import CausalInferencePipeline
# except ImportError:
#     # Define a placeholder class if the old pipeline doesn't exist
#     class CausalInferencePipeline:
#         """Placeholder for CausalInferencePipeline."""
#         
#         def __init__(self, *args, **kwargs):
#             pass

# Update __all__ to export the main function
__all__ = [
    'run_causal_analysis'
]