• Home
• Index

Index

_

__aenter__() (airflow.providers.common.ai.toolsets.mcp.MCPToolset method)

__aexit__() (airflow.providers.common.ai.toolsets.mcp.MCPToolset method) __version__ (in module airflow.providers.common.ai)

A

agent_params (airflow.providers.common.ai.operators.agent.AgentOperator attribute) (airflow.providers.common.ai.operators.llm.LLMOperator attribute) agent_task() (in module airflow.providers.common.ai.decorators.agent) AgentOperator (class in airflow.providers.common.ai.operators.agent) airflow.providers.common.ai module airflow.providers.common.ai.decorators module airflow.providers.common.ai.decorators.agent module airflow.providers.common.ai.decorators.llm module airflow.providers.common.ai.decorators.llm_branch module airflow.providers.common.ai.decorators.llm_schema_compare module airflow.providers.common.ai.decorators.llm_sql module airflow.providers.common.ai.example_dags module airflow.providers.common.ai.example_dags.example_agent module airflow.providers.common.ai.example_dags.example_llm module airflow.providers.common.ai.example_dags.example_llm_analysis_pipeline module airflow.providers.common.ai.example_dags.example_llm_branch module airflow.providers.common.ai.example_dags.example_llm_classification module airflow.providers.common.ai.example_dags.example_llm_schema_compare module airflow.providers.common.ai.example_dags.example_llm_sql module airflow.providers.common.ai.example_dags.example_mcp module airflow.providers.common.ai.example_dags.example_pydantic_ai_hook module airflow.providers.common.ai.get_provider_info module airflow.providers.common.ai.hooks module airflow.providers.common.ai.hooks.mcp module

airflow.providers.common.ai.hooks.pydantic_ai module airflow.providers.common.ai.mixins module airflow.providers.common.ai.mixins.approval module airflow.providers.common.ai.operators module airflow.providers.common.ai.operators.agent module airflow.providers.common.ai.operators.llm module airflow.providers.common.ai.operators.llm_branch module airflow.providers.common.ai.operators.llm_schema_compare module airflow.providers.common.ai.operators.llm_sql module airflow.providers.common.ai.toolsets module airflow.providers.common.ai.toolsets.datafusion module airflow.providers.common.ai.toolsets.hook module airflow.providers.common.ai.toolsets.logging module airflow.providers.common.ai.toolsets.mcp module airflow.providers.common.ai.toolsets.sql module airflow.providers.common.ai.triggers module airflow.providers.common.ai.utils module airflow.providers.common.ai.utils.logging module airflow.providers.common.ai.utils.sql_validation module allow_modifications (airflow.providers.common.ai.mixins.approval.DeferForApprovalProtocol attribute) (airflow.providers.common.ai.operators.llm.LLMOperator attribute) allow_multiple_branches (airflow.providers.common.ai.operators.llm_branch.LLMBranchOperator attribute) allowed_sql_types (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator attribute) approval_timeout (airflow.providers.common.ai.mixins.approval.DeferForApprovalProtocol attribute) (airflow.providers.common.ai.operators.llm.LLMOperator attribute) APPROVE (airflow.providers.common.ai.mixins.approval.LLMApprovalMixin attribute)

C

call_tool() (airflow.providers.common.ai.toolsets.datafusion.DataFusionToolset method) (airflow.providers.common.ai.toolsets.hook.HookToolset method) (airflow.providers.common.ai.toolsets.logging.LoggingToolset method) (airflow.providers.common.ai.toolsets.mcp.MCPToolset method) (airflow.providers.common.ai.toolsets.sql.SQLToolset method) column (airflow.providers.common.ai.operators.llm_schema_compare.SchemaMismatch attribute)

compatible (airflow.providers.common.ai.operators.llm_schema_compare.SchemaCompareResult attribute) conn_name_attr (airflow.providers.common.ai.hooks.mcp.MCPHook attribute) (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook attribute) conn_type (airflow.providers.common.ai.hooks.mcp.MCPHook attribute) (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook attribute) context_strategy (airflow.providers.common.ai.operators.llm_schema_compare.LLMSchemaCompareOperator attribute) create_agent() (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook method)

D

