Katon Minhas
afb6d5185d
Feature/lesser table caching refactor hybrid * chore: Remove unused duplicate main.py from shared pipeline * fix: Correct crosswalk paths in aarete_derived.py * chore: Remove unused documentation files from fieldExtraction * docs: Add documentation files to documentation folder * docs: Update README with uv setup, expanded project structure, and branching conventions * docs: Add uv installation steps with Ubuntu/WSL emphasis * Enable prompt caching for all remaining LLM calls - Add _INSTRUCTION() functions for: EXHIBIT_HEADER, EXHIBIT_LINKAGE, EXHIBIT_TITLE_MATCH, DATE_FIX, DERIVED_TERM_DATE, CHECK_PROVIDER_NAME_MATCH, SPECIAL_CASE_ASSIGNMENT - Update all invoke_claude() calls in saas and clover pipelines to use cache=True with corresponding _INSTRUCTION() functions - Add new instructions to get_cacheable_instructions() for cache warming - Update tests for new instruction functions Functions now using caching: - prompt_exhibit_level - prompt_exhibit_lesser (EXHIBIT_LEVEL_LESSER_OF) - prompt_fee_schedule_breakout - prompt_grouper_breakout - prompt_special_case_assignment - prompt_exhibit_linkage - prompt_exhibit_header - prompt_smart_chunked (ONE_TO_ONE templates) - prompt_date_fix - prompt_derived_term_date - prompt_exhibit_title_match - provider_name_match_check 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * Reorder * feat: Add bcbs_promise client pipeline with OFFSET_TERM extraction - Add new bcbs_promise client with HSC-based OFFSET_TERM field extraction - Extract full paragraph text of offset/recoupment provisions from contracts - Derive OFFSET_INDICATOR (Y/N) from OFFSET_TERM presence - Fix reorder_columns to preserve extra columns not in COLUMN_ORDER - Update QC/QA output path to outputs/qc_qa/ * fix: Update dev deps and test assertions for QC/QA output path - Add pytest/pytest-mock to dev dependencies for mypy type checking - Update test assertions to expect outputs/qc_qa instead of qa_qc_output * style: Apply black formatting to prompt_templates.py * Merge main, move scripts * Archive some scripts * update py version * remove .py version file * Remove ASCII characters * Restore testbed code * restore tracking * Update testbed metrics * Enable prompt caching for CODE_LAST_CHECK, FILL_BILL_TYPE, DUAL_LOB_CHECK, and GROUPER_BREAKOUT - Add CODE_LAST_CHECK_INSTRUCTION() for service specificity classification - Add FILL_BILL_TYPE_INSTRUCTION() for bill type code determination - Add DUAL_LOB_CHECK_INSTRUCTION() for Medicare/Medicaid classification - Update code_funcs.py to use caching for CODE_LAST_CHECK, FILL_BILL_TYPE, GROUPER_BREAKOUT - Update postprocessing_funcs.py to use caching for DUAL_LOB_CHECK - Add new instructions to get_cacheable_instructions() for cache warming - Add unit tests for new instruction functions 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * Fix postprocessing_funcs to remove invalid columns * Merge branch 'main' into feature/lesser-table-caching-refactor-hybrid * Revert prompt caching changes from aed1b73c * update formatting * Update imports Approved-by: Sha Brown Approved-by: Praneel Panchigar
79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
import streamlit as st
|
|
import msal
|
|
import requests
|
|
import boto3
|
|
from botocore.exceptions import ClientError
|
|
import json
|
|
|
|
# Replace with your own values
|
|
CLIENT_ID = "effafe90-7ed7-43a3-ab03-19a0be2f1758"
|
|
# CLIENT_SECRET = 'bjQ8Q~lpR2uBcGI34VDu16t73doz8Crj0YY_~dgD'
|
|
# TENANT_ID = ''
|
|
|
|
AUTHORITY = "https://login.microsoftonline.com/organizations/"
|
|
SCOPE = ["User.Read"]
|
|
REDIRECT_URI = "https://172.29.20.102:8501"
|
|
|
|
# Initialize boto3 client to interact with AWS Secrets Manager
|
|
|
|
|
|
def get_secret():
|
|
|
|
secret_name = "doczy-sso-azure-app-key"
|
|
region_name = "us-east-2"
|
|
|
|
# Create a Secrets Manager client
|
|
# session = boto3.session.Session()
|
|
# client = session.client(
|
|
# service_name='secretsmanager',
|
|
# region_name=region_name
|
|
# )
|
|
client = boto3.client("secretsmanager", region_name=region_name)
|
|
try:
|
|
get_secret_value_response = client.get_secret_value(SecretId=secret_name)
|
|
except ClientError as e:
|
|
raise e
|
|
|
|
secret = get_secret_value_response["SecretString"]
|
|
secret = json.loads(secret)["CLIENT_SECRET"]
|
|
|
|
return secret
|
|
|
|
|
|
CLIENT_SECRET = get_secret()
|
|
|
|
app = msal.ConfidentialClientApplication(
|
|
CLIENT_ID, authority=AUTHORITY, client_credential=CLIENT_SECRET
|
|
)
|
|
|
|
|
|
def get_auth_url(REDIRECT_URI):
|
|
auth_url = app.get_authorization_request_url(SCOPE, redirect_uri=REDIRECT_URI)
|
|
return auth_url
|
|
|
|
|
|
@st.cache_data
|
|
def get_token_from_code(auth_code, REDIRECT_URI):
|
|
app = msal.ConfidentialClientApplication(
|
|
CLIENT_ID, authority=AUTHORITY, client_credential=CLIENT_SECRET
|
|
)
|
|
result = app.acquire_token_by_authorization_code(
|
|
auth_code, scopes=SCOPE, redirect_uri=REDIRECT_URI
|
|
)
|
|
return result["access_token"]
|
|
|
|
|
|
def get_user_info(access_token):
|
|
headers = {"Authorization": f"Bearer {access_token}"}
|
|
response = requests.get("https://graph.microsoft.com/v1.0/me", headers=headers)
|
|
return response.json()
|
|
|
|
|
|
def handle_redirect(REDIRECT_URI):
|
|
if not st.session_state.get("access_token"):
|
|
code = st.query_params.get("code")
|
|
if code:
|
|
access_token = get_token_from_code(code, REDIRECT_URI)
|
|
st.session_state["access_token"] = access_token
|
|
st.session_state
|