SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 20:12:55 +01:00
Code Issues Projects Releases Packages Activity

fix: Replacing unsafe eval() with json.loads() (#1890)
Some checks failed
release-please / release-please (push) Has been cancelled
tests / setup (push) Has been cancelled
tests / ${{ matrix.quality-command }} (black) (push) Has been cancelled
tests / ${{ matrix.quality-command }} (mypy) (push) Has been cancelled
tests / ${{ matrix.quality-command }} (ruff) (push) Has been cancelled
tests / test (push) Has been cancelled
tests / all_checks_passed (push) Has been cancelled

Browse source
This commit is contained in:
Patrick Peng 2024-04-30 15:58:19 +08:00 • committed by GitHub
parent e21bf20c10
commit 9d0d614706
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
private_gpt/components/llm/custom/sagemaker.py
View file

@ -218,7 +218,7 @@ class SagemakerLLM(CustomLLM):
response_body = resp["Body"] response_body = resp["Body"]
response_str = response_body.read().decode("utf-8") response_str = response_body.read().decode("utf-8")
response_dict = eval(response_str) response_dict = json.loads(response_str)
return CompletionResponse( return CompletionResponse(
text=response_dict[0]["generated_text"][len(prompt) :], raw=resp text=response_dict[0]["generated_text"][len(prompt) :], raw=resp