From f34bf5139900d53469586664e0fbb8f85e91b334 Mon Sep 17 00:00:00 2001
From: Tom Viner <tom@viner.tv>
Date: Mon, 17 Apr 2023 23:19:48 +0100
Subject: [PATCH] Access prompt.txt via relative path

---
 wolverine.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/wolverine.py b/wolverine.py
index 8f729fe..736c3ef 100644
--- a/wolverine.py
+++ b/wolverine.py
@@ -10,6 +10,8 @@ from termcolor import cprint
 from dotenv import load_dotenv
 
 
+FOLDER = os.path.realpath(os.path.dirname(__file__))
+
 # Set up the OpenAI API
 load_dotenv()
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -17,7 +19,7 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 DEFAULT_MODEL = os.environ.get("DEFAULT_MODEL", "gpt-4")
 
 
-with open("prompt.txt") as f:
+with open(os.path.join(FOLDER, "prompt.txt")) as f:
     SYSTEM_PROMPT = f.read()