Q ig@dZddlZddlmZmZmZGddZdS)z=SQL Query class for executing SQL-like queries against Redis.N)AnyDictOptionalceZdZdZd dedeeeeffdZdedeeefdefdZ d d eed edefd Z dS)SQLQueryaA query class that translates SQL-like syntax into Redis queries. This class allows users to write SQL SELECT statements that are automatically translated into Redis FT.SEARCH or FT.AGGREGATE commands. .. code-block:: python from redisvl.query import SQLQuery from redisvl.index import SearchIndex index = SearchIndex.from_existing("products", redis_url="redis://localhost:6379") sql_query = SQLQuery(''' SELECT title, price, category FROM products WHERE category = 'electronics' AND price < 100 ''') results = index.query(sql_query) Note: Requires the optional `sql-redis` package. Install with: ``pip install redisvl[sql-redis]`` Nsqlparamsc&||_|pi|_dS)aInitialize a SQLQuery. Args: sql: The SQL SELECT statement to execute. params: Optional dictionary of parameters for parameterized queries. Useful for passing vector data for similarity searches. N)rr )selfrr s C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\redisvl/query/sql.py__init__zSQLQuery.__init__!sl returncL|s|Stjd|}g}|D]}|dr|dd}||vr||}t|tt fr#|t|lt|tr0|dd}|d|d||||||d |S)aSubstitute parameter placeholders in SQL with actual values. Uses token-based approach: splits SQL on :param patterns, then rebuilds with substituted values. This prevents partial matching (e.g., :id won't match inside :product_id) and is faster than regex at scale. Args: sql: The SQL string with :param placeholders. params: Dictionary mapping parameter names to values. Returns: SQL string with parameters substituted. Note: - String values are wrapped in single quotes with proper escaping - Numeric values are converted to strings - Bytes values (e.g., vectors) are NOT substituted here z(:[a-zA-Z_][a-zA-Z0-9_]*):N'z'') resplit startswith isinstanceintfloatappendstrreplacejoin) r rr tokensresulttokenkeyvalueescapeds r _substitute_paramszSQLQuery._substitute_params,s4& J6<< % %E$$ %ABBi&=="3KE!%#u66- c%jj1111#E3//-"'--T":": n'nnn5555 e,,,,MM%(((( e$$$$wwvrredis://localhost:6379 redis_client redis_urlcx ddlm}ddlm}n#t$rt dwxYw|ddlm}||}||}|||}| |j |j }| |} | S)a?Translate the SQL query to a Redis command string. This method uses the sql-redis translator to convert the SQL statement into the equivalent Redis FT.SEARCH or FT.AGGREGATE command. Args: redis_client: A Redis client connection used to load index schemas. If not provided, a connection will be created using redis_url. redis_url: The Redis URL to connect to if redis_client is not provided. Defaults to "redis://localhost:6379". Returns: The Redis command string (e.g., 'FT.SEARCH products "@category:{electronics}"'). Raises: ImportError: If sql-redis package is not installed. Example: .. code-block:: python from redisvl.query import SQLQuery sql_query = SQLQuery("SELECT * FROM products WHERE category = 'electronics'") # Using redis_url redis_cmd = sql_query.redis_query_string(redis_url="redis://localhost:6379") # Or using an existing client from redis import Redis client = Redis() redis_cmd = sql_query.redis_query_string(redis_client=client) print(redis_cmd) # Output: FT.SEARCH products "@category:{electronics}" r)SchemaRegistry) Translatorz\sql-redis is required for SQL query support. Install it with: pip install redisvl[sql-redis]N)Redis)sql_redis.schemar*sql_redis.translatorr+ ImportErrorredisr,from_urlload_allr%rr translateto_command_string) r r'r(r*r+r,registry translatorr translateds r redis_query_stringzSQLQuery.redis_query_string]sP  7 7 7 7 7 7 7 7 7 7 7 7 7   B     # # # # # # >>)44L">,// Z)) %%dh <<))#.. ++---s ))N)Nr&) __name__ __module__ __qualname____doc__rrrrr r%r8rr rrs2 # #C #$sCx.)A # # # #/c/4S>/c////f'+1B.B.smB.B. B.B.B.B.B.B.rr)r<rtypingrrrrr=rr r?snCC &&&&&&&&&&X.X.X.X.X.X.X.X.X.X.r