black format

This commit is contained in:
Venkat
2026-02-03 13:41:09 +00:00
parent e8e8f1ba5a
commit 2aa68f75b9
15 changed files with 253 additions and 231 deletions
+7 -9
View File
@@ -141,9 +141,9 @@ def code_category(service, proc_category, hcpcs_level2_mapping, filename):
for key, value in hcpcs_level2_mapping.items()
if key.startswith(category)
]
prompt, _parser = prompt_templates.CODE_CATEGORY(service, matching_values)
llm_answer_raw = llm_utils.invoke_claude(
prompt,
"sonnet_latest",
@@ -151,9 +151,7 @@ def code_category(service, proc_category, hcpcs_level2_mapping, filename):
cache=True,
instruction=prompt_templates.CODE_CATEGORY_INSTRUCTION(),
)
llm_answer_final = _parser(
llm_answer_raw
) # Returns list
llm_answer_final = _parser(llm_answer_raw) # Returns list
code_answer_dict = {"PROCEDURE_CD": [], "PROCEDURE_CD_DESC": []}
for answer in llm_answer_final:
@@ -355,8 +353,10 @@ def code_implicit_rag(service, implicit_run_dict, filename, constants):
)
)
prompt, _parser = prompt_templates.CODE_IMPLICIT(service, level_dict["match_list"])
prompt, _parser = prompt_templates.CODE_IMPLICIT(
service, level_dict["match_list"]
)
# Run Prompt
llm_answer_raw = llm_utils.invoke_claude(
prompt,
@@ -458,7 +458,6 @@ def code_last_check(service, filename):
return "Generic"
def fill_bill_type(
service,
answer_dict,
@@ -488,7 +487,6 @@ def fill_bill_type(
service, valid_bill_type, reimb_term=reimb_term, exhibit_text=None
)
# First attempt: Service term + reimbursement term
llm_answer_raw = llm_utils.invoke_claude(
prompt,