bobseares-2/csv-answers-studio-v1
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fail CSVAnswers persistence closed

Browse Source
This commit is contained in:
bobseares@gmail.com
2026-07-18 06:07:23 -03:00
parent c044d9deab
commit 2ac520acbe
3 changed files with 83 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

79
agent.py
View File

@@ -12,7 +12,6 @@ import io
import json
import os
import re
from datetime import datetime, timezone
from decimal import Decimal, InvalidOperation
from typing import Annotated, Any
@@ -40,6 +39,13 @@ MAX_CELL_CHARS = 2_000
CSV_MEDIA_TYPES = {"text/csv", "application/csv", "text/plain", "application/vnd.ms-excel"}
SAFE_ID_RE = re.compile(r"^[A-Za-z0-9][A-Za-z0-9_.:-]{0,127}$")
AnalysisId = Annotated[
str,
Field(min_length=1, max_length=128, pattern=r"^[A-Za-z0-9][A-Za-z0-9_.:-]{0,127}$"),
]
CsvText = Annotated[str, Field(min_length=1, max_length=MAX_CSV_BYTES)]
Question = Annotated[str, Field(min_length=1, max_length=500)]
# Local-only fallback lets sandbox/unit tests prove behavior without receiving
# managed DATABASE_URL. Hosted deployments use managed Postgres exclusively.
_LOCAL_ANALYSES: dict[tuple[str, str], dict[str, Any]] = {}
@@ -88,7 +94,7 @@ class CsvAnswersStudioV1(A2AAgent[CsvAnswersStudioV1Config, PlatformUserAuth]):
"highest numeric value, persist the grounded answer in user-scoped "
"managed Postgres, and reopen it from a polished one-page app."
)
version = "0.1.0"
version = "0.1.1"
config_model = CsvAnswersStudioV1Config
auth_model = PlatformUserAuth
@@ -115,45 +121,52 @@ class CsvAnswersStudioV1(A2AAgent[CsvAnswersStudioV1Config, PlatformUserAuth]):
@a2a.tool(
description="Analyze bounded pasted CSV text, answer the highest numeric row question, save the result, and persist a receipt.",
timeout_seconds=30,
idempotent=True,
idempotent=False,
cost_class="cheap",
)
async def analyze_csv(
self,
ctx: RunContext[PlatformUserAuth],
analysis_id: str,
csv_text: str,
question: str,
analysis_id: AnalysisId,
csv_text: CsvText,
question: Question,
) -> CsvAnalysisResult:
tenant = _tenant_key(ctx)
result = _analyze_csv_text(analysis_id=analysis_id, csv_text=csv_text, question=question)
if not result.ok:
return result
saved, receipt_id = _persist_analysis_and_receipt(
tenant_key=tenant,
analysis_id=analysis_id,
question=question,
csv_text=csv_text,
result=result.model_dump(mode="json"),
source="pasted_csv",
)
try:
saved, receipt_id = _persist_analysis_and_receipt(
tenant_key=tenant,
analysis_id=analysis_id,
question=question,
csv_text=csv_text,
result=result.model_dump(mode="json"),
source="pasted_csv",
)
except Exception: # noqa: BLE001
return _error(
analysis_id,
"database_error",
"CSVAnswers could not persist this analysis. Try again shortly.",
)
result.saved = saved
result.receipt_id = receipt_id
await ctx.emit_progress(f"saved analysis {analysis_id} for {tenant}")
await ctx.emit_progress(f"saved analysis {analysis_id}")
return result
@a2a.tool(
description="Analyze a bounded browser CSV upload encoded as base64 JSON, then save the result and receipt.",
timeout_seconds=30,
idempotent=True,
idempotent=False,
cost_class="cheap",
)
async def analyze_csv_upload_base64(
self,
ctx: RunContext[PlatformUserAuth],
analysis_id: str,
analysis_id: AnalysisId,
upload: BrowserCsvUpload,
question: str,
question: Question,
) -> CsvAnalysisResult:
decoded = _decode_browser_upload(upload)
if isinstance(decoded, CsvAnalysisResult):
@@ -163,13 +176,13 @@ class CsvAnswersStudioV1(A2AAgent[CsvAnswersStudioV1Config, PlatformUserAuth]):
@a2a.tool(
description="Analyze a typed external FileUpload CSV, then save the result and receipt.",
timeout_seconds=30,
idempotent=True,
idempotent=False,
cost_class="cheap",
)
async def analyze_csv_file(
self,
ctx: RunContext[PlatformUserAuth],
analysis_id: str,
analysis_id: AnalysisId,
file: Annotated[
UploadedFile,
FileUpload(
@@ -178,7 +191,7 @@ class CsvAnswersStudioV1(A2AAgent[CsvAnswersStudioV1Config, PlatformUserAuth]):
description="Bounded CSV file for CSVAnswers analysis.",
),
],
question: str,
question: Question,
) -> CsvAnalysisResult:
if file.size_bytes and file.size_bytes > MAX_CSV_BYTES:
return _error(analysis_id, "csv_too_large", f"CSV must be at most {MAX_CSV_BYTES} bytes.")
@@ -203,10 +216,17 @@ class CsvAnswersStudioV1(A2AAgent[CsvAnswersStudioV1Config, PlatformUserAuth]):
async def get_analysis(
self,
ctx: RunContext[PlatformUserAuth],
analysis_id: str,
analysis_id: AnalysisId,
) -> CsvAnalysisResult:
tenant = _tenant_key(ctx)
record = _load_analysis(tenant, analysis_id)
try:
record = _load_analysis(tenant, analysis_id)
except Exception: # noqa: BLE001
return _error(
analysis_id,
"database_error",
"CSVAnswers could not reload this analysis. Try again shortly.",
)
if record is None:
return _error(analysis_id, "not_found", "No saved analysis with that id exists for this signed-in user.")
return CsvAnalysisResult.model_validate(record)
@@ -224,7 +244,14 @@ class CsvAnswersStudioV1(A2AAgent[CsvAnswersStudioV1Config, PlatformUserAuth]):
) -> CsvAnalysisListResult:
tenant = _tenant_key(ctx)
safe_limit = min(max(int(limit), 1), 50)
return CsvAnalysisListResult(ok=True, analyses=_list_analyses(tenant, safe_limit))
try:
return CsvAnalysisListResult(ok=True, analyses=_list_analyses(tenant, safe_limit))
except Exception: # noqa: BLE001
return CsvAnalysisListResult(
ok=False,
code="database_error",
message="CSVAnswers could not list saved analyses. Try again shortly.",
)
def _tenant_key(ctx: RunContext[PlatformUserAuth]) -> str:
@@ -439,11 +466,12 @@ def _format_decimal(value: Decimal) -> str:
def _connect():
url = os.environ.get("DATABASE_URL")
if not url:
return None
raise RuntimeError("managed database is not configured")
import psycopg
return psycopg.connect(
url,
connect_timeout=3,
options="-c statement_timeout=10000 -c lock_timeout=5000 -c idle_in_transaction_session_timeout=30000",
)
@@ -467,7 +495,6 @@ def _persist_analysis_and_receipt(
"input_sha256": hashlib.sha256(csv_text.encode("utf-8")).hexdigest(),
"question_sha256": hashlib.sha256(question.encode("utf-8")).hexdigest(),
"source": source,
"created_at": datetime.now(timezone.utc).isoformat(),
"ok": True,
}
conn = _connect()