From cb7c6ea59762a3822c3ba7deeee6bbc8d1519a37 Mon Sep 17 00:00:00 2001 From: ArslanSaleem Date: Tue, 30 Jan 2024 09:11:47 +0100 Subject: [PATCH] fix(prompt_path): path issue if pandasai used outside of pandas env --- pandasai/prompts/base.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/pandasai/prompts/base.py b/pandasai/prompts/base.py index 787caafa0..6952579c4 100644 --- a/pandasai/prompts/base.py +++ b/pandasai/prompts/base.py @@ -1,6 +1,8 @@ import re from jinja2 import Environment, FileSystemLoader from typing import Optional +import os +from pathlib import Path class BasePrompt: @@ -20,7 +22,10 @@ def __init__(self, **kwargs): env = Environment() self.prompt = env.from_string(self.template) elif self.template_path: - env = Environment(loader=FileSystemLoader("pandasai/prompts/templates")) + # find path to template file + current_dir_path = Path(__file__).parent + path_to_template = os.path.join(current_dir_path, "templates") + env = Environment(loader=FileSystemLoader(path_to_template)) self.prompt = env.get_template(self.template_path) def render(self):