yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-10-29 00:39:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove unecessary literal_eval (#4570)

Browse Source
This commit is contained in:
Silvano Cerza 2023-04-06 13:30:45 +02:00 committed by GitHub
parent e0895f0ac2
commit ee7b25b8cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
haystack/nodes/prompt/prompt_template.py
View File

@ -2,6 +2,7 @@ from typing import Optional, List, Union, Tuple, Dict, Iterator, Any
import logging
import os
import ast
import json
from abc import ABC
from uuid import uuid4
@ -21,7 +22,7 @@ from haystack.schema import Document, MultiLabel
logger = logging.getLogger(__name__)
PROMPT_TEMPLATE_ALLOWED_FUNCTIONS = ast.literal_eval(
PROMPT_TEMPLATE_ALLOWED_FUNCTIONS = json.loads(
os.environ.get(HAYSTACK_PROMPT_TEMPLATE_ALLOWED_FUNCTIONS, '["join", "to_strings", "replace", "enumerate", "str"]')
)
PROMPT_TEMPLATE_SPECIAL_CHAR_ALIAS = {"new_line": "\n", "tab": "\t", "double_quote": '"', "carriage_return": "\r"}