eladkal commented on code in PR #76230:
URL: https://github.com/apache/airflow/pull/76230#discussion_r4436154291
##########
providers/common/ai/src/airflow/providers/common/ai/operators/llm_sql.py:
##########
@@ -80,13 +83,21 @@ class LLMSQLQueryOperator(LLMOperator):
:param db_conn_id: Connection ID for database schema introspection.
The connection must resolve to a ``DbApiHook``.
:param table_names: Tables to include in the LLM's schema context.
- Used with ``db_conn_id`` for automatic introspection.
+ Used with ``db_conn_id`` for automatic introspection. This does not
restrict
+ the tables the generated SQL may reference; use ``allowed_tables`` for
that.
:param schema_context: Manual schema context string. When provided,
this is used instead of ``db_conn_id`` introspection.
:param validate_sql: Whether to validate generated SQL via AST parsing.
Default ``True`` (safe by default).
:param allowed_sql_types: SQL statement types to allow.
Default: ``(Select, Union, Intersect, Except)``.
+ :param allowed_tables: Tables the generated SQL may reference, as
``table`` or
Review Comment:
I wonder if we need something more generic? One might want to allow list a
schema.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
