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
+3 -2
View File
@@ -2,9 +2,10 @@ import re
from collections import defaultdict
from itertools import chain, count, groupby
from src.prompts import preprocessing_prompts
import src.utils.llm_utils as llm_utils
import src.utils.string_utils as string_utils
from src import config, keywords, prompts, smart_chunking_funcs
from src import config, keywords, smart_chunking_funcs
from src.regex.regex_patterns import PIPE_PATTERN
@@ -298,7 +299,7 @@ def filter_quick_review(text_dict):
def get_exhibit_pages(text_dict, filename):
exhibit_pages = []
for page_num, page in text_dict.items():
prompt = prompts.EXHIBIT_CHECK(page[0:100])
prompt = preprocessing_prompts.EXHIBIT_CHECK(page[0:100])
claude_answer_raw = llm_utils.invoke_claude(
prompt, config.MODEL_ID_CLAUDE3_HAIKU, filename, max_tokens=10
)