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
|
||||
import re
|
||||
from datetime import datetime
|
||||
from io import BytesIO
|
||||
import requests
|
||||
|
||||
LOG_PATH = "/var/log/automation-service.log"
|
||||
|
||||
@@ -832,10 +834,7 @@ def detect_template(text):
|
||||
async def metadata(text: str = Body(..., embed=True)):
|
||||
return extract_report_metadata(text)
|
||||
|
||||
import httpx
|
||||
from io import BytesIO
|
||||
|
||||
async def call_extractor(text: str, lab: str = "AHK"):
|
||||
def call_extractor(text: str, lab: str = "AHK"):
|
||||
url = "http://62.72.36.116:8090/extract"
|
||||
params = {"lab": lab}
|
||||
|
||||
@@ -845,10 +844,11 @@ async def call_extractor(text: str, lab: str = "AHK"):
|
||||
"file": ("document.txt", fake_file, "text/plain")
|
||||
}
|
||||
|
||||
async with httpx.AsyncClient(timeout=60) as client:
|
||||
r = await client.post(url, params=params, files=files)
|
||||
r.raise_for_status()
|
||||
return r.json()
|
||||
response = requests.post(url, params=params, files=files, timeout=60)
|
||||
response.raise_for_status()
|
||||
|
||||
return response.json()
|
||||
|
||||
|
||||
@app.post("/parse")
|
||||
async def parse_endpoint(text: str = Body(..., embed=True)):
|
||||
|
||||
Reference in New Issue
Block a user