fix: Replacing unsafe `eval()` with `json.loads()` (#1890)

2024-04-30 15:58:19 +08:00 · 2024-04-30 15:58:19 +08:00 · 9d0d614706
parent e21bf20c10
commit 9d0d614706
1 changed files with 1 additions and 1 deletions
--- a/private_gpt/components/llm/custom/sagemaker.py
+++ b/private_gpt/components/llm/custom/sagemaker.py
@ -218,7 +218,7 @@ class SagemakerLLM(CustomLLM):

        response_body = resp["Body"]
        response_str = response_body.read().decode("utf-8")
-        response_dict = eval(response_str)
+        response_dict = json.loads(response_str)

        return CompletionResponse(
            text=response_dict[0]["generated_text"][len(prompt) :], raw=resp