zio/guardia-itsm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

manual-deploy 2026-06-08 00:13

Browse Source
This commit is contained in:
DESKTOP-TKLFCPR\ython 2026-06-08 00:13:45 +09:00
parent 738afda78a
commit 4b68523825
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
routers/upstage_ocr.py
View File

@ -200,7 +200,7 @@ async def get_ocr_config(
cfg = row.scalar_one_or_none() cfg = row.scalar_one_or_none()
if not cfg: if not cfg:
return {"configured": False} return {"configured": False}
key = cfg.api_key_enc or "" key = decrypt_secret(cfg.api_key_enc) or ""
masked_key = f"{key[:6]}{'*' * (len(key) - 10)}{key[-4:]}" if len(key) > 10 else "***" masked_key = f"{key[:6]}{'*' * (len(key) - 10)}{key[-4:]}" if len(key) > 10 else "***"
return { return {
"configured": True, "configured": True,
@ -232,7 +232,7 @@ async def parse_document(
async with httpx.AsyncClient(timeout=120) as client: async with httpx.AsyncClient(timeout=120) as client:
r = await client.post( r = await client.post(
f"{UPSTAGE_BASE}/document-digitization", f"{UPSTAGE_BASE}/document-digitization",
headers={"Authorization": f"Bearer {cfg.api_key_enc}"}, headers={"Authorization": f"Bearer {decrypt_secret(cfg.api_key_enc)}"},
files={"document": (file.filename, file_bytes, mime)}, files={"document": (file.filename, file_bytes, mime)},
data={ data={
"model": model or cfg.model, "model": model or cfg.model,
@ -286,7 +286,7 @@ async def extract_information(
async with httpx.AsyncClient(timeout=120) as client: async with httpx.AsyncClient(timeout=120) as client:
r = await client.post( r = await client.post(
f"{UPSTAGE_BASE}/information-extraction", f"{UPSTAGE_BASE}/information-extraction",
headers={"Authorization": f"Bearer {cfg.api_key_enc}"}, headers={"Authorization": f"Bearer {decrypt_secret(cfg.api_key_enc)}"},
files={"document": (file.filename, file_bytes, mime)}, files={"document": (file.filename, file_bytes, mime)},
data={"schema": json.dumps(schema_dict, ensure_ascii=False)} data={"schema": json.dumps(schema_dict, ensure_ascii=False)}
) )
@ -342,7 +342,7 @@ async def document_qa(
async with httpx.AsyncClient(timeout=120) as client: async with httpx.AsyncClient(timeout=120) as client:
r = await client.post( r = await client.post(
f"{UPSTAGE_BASE}/document-qa", f"{UPSTAGE_BASE}/document-qa",
headers={"Authorization": f"Bearer {cfg.api_key_enc}"}, headers={"Authorization": f"Bearer {decrypt_secret(cfg.api_key_enc)}"},
files={"document": (file.filename, file_bytes, mime)}, files={"document": (file.filename, file_bytes, mime)},
data={"question": question} data={"question": question}
) )
@ -386,14 +386,14 @@ async def batch_parse(
if mode == "extract" and schema: if mode == "extract" and schema:
r = await client.post( r = await client.post(
f"{UPSTAGE_BASE}/information-extraction", f"{UPSTAGE_BASE}/information-extraction",
headers={"Authorization": f"Bearer {cfg.api_key_enc}"}, headers={"Authorization": f"Bearer {decrypt_secret(cfg.api_key_enc)}"},
files={"document": (file.filename, file_bytes, mime)}, files={"document": (file.filename, file_bytes, mime)},
data={"schema": schema} data={"schema": schema}
) )
else: else:
r = await client.post( r = await client.post(
f"{UPSTAGE_BASE}/document-digitization", f"{UPSTAGE_BASE}/document-digitization",
headers={"Authorization": f"Bearer {cfg.api_key_enc}"}, headers={"Authorization": f"Bearer {decrypt_secret(cfg.api_key_enc)}"},
files={"document": (file.filename, file_bytes, mime)}, files={"document": (file.filename, file_bytes, mime)},
data={"model": cfg.model, "ocr": "auto", "output_formats": '["text"]'} data={"model": cfg.model, "ocr": "auto", "output_formats": '["text"]'}
) )