Documentation:
API Reference:
add_history_messages()
AgentState
AgentState.history_messages
AgentState.agent_next_node
AgentState.sends
AgentState.sql
AgentState.final_answer
AgentState.messages
SQLGraphState
SQLGraphState.rewrite_question
SQLGraphState.tables
SQLGraphState.info_entities
SQLGraphState.sql
SQLGraphState.sql_retry_count
SQLGraphState.sql_execution_result
SQLGraphState.schema_info
SQLGraphState.data
SQLGraphState.previous_sql_errors
SQLGraphState.visualization_dsl
SQLGraphState.messages
InputState
InputState.messages
OutputState
OutputState.messages
SQLOutputState
SQLOutputState.rewrite_question
SQLOutputState.tables
SQLOutputState.sql
SQLOutputState.schema_info
SQLOutputState.data
SQLOutputState.visualization_dsl
SQLOutputState.messages
log()
get_text_from_content()
get_text_from_message_chunk()
extract_json_from_answer()
get_report_download_response()
create_vector_db()
recover_incomplete_tool_calls()
Config
Config.organization
Config.dialect
Config.default_llm
Config.embedding_model
Config.text2sql_llm
Config.bi_config
Config.data_warehouse_config
Config.model_config
Config.visualization_mode
Config.context_config
Config.timeseries_forecasting_service_url
Config.from_dict()
ConfigLoader
ConfigLoader.llm_configs
ConfigLoader.get()
ConfigLoader.load()
ConfigLoader.load_bi_config()
ConfigLoader.set()
CatalogStore
CatalogStore.get_data_warehouse_config()
CatalogStore.get_sql_engine()
CatalogStore.get_database_list()
CatalogStore.get_table_list()
CatalogStore.get_column_list()
CatalogStore.get_table_information()
CatalogStore.get_sql_examples()
CatalogStore.get_table_selection_examples()
CatalogStore.save_table_information()
CatalogStore.save_table_sql_examples()
CatalogStore.save_table_selection_examples()
CatalogStore.check_exists()
split_db_table_name()
FileSystemCatalogStore
DataCatalogLoader
DataCatalogLoader.__init__()
DataCatalogLoader.get_tables_and_columns()
DataCatalogLoader.get_table_indexes()
DataCatalogLoader.get_foreign_keys()
DataCatalogLoader.save_to_catalog_store()
load_catalog_from_data_warehouse()
Natural language to SQL conversion pipeline with schema linking and prompt engineering.