feat: add daemon sql batch analysis

This commit is contained in:
Andrey Avtomonov 2026-05-11 16:56:50 +02:00
parent c45d131a1f
commit ffbbaf417a
4 changed files with 260 additions and 0 deletions

View file

@ -48,6 +48,11 @@ from ktx_daemon.source_generation import (
GenerateSourcesResponse,
generate_sources_response,
)
from ktx_daemon.sql_analysis import (
AnalyzeSqlBatchRequest,
AnalyzeSqlBatchResponse,
analyze_sql_batch_response,
)
from ktx_daemon.table_identifier import (
ParseTableIdentifierBatchRequest,
ParseTableIdentifierBatchResponse,
@ -193,6 +198,19 @@ def create_app(
detail=f"Table identifier parsing failed: {error}",
) from error
@app.post("/sql/analyze-batch", response_model=AnalyzeSqlBatchResponse)
async def sql_analyze_batch(
request: AnalyzeSqlBatchRequest,
) -> AnalyzeSqlBatchResponse:
try:
return analyze_sql_batch_response(request)
except Exception as error:
logger.exception("SQL batch analysis failed: %s", error)
raise HTTPException(
status_code=500,
detail=f"SQL batch analysis failed: {error}",
) from error
@app.post(
"/semantic-layer/generate-sources", response_model=GenerateSourcesResponse
)