| from langchain.prompts import PromptTemplate | |
| def create_prompt_template(role_template, format_instructions=None): | |
| prompt = role_template.get("prompt_template", "") | |
| format_response = { | |
| "format_instructions": format_instructions | |
| } if role_template.get("name") == "summary" else {} | |
| return PromptTemplate( | |
| template=prompt, | |
| input_variables=[role_template.get("input_variables", "")], | |
| partial_variables=format_response | |
| ) | |
| def check_format(response): | |
| if not 'json' in response: | |
| response = f""" | |
| ```json\n | |
| {response} | |
| ``` | |
| """ | |
| return response |