critical_thinking.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

from question import get_response
from prompt_template import PromptTemplate
import json
import traceback

class CriticalThinking:
    def __init__(self, main_question: str, prompt_template_path: str, background_knowledge_path: str):
        with open(background_knowledge_path, 'r') as file:
            self.background_knowledge = file.read()
        prompt_template =  PromptTemplate(prompt_template_path)
        self.query = prompt_template.get_prompt(MainQuestion=main_question, BackgroundKnowledges = self.background_knowledge)

    def create(self):
        print(self.query)
        try:
            self.reply_raw = get_response(self.query)
        except Exception as e:
            traceback_str = traceback.format_exc()
            error_message = f"ERROR: {str(e)}"
            print(traceback_str + error_message)
            sys.exit(1)
        print(self.reply_raw)

    def save_to_raw(self, file_path):
        with open(file_path, 'w') as file:
            file.write(self.reply_raw)

    def save_to_json(self, file_path):
        with open(file_path, 'w') as file:
            json.dump(json.loads(self.reply_raw), file, indent=4, ensure_ascii=False)


if __name__ == "__main__":
    import sys
    from params import get_param
    prompt_template_path = get_param("prompt_templates_path")

    if len(sys.argv) != 3:
        print("Usage: <MainQuestion> <BackgroundKnowledge>")
        sys.exit(1)
    main_question = sys.argv[1]
    background_knowledge_path = sys.argv[2]
    think = CriticalThinking(main_question, prompt_template_path + "/ptemplate_critical_thinking.txt", background_knowledge_path)
    think.create()
    think.save_to_raw("test/result/critical_thinking.json")