File size: 898 Bytes
f4abbca 8786bb1 f4abbca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
from pydantic import BaseModel
from typing import Optional
from typing import List
import time
class NER_Request (BaseModel):
text: str
entities: List[str]
class NER_Response (BaseModel):
success: int
result: str
description: Optional[str] = ""
errorCode: Optional[int] = 0
errorDescriptin: Optional[str] = ""
entity_labels_sample = [
"team",
"developer",
"technology",
"tool",
"amount",
"duration",
"capacity",
"company",
"currency"
]
def entities_list_to_dict(entitiesList: List[str]):
return {key: '' for key in entitiesList}
def set_start () -> time:
return time.time()
def audit_elapsedtime(function: str, start: time):
end = time.time()
elapsedtime = end-start
print("------------------")
print(f"[{function}] Elapsed time: {elapsedtime}")
print("------------------")
return elapsedtime |