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:
@@ -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
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user