doczyai-pipelines/archive/scripts/adhoc/unit_test.py

# """
# Run N iterations of a single prompt, count unique outputs
# """

# import pandas as pd

# import prompts
# import claude_funcs
# import config
# import dict_operations


# N = 1

# # Paste contract text here
# text = """The maximum compensation for Covered Services rendered to a Covered Person shall be the "Allowed Amount." 
# The Allowed Amount for Covered Services is the lesser of: (i) Allowable Charges; or (ii) as applicable, 
# (a) 100% of the Payor's Medicaid fee schedule in effect on the date of service, 
# or (b) for those Covered Services included in the Service Categories in Table 1 below, 
# the applicable "Contracted Rate" set forth in Table 1 below. Physical, Occupational and Speech Therapy | 80% of the Medicaid Rate in effect on the date of service.
# """

# # Change prompt below
# prompt = prompts.BOTTOM_UP_METHODOLOGY_BREAKOUT({'FULL_METHODOLOGY' : text})

# all_dicts = []
# for i in range(N):
#     answer = claude_funcs.invoke_claude(prompt, config.MODEL_ID_CLAUDE35_SONNET, "", 512)
#     answer_dict = dict_operations.secondary_string_to_dict(answer, "")
#     print(answer_dict)
#     all_dicts.append(answer_dict)
    
# # answer_df = pd.DataFrame(all_dicts)
# # answer_df.to_csv('full_methodology_unit_test.csv')