data_sources (airflow.providers.common.ai.operators.llm_schema_compare.LLMSchemaCompareOperator attribute) DataFusionToolset (class in airflow.providers.common.ai.toolsets.datafusion) datasource_config (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator attribute) db_conn_id (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator attribute) db_conn_ids (airflow.providers.common.ai.operators.llm_schema_compare.LLMSchemaCompareOperator attribute) db_hook (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator property) DEFAULT_ALLOWED_TYPES (in module airflow.providers.common.ai.utils.sql_validation)

default_conn_name (airflow.providers.common.ai.hooks.mcp.MCPHook attribute) (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook attribute) DEFAULT_SYSTEM_PROMPT (in module airflow.providers.common.ai.operators.llm_schema_compare) defer (airflow.providers.common.ai.mixins.approval.DeferForApprovalProtocol attribute) defer_for_approval() (airflow.providers.common.ai.mixins.approval.LLMApprovalMixin method) DeferForApprovalProtocol (class in airflow.providers.common.ai.mixins.approval) description (airflow.providers.common.ai.operators.llm_schema_compare.SchemaMismatch attribute) dialect (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator attribute)

E

enable_tool_logging (airflow.providers.common.ai.operators.agent.AgentOperator attribute) example_agent_chain() (in module airflow.providers.common.ai.example_dags.example_agent) example_agent_decorator() (in module airflow.providers.common.ai.example_dags.example_agent) example_agent_operator_hook() (in module airflow.providers.common.ai.example_dags.example_agent) example_agent_operator_sql() (in module airflow.providers.common.ai.example_dags.example_agent) example_agent_structured_output() (in module airflow.providers.common.ai.example_dags.example_agent) example_llm_analysis_pipeline() (in module airflow.providers.common.ai.example_dags.example_llm_analysis_pipeline) example_llm_branch_decorator() (in module airflow.providers.common.ai.example_dags.example_llm_branch) example_llm_branch_decorator_multi() (in module airflow.providers.common.ai.example_dags.example_llm_branch) example_llm_branch_multi() (in module airflow.providers.common.ai.example_dags.example_llm_branch) example_llm_branch_operator() (in module airflow.providers.common.ai.example_dags.example_llm_branch) example_llm_classification() (in module airflow.providers.common.ai.example_dags.example_llm_classification) example_llm_decorator() (in module airflow.providers.common.ai.example_dags.example_llm) example_llm_decorator_structured() (in module airflow.providers.common.ai.example_dags.example_llm) example_llm_operator() (in module airflow.providers.common.ai.example_dags.example_llm) example_llm_operator_agent_params() (in module airflow.providers.common.ai.example_dags.example_llm) example_llm_operator_approval() (in module airflow.providers.common.ai.example_dags.example_llm) example_llm_operator_structured() (in module airflow.providers.common.ai.example_dags.example_llm) example_llm_schema_compare_basic() (in module airflow.providers.common.ai.example_dags.example_llm_schema_compare) example_llm_schema_compare_conditional() (in module airflow.providers.common.ai.example_dags.example_llm_schema_compare)

example_llm_schema_compare_decorator() (in module airflow.providers.common.ai.example_dags.example_llm_schema_compare) example_llm_schema_compare_full_context() (in module airflow.providers.common.ai.example_dags.example_llm_schema_compare) example_llm_schema_compare_with_object_storage() (in module airflow.providers.common.ai.example_dags.example_llm_schema_compare) example_llm_sql_approval() (in module airflow.providers.common.ai.example_dags.example_llm_sql) example_llm_sql_basic() (in module airflow.providers.common.ai.example_dags.example_llm_sql) example_llm_sql_decorator() (in module airflow.providers.common.ai.example_dags.example_llm_sql) example_llm_sql_expand() (in module airflow.providers.common.ai.example_dags.example_llm_sql) example_llm_sql_schema_introspection() (in module airflow.providers.common.ai.example_dags.example_llm_sql) example_llm_sql_with_object_storage() (in module airflow.providers.common.ai.example_dags.example_llm_sql) example_mcp_multiple_servers() (in module airflow.providers.common.ai.example_dags.example_mcp) example_mcp_toolset() (in module airflow.providers.common.ai.example_dags.example_mcp) example_pydantic_ai_hook() (in module airflow.providers.common.ai.example_dags.example_pydantic_ai_hook) example_pydantic_ai_structured_output() (in module airflow.providers.common.ai.example_dags.example_pydantic_ai_hook) execute() (airflow.providers.common.ai.operators.agent.AgentOperator method) (airflow.providers.common.ai.operators.llm.LLMOperator method) (airflow.providers.common.ai.operators.llm_branch.LLMBranchOperator method) (airflow.providers.common.ai.operators.llm_schema_compare.LLMSchemaCompareOperator method) (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator method) execute_complete() (airflow.providers.common.ai.mixins.approval.LLMApprovalMixin method) (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator method)

