This commit is contained in:
2026-01-13 15:56:13 +01:00
parent 6bc7ece057
commit e385808547

16
app.py
View File

@@ -12,6 +12,8 @@ import io
from logging.handlers import RotatingFileHandler from logging.handlers import RotatingFileHandler
import re import re
from datetime import datetime from datetime import datetime
from io import BytesIO
import requests
LOG_PATH = "/var/log/automation-service.log" LOG_PATH = "/var/log/automation-service.log"
@@ -832,10 +834,7 @@ def detect_template(text):
async def metadata(text: str = Body(..., embed=True)): async def metadata(text: str = Body(..., embed=True)):
return extract_report_metadata(text) return extract_report_metadata(text)
import httpx def call_extractor(text: str, lab: str = "AHK"):
from io import BytesIO
async def call_extractor(text: str, lab: str = "AHK"):
url = "http://62.72.36.116:8090/extract" url = "http://62.72.36.116:8090/extract"
params = {"lab": lab} params = {"lab": lab}
@@ -845,10 +844,11 @@ async def call_extractor(text: str, lab: str = "AHK"):
"file": ("document.txt", fake_file, "text/plain") "file": ("document.txt", fake_file, "text/plain")
} }
async with httpx.AsyncClient(timeout=60) as client: response = requests.post(url, params=params, files=files, timeout=60)
r = await client.post(url, params=params, files=files) response.raise_for_status()
r.raise_for_status()
return r.json() return response.json()
@app.post("/parse") @app.post("/parse")
async def parse_endpoint(text: str = Body(..., embed=True)): async def parse_endpoint(text: str = Body(..., embed=True)):