| #!/usr/bin/env python | |
| # -*- coding: utf-8 -*- | |
| import hashlib | |
| def get_org_and_model_names_from_filepath(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_from_filepath(file_path: str) -> str: | |
| org, model = get_org_and_model_names_from_filepath(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)) | |