G

get_conn() (airflow.providers.common.ai.hooks.mcp.MCPHook method) (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook method) get_provider_info() (in module airflow.providers.common.ai.get_provider_info) get_tools() (airflow.providers.common.ai.toolsets.datafusion.DataFusionToolset method) (airflow.providers.common.ai.toolsets.hook.HookToolset method) (airflow.providers.common.ai.toolsets.mcp.MCPToolset method) (airflow.providers.common.ai.toolsets.sql.SQLToolset method)

get_ui_field_behaviour() (airflow.providers.common.ai.hooks.mcp.MCPHook static method) (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook static method)

H

hook_name (airflow.providers.common.ai.hooks.mcp.MCPHook attribute) (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook attribute)

HookToolset (class in airflow.providers.common.ai.toolsets.hook)

I

id (airflow.providers.common.ai.toolsets.datafusion.DataFusionToolset property) (airflow.providers.common.ai.toolsets.hook.HookToolset property) (airflow.providers.common.ai.toolsets.mcp.MCPToolset property) (airflow.providers.common.ai.toolsets.sql.SQLToolset property)

inherits_from_skipmixin (airflow.providers.common.ai.operators.llm_branch.LLMBranchOperator attribute)

L

llm_branch_task() (in module airflow.providers.common.ai.decorators.llm_branch) llm_conn_id (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook attribute) (airflow.providers.common.ai.operators.agent.AgentOperator attribute) (airflow.providers.common.ai.operators.llm.LLMOperator attribute) llm_hook (airflow.providers.common.ai.operators.agent.AgentOperator property) (airflow.providers.common.ai.operators.llm.LLMOperator property) llm_schema_compare_task() (in module airflow.providers.common.ai.decorators.llm_schema_compare) llm_sql_task() (in module airflow.providers.common.ai.decorators.llm_sql) llm_task() (in module airflow.providers.common.ai.decorators.llm)

LLMApprovalMixin (class in airflow.providers.common.ai.mixins.approval) LLMBranchOperator (class in airflow.providers.common.ai.operators.llm_branch) LLMOperator (class in airflow.providers.common.ai.operators.llm) LLMSchemaCompareOperator (class in airflow.providers.common.ai.operators.llm_schema_compare) LLMSQLQueryOperator (class in airflow.providers.common.ai.operators.llm_sql) log (in module airflow.providers.common.ai.mixins.approval) (in module airflow.providers.common.ai.toolsets.datafusion) log_run_summary() (in module airflow.providers.common.ai.utils.logging) logger (airflow.providers.common.ai.toolsets.logging.LoggingToolset attribute) LoggingToolset (class in airflow.providers.common.ai.toolsets.logging)

M

mcp_conn_id (airflow.providers.common.ai.hooks.mcp.MCPHook attribute) MCPHook (class in airflow.providers.common.ai.hooks.mcp) MCPToolset (class in airflow.providers.common.ai.toolsets.mcp) migration_query (airflow.providers.common.ai.operators.llm_schema_compare.SchemaMismatch attribute) mismatches (airflow.providers.common.ai.operators.llm_schema_compare.SchemaCompareResult attribute) model_id (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook attribute) (airflow.providers.common.ai.operators.agent.AgentOperator attribute) (airflow.providers.common.ai.operators.llm.LLMOperator attribute) module airflow.providers.common.ai airflow.providers.common.ai.decorators airflow.providers.common.ai.decorators.agent airflow.providers.common.ai.decorators.llm airflow.providers.common.ai.decorators.llm_branch airflow.providers.common.ai.decorators.llm_schema_compare airflow.providers.common.ai.decorators.llm_sql airflow.providers.common.ai.example_dags airflow.providers.common.ai.example_dags.example_agent airflow.providers.common.ai.example_dags.example_llm airflow.providers.common.ai.example_dags.example_llm_analysis_pipeline airflow.providers.common.ai.example_dags.example_llm_branch airflow.providers.common.ai.example_dags.example_llm_classification airflow.providers.common.ai.example_dags.example_llm_schema_compare airflow.providers.common.ai.example_dags.example_llm_sql airflow.providers.common.ai.example_dags.example_mcp airflow.providers.common.ai.example_dags.example_pydantic_ai_hook airflow.providers.common.ai.get_provider_info airflow.providers.common.ai.hooks airflow.providers.common.ai.hooks.mcp airflow.providers.common.ai.hooks.pydantic_ai airflow.providers.common.ai.mixins airflow.providers.common.ai.mixins.approval airflow.providers.common.ai.operators airflow.providers.common.ai.operators.agent airflow.providers.common.ai.operators.llm airflow.providers.common.ai.operators.llm_branch airflow.providers.common.ai.operators.llm_schema_compare airflow.providers.common.ai.operators.llm_sql airflow.providers.common.ai.toolsets airflow.providers.common.ai.toolsets.datafusion airflow.providers.common.ai.toolsets.hook airflow.providers.common.ai.toolsets.logging airflow.providers.common.ai.toolsets.mcp airflow.providers.common.ai.toolsets.sql airflow.providers.common.ai.triggers airflow.providers.common.ai.utils airflow.providers.common.ai.utils.logging airflow.providers.common.ai.utils.sql_validation tests.system.common.ai

