jacob-mathison/doczyai-pipelines
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Activity

Merged in bugfix/code-funcs (pull request #872)

Browse Source 
Bugfix/code funcs

* Try-except code extraction

* Additional try-except for protection

* Black format


Approved-by: Praneel Panchigar
This commit is contained in:
Katon Minhas
2026-02-05 22:10:50 +00:00
parent 49407bfb93
commit c62bd1cd6a
3 changed files with 32 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/codes/code_funcs.py
+29 -14
View File
@@ -116,8 +116,11 @@ def code_explicit(service, methodology, filename):
instruction=prompt_templates.CODE_EXPLICIT_INSTRUCTION(),
usage_label="CODE_EXPLICIT",
)
code_answer_dict = _parser(llm_answer_raw)
return code_answer_dict
try:
code_answer_dict = _parser(llm_answer_raw)
return code_answer_dict
except:
return {}
def code_category(service, proc_category, hcpcs_level2_mapping, filename):
@@ -151,7 +154,10 @@ 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
try:
llm_answer_final = _parser(llm_answer_raw) # Returns list
except:
{}
code_answer_dict = {"PROCEDURE_CD": [], "PROCEDURE_CD_DESC": []}
for answer in llm_answer_final:
@@ -179,16 +185,19 @@ def code_implicit_special(service, filename):
Returns:
dict: A dictionary containing the implicit codes found for the service. If no codes are found, returns an empty dictionary.
"""
prompt, _parser = prompt_templates.CODE_IMPLICIT_SPECIAL(service)
llm_answer_raw = llm_utils.invoke_claude(
prompt,
"sonnet_latest",
filename,
cache=True,
instruction=prompt_templates.CODE_IMPLICIT_SPECIAL_INSTRUCTION(),
)
llm_answer_final = _parser(llm_answer_raw)
prompt, _parser = prompt_templates.CODE_IMPLICIT_SPECIAL(service)
try:
llm_answer_raw = llm_utils.invoke_claude(
prompt,
"sonnet_latest",
filename,
cache=True,
instruction=prompt_templates.CODE_IMPLICIT_SPECIAL_INSTRUCTION(),
)
llm_answer_final = _parser(llm_answer_raw)
except:
return {}
special_case_mapping = {
"Drugs": ["J0000-J9999"],
@@ -512,7 +521,10 @@ def fill_bill_type(
cache=True,
instruction=prompt_templates.FILL_BILL_TYPE_INSTRUCTION(),
)
llm_answer_final = _parser(llm_answer_raw)
try:
llm_answer_final = _parser(llm_answer_raw)
except:
return answer_dict
# Second attempt: Service term + reimbursement term + exhibit context (if first attempt failed and context available)
if string_utils.is_empty(llm_answer_final) and exhibit_text:
@@ -530,7 +542,10 @@ def fill_bill_type(
cache=True,
instruction=prompt_templates.FILL_BILL_TYPE_INSTRUCTION(),
)
llm_answer_final = _parser(llm_answer_raw)
try:
llm_answer_final = _parser(llm_answer_raw)
except:
return answer_dict
# Keep legacy behavior: return answer_dict unchanged if no result found
if not llm_answer_final: