13.01.26
This commit is contained in:
16
app.py
16
app.py
@@ -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)):
|
||||||
|
|||||||
Reference in New Issue
Block a user