O

output_type (airflow.providers.common.ai.operators.agent.AgentOperator attribute) (airflow.providers.common.ai.operators.llm.LLMOperator attribute)

OutputT (in module airflow.providers.common.ai.hooks.pydantic_ai)

P

prompt (airflow.providers.common.ai.mixins.approval.DeferForApprovalProtocol attribute) (airflow.providers.common.ai.operators.agent.AgentOperator attribute) (airflow.providers.common.ai.operators.llm.LLMOperator attribute)

PydanticAIHook (class in airflow.providers.common.ai.hooks.pydantic_ai)

R

REJECT (airflow.providers.common.ai.mixins.approval.LLMApprovalMixin attribute)

require_approval (airflow.providers.common.ai.operators.llm.LLMOperator attribute)

S

schema_context (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator attribute) SchemaCompareResult (class in airflow.providers.common.ai.operators.llm_schema_compare) SchemaMismatch (class in airflow.providers.common.ai.operators.llm_schema_compare) severity (airflow.providers.common.ai.operators.llm_schema_compare.SchemaMismatch attribute) source (airflow.providers.common.ai.operators.llm_schema_compare.SchemaMismatch attribute) source_type (airflow.providers.common.ai.operators.llm_schema_compare.SchemaMismatch attribute)

SQLSafetyError SQLToolset (class in airflow.providers.common.ai.toolsets.sql) suggested_action (airflow.providers.common.ai.operators.llm_schema_compare.SchemaMismatch attribute) summary (airflow.providers.common.ai.operators.llm_schema_compare.SchemaCompareResult attribute) system_prompt (airflow.providers.common.ai.operators.agent.AgentOperator attribute) (airflow.providers.common.ai.operators.llm.LLMOperator attribute) (airflow.providers.common.ai.operators.llm_schema_compare.LLMSchemaCompareOperator attribute)

T

table_names (airflow.providers.common.ai.operators.llm_schema_compare.LLMSchemaCompareOperator attribute) (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator attribute) target (airflow.providers.common.ai.operators.llm_schema_compare.SchemaMismatch attribute) target_type (airflow.providers.common.ai.operators.llm_schema_compare.SchemaMismatch attribute) task_id (airflow.providers.common.ai.mixins.approval.DeferForApprovalProtocol attribute) template_fields (airflow.providers.common.ai.operators.agent.AgentOperator attribute) (airflow.providers.common.ai.operators.llm.LLMOperator attribute) (airflow.providers.common.ai.operators.llm_branch.LLMBranchOperator attribute) (airflow.providers.common.ai.operators.llm_schema_compare.LLMSchemaCompareOperator attribute) (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator attribute)

test_connection() (airflow.providers.common.ai.hooks.mcp.MCPHook method) (airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIHook method) tests.system.common.ai module tool_prefix (airflow.providers.common.ai.hooks.mcp.MCPHook attribute) toolsets (airflow.providers.common.ai.operators.agent.AgentOperator attribute)

V

validate_sql (airflow.providers.common.ai.operators.llm_sql.LLMSQLQueryOperator attribute)

validate_sql() (in module airflow.providers.common.ai.utils.sql_validation)

W

wrap_toolsets_for_logging() (in module airflow.providers.common.ai.utils.logging)

PreviousNext

Was this entry helpful?