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

zylon-ai · Apr 30, 2024 · 9d0d614 · 9d0d614
1 parent e21bf20
commit 9d0d614
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/private_gpt/components/llm/custom/sagemaker.py b/private_gpt/components/llm/custom/sagemaker.py
@@ -218,7 +218,7 @@ def complete(self, prompt: str, **kwargs: Any) -> CompletionResponse:
 
         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