Skip to content

Commit d26d8b2

Browse files
committed
fix(text2sql): import QueryResult for type hints
1 parent e42d838 commit d26d8b2

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

app/text2sql_engine.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
from app.logging_config import get_logger
3232
from app.resilience import CircuitBreaker, CircuitBreakerConfig, compute_backoff_seconds
3333
from db.connection import DatabaseManager
34+
from db.executor import QueryResult
3435
from db.schema import SchemaInfo, SchemaIntrospector
3536

3637
logger = get_logger(__name__)
@@ -999,7 +1000,7 @@ async def execute_sql(
9991000
sql: str,
10001001
database_id: str | None = None,
10011002
max_rows: int = 1000,
1002-
) -> "QueryResult":
1003+
) -> QueryResult:
10031004
"""
10041005
Execute a SQL query using the safe executor.
10051006
@@ -1011,7 +1012,7 @@ async def execute_sql(
10111012
Returns:
10121013
QueryResult with execution results
10131014
"""
1014-
from db.executor import QueryResult, SafeQueryExecutor
1015+
from db.executor import SafeQueryExecutor
10151016

10161017
async with self._db_manager.session() as session:
10171018
executor = SafeQueryExecutor(

0 commit comments

Comments
 (0)