Merged in refactor/prompt_refactor (pull request #353)

READY: Refactor/prompt refactor

* Prompt refactor

* refactor imports

* Prompt refactor

* Import refactor

* Remove excess prompts from investment_prompts.py


Approved-by: Alex Galarce
This commit is contained in:
Katon Minhas
2025-01-13 21:11:36 +00:00
parent a9e48333de
commit dfbe47ef28
13 changed files with 651 additions and 120 deletions
+4 -3
View File
@@ -1,7 +1,8 @@
import pytest
from src.prompts import preprocessing_prompts
import src.utils as utils
from src.utils.table_utils import align_and_format_tables
from src import (prompts, config)
from src import (config)
import copy
class TestAlignAndFormatTables:
@@ -30,7 +31,7 @@ class TestAlignAndFormatTables:
}
filename = "test_file.txt"
expected_prompt = prompts.ALIGN_AND_FORMAT_TABLES(text_dict["1"])
expected_prompt = preprocessing_prompts.ALIGN_AND_FORMAT_TABLES(text_dict["1"])
input_text_dict = copy.deepcopy(text_dict)
result = align_and_format_tables(text_dict, filename)
@@ -91,7 +92,7 @@ class TestAlignAndFormatTables:
}
filename = "test_file.txt"
expected_prompt = prompts.ALIGN_AND_FORMAT_TABLES(text_dict["1"])
expected_prompt = preprocessing_prompts.ALIGN_AND_FORMAT_TABLES(text_dict["1"])
input_text_dict = copy.deepcopy(text_dict)
result = align_and_format_tables(text_dict, filename)