' jzdZddlmZddlZddlmZmZddlmZddl Z ddl m Z eGdde Z dS) uNL2SQLDeps — per-request context dataclass for NL-to-SQL agents. Extends ``RequestContext`` with a side-channel that tools write into so the ``/query`` REST endpoint can return structured rows without re-parsing the agent's prose output. ) annotationsN) dataclassfield)Any)RequestContextcFeZdZUdZedZded<edZded<edZded<ee Z d ed <ed Z d ed<ed Z d ed<ee Z ded<edZded<edZded<dS) NL2SQLDepsaGeneric per-request deps for any NL-to-SQL agent. The ``run_query`` tool writes results here during execution. The ``/query`` REST handler reads them after ``agent.run()`` completes. Attributes: query_result: Raw rows returned by the most recent ``run_query`` call. Each element is a ``{column: value}`` dict. generated_sql: The exact SQL that was executed (after LIMIT injection). truncated: ``True`` when the result was capped by the row limit. widget_mode: ``True`` when the caller is the REST ``/query`` endpoint and expects structured widget JSON. ``False`` (default) means the agent is running inside a pylogue chat session and should respond in plain prose. formatted_widgets: Populated by the ``format_response`` tool only when ``widget_mode=True``. N)defaultzlist[dict[str, Any]] | None query_resultz str | None generated_sqlsql_explanation)default_factoryzlist[dict[str, str]] query_logFbool truncated widget_modezlist[dict[str, Any]]formatted_widgetszasyncio.Queue | Noneprogress_queuezasyncpg.Pool | Nonepool)__name__ __module__ __qualname____doc__rr __annotations__r r listrrrrrrF/var/www/html/ai-enterprise-brain/backend/chanakya/nl2sql_core/deps.pyr r s-"16d0C0C0CLCCCC %d 3 3 3M3333"'%"5"5"5O5555&+eD&A&A&AIAAAAeE***I****e,,,K,,,,.3eD.I.I.IIIII+05+>+>+>N>>>> %d 3 3 3D333333rr ) r __future__rasyncio dataclassesrrtypingrasyncpgbackend.chanakya.a2a_contextrr rrrr%s #"""""((((((((777777 4444444 444r