Update llm/llm.py

llm/llm.py

@@ -1,3 +1,5 @@
+import os
+
 import yaml
 import logging
 
@@ -14,7 +16,11 @@ class LLM_chain:
 
     def __read_yaml(self):
         try:
-            with open('./prompts.yaml', 'r') as file:
+            # get current dir 
+            current_dir = os.path.dirname(os.path.realpath(__file__))
+            yaml_file = os.path.join(current_dir, 'prompts.yaml')
+            
+            with open(yaml_file, 'r') as file:
                 data = yaml.safe_load(file)
             return data
         except Exception as e: