causal_agent.interpret_query

causal_agent.interpret_query(query_info, dataset_analysis, dataset_description=None)[source]

Interpret query using hybrid heuristic/LLM approach to identify variables.

Parameters:

  • query_info (Dict[str, Any]) – Information extracted from the user’s query (text, hints).

  • dataset_analysis (Dict[str, Any]) – Information about the dataset structure (columns, types, etc.).

  • dataset_description (str | None) – Optional textual description of the dataset.

  • llm – Optional language model instance.

Returns:

Dict containing identified variables (treatment, outcome, covariates, etc., and is_rct).

Return type:

Dict[str, Any]