Merge branch 'task_3255/phjirsa' of https://dev.azure.com/codewithai/UHG%20Optum%20RX%20-%20PBM%20Prior%20Auth/_git/UHG%20Optum%20RX%20-%20PBM%20Prior%20Auth

Author: juan-trujillo

.gitignore

@@ -1,5 +1,7 @@
 **/*.pyc
 .venv/
+.env
 .azure
 __pycache__/
 .coverage
+sample_data/

api.http

@@ -1,4 +1,25 @@
 GET http://localhost:8000/generate_name
 
 ###
-GET http://localhost:8000/generate_name?starts_with=p
+GET http://localhost:8000/generate_name?starts_with=p
+
+###
+POST http://localhost:8000/generate_plan
+Content-Type: application/json
+
+{
+    "drug": "ozempic",
+    "rules": [
+        {
+            "name": "1.1",
+            "value": "For patients requiring ongoing drug treatment for type 2 diabetes mellitus, submission of medical records (e.g., chart notes) confirming diagnosis of type 2 diabetes mellitus"
+        },
+        {
+            "name": "1.2",
+            "value": "A1C greater than or equal to 6.5% OR Fasting plasma glucose (FPG) greater than or equal to 126 mg/dL"
+        }
+    ]
+}
+
+###
+GET http://localhost:8000/get_plan?drug=ozempic

notebooks/rewoo.ipynb

@@ -0,0 +1,18 @@
+from langchain_core.tools import tool
+
+@tool
+def get_patient_notes(patient_id: str):
+    """Get patient notes by patient ID."""
+    return {
+        "id": patient_id,
+        "notes": "Patient was diagnosed with Type 2 diabetes on 2023-01-09. Patient has an A1C level of 9.5%."
+    }
+    
+@tool
+def get_patient_data(patient_id: str):
+    """Get patient data by patient ID."""
+    return {
+        "id": patient_id,
+        "name": "John Doe",
+        "birthdate": "2010-01-05"
+    }

notebooks/tools.py

@@ -0,0 +1,20 @@
+prompt = """For the following rule, make plans that can validate a patient meets the criteria defined by the rule. For each plan, indicate \
+which external tool together with tool input to retrieve evidence. You can store the evidence into a \
+variable #E that can be called by later tools. (Plan, #E1, Plan, #E2, Plan, ...)
+
+Tools can be one of the following:
+(1) ChartNotes[input]: Worker that searches for patient visit notes. Useful when you need to confirm a diagnosis or treatment.
+(2) PatientData[input]: Worker that searches for specific patient data. Useful when looking up patient data.
+(2) LLM[input]: A pretrained LLM like yourself. Useful when you need to act with general
+world knowledge and common sense. Prioritize it when you are confident in solving the problem
+yourself. Input can be any instruction.
+
+For example,
+Rule: The outside air temperature must be below 32 degrees Fahrenheit in order for it to be snowing.
+Plan: Look up the outside air temperature. #E1 = WeatherData[lookup outside air temperature]
+
+Begin! 
+Describe your plans with rich details. Each Plan should be followed by only one #E.
+
+Rule: {rule}"""
+

src/api/agents/planner.py

@@ -1,11 +1,20 @@
 import random
 
+from pydantic import BaseModel
+from langchain_core.tools import tool
+from langgraph.checkpoint.memory import MemorySaver
+from langgraph.graph import END, START, StateGraph, MessagesState
+from langgraph.prebuilt import ToolNode
+
 import fastapi
 
 from .data import names
 
 app = fastapi.FastAPI()
 
+class plan_request(BaseModel):
+    drug: str
+    rules: list
 
 @app.get("/generate_name")
 async def generate_name(starts_with: str = None):
@@ -14,3 +23,11 @@ async def generate_name(starts_with: str = None):
         name_choices = [name for name in name_choices if name.lower().startswith(starts_with.lower())]
     random_name = random.choice(name_choices)
     return {"name": random_name}
+
+@app.post("/generate_plan")
+async def generate_plan(request: plan_request):
+    return {"plan": f"Take 1 of {request.drug} pill every 4 hours", "rules": request.rules}
+
+@app.get("/get_plan")
+async def get_plan(drug: str):
+    return {"plan": f"Take 1 pill of {drug} every 4 hours"}

src/api/main.py

@@ -0,0 +1,8 @@
+# write a function to return a list of drug qualification rules
+def fetch_drug_rules(drug: str):
+    return [
+        {"rule": "Patient must be 18 years or older", "logical": "AND"},
+        {"rule": "Patient must have a confirmed diagnosis of Type 2 diabetes", "logical": "AND"},
+        {"rule": "Patient must have an A1C level of 7.0% or higher", "logical": "AND"}
+    ]
+        

src/api/tools/__init__.py

@@ -1,3 +1,5 @@
 fastapi==0.111.0
 uvicorn[standard]==0.29.0
 gunicorn==22.0.0
+langgraph
+