Files
onyx/backend/danswer/server/utils.py
T
2023-08-31 15:33:39 -07:00

24 lines
650 B
Python

import json
from typing import Any
def get_json_line(json_dict: dict) -> str:
return json.dumps(json_dict) + "\n"
def mask_string(sensitive_str: str) -> str:
return "****...**" + sensitive_str[-4:]
def mask_credential_dict(credential_dict: dict[str, Any]) -> dict[str, str]:
masked_creds = {}
for key, val in credential_dict.items():
if not isinstance(val, str):
raise ValueError(
f"Unable to mask credentials of type other than string, cannot process request."
f"Recieved type: {type(val)}"
)
masked_creds[key] = mask_string(val)
return masked_creds