|
|
|
|
|
|
|
|
|
|
|
import hashlib |
|
|
|
|
|
|
|
|
def get_org_and_model_names(file_path: str) -> str: |
|
|
org, model = file_path.split("/")[-3:-1] |
|
|
model = model.removesuffix(".json") |
|
|
model = model.split('_request_')[0] |
|
|
return org, model |
|
|
|
|
|
|
|
|
def get_model_name(file_path: str) -> str: |
|
|
org, model = get_org_and_model_names(file_path) |
|
|
return f"{org}/{model}" |
|
|
|
|
|
|
|
|
def get_hash(key: str) -> str: |
|
|
sha256_hash = hashlib.sha256() |
|
|
sha256_hash.update(key.encode('utf-8')) |
|
|
return sha256_hash.hexdigest()[:16] |
|
|
|
|
|
|
|
|
def get_request_id(model: str, revision: str, precision: str) -> str: |
|
|
return f"{model}_{revision}_{precision}" |
|
|
|
|
|
|
|
|
def get_request_hash(model: str, revision: str, precision: str) -> str: |
|
|
return get_hash(get_request_id(model, revision, precision)) |
|
|
|
