Spaces:
Sleeping
Sleeping
priyansh-saxena1 commited on
Commit Β·
c6b9370
1
Parent(s): 4e16e37
fix: docker issue
Browse files- Dockerfile +2 -0
- app/graph.py +0 -11
- app/llm.py +0 -3
Dockerfile
CHANGED
|
@@ -8,8 +8,10 @@ RUN apt-get update && apt-get install -y \
|
|
| 8 |
curl \
|
| 9 |
ca-certificates \
|
| 10 |
bash \
|
|
|
|
| 11 |
&& rm -rf /var/lib/apt/lists/*
|
| 12 |
|
|
|
|
| 13 |
# βββ Install Ollama βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 14 |
RUN curl -fsSL https://ollama.com/install.sh | bash
|
| 15 |
|
|
|
|
| 8 |
curl \
|
| 9 |
ca-certificates \
|
| 10 |
bash \
|
| 11 |
+
zstd \
|
| 12 |
&& rm -rf /var/lib/apt/lists/*
|
| 13 |
|
| 14 |
+
|
| 15 |
# βββ Install Ollama βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 16 |
RUN curl -fsSL https://ollama.com/install.sh | bash
|
| 17 |
|
app/graph.py
CHANGED
|
@@ -121,17 +121,6 @@ def agent_node(state: IntakeState) -> dict:
|
|
| 121 |
|
| 122 |
print(f"[{time.time():.3f}] [Graph Node] LLM returned. Preparing node dictionaries...")
|
| 123 |
|
| 124 |
-
if result.emergency:
|
| 125 |
-
return {
|
| 126 |
-
"messages": [{"role": "assistant", "content": (
|
| 127 |
-
"π¨ EMERGENCY: Your symptoms require immediate attention. "
|
| 128 |
-
"Please call 911 or go to your nearest emergency room right away."
|
| 129 |
-
)}],
|
| 130 |
-
"clinical_state": result.model_dump_json(),
|
| 131 |
-
"current_node": "done",
|
| 132 |
-
"frontend_stage": "done",
|
| 133 |
-
}
|
| 134 |
-
|
| 135 |
stage = compute_stage(result)
|
| 136 |
missing = missing_from(result)
|
| 137 |
reply = result.reply or "Could you tell me more?"
|
|
|
|
| 121 |
|
| 122 |
print(f"[{time.time():.3f}] [Graph Node] LLM returned. Preparing node dictionaries...")
|
| 123 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 124 |
stage = compute_stage(result)
|
| 125 |
missing = missing_from(result)
|
| 126 |
reply = result.reply or "Could you tell me more?"
|
app/llm.py
CHANGED
|
@@ -14,7 +14,6 @@ CRITICAL RULES:
|
|
| 14 |
- Do NOT diagnose or give medical advice.
|
| 15 |
- Do NOT ask more than one question.
|
| 16 |
- If all fields are complete, set reply to "Thank you β I have everything I need."
|
| 17 |
-
- EMERGENCY RULE: Set emergency=true ONLY IF the patient explicitly describes an acute life-threatening event right now (e.g., EXACT words "crushing chest pain", "I can't breathe right now", "I want to kill myself"). Do NOT set emergency=true for generic "chest pain", "headache", or past symptoms. Default is always false.
|
| 18 |
|
| 19 |
OUTPUT FORMAT (strictly follow this, no extra text):
|
| 20 |
{
|
|
@@ -27,7 +26,6 @@ OUTPUT FORMAT (strictly follow this, no extra text):
|
|
| 27 |
"aggravating": "...",
|
| 28 |
"relieving": "...",
|
| 29 |
"ros": {"system_name": ["finding1", "finding2"]},
|
| 30 |
-
"emergency": false,
|
| 31 |
"reply": "The single question to ask the patient next"
|
| 32 |
}
|
| 33 |
|
|
@@ -44,7 +42,6 @@ class CombinedOutput(BaseModel):
|
|
| 44 |
aggravating: str | None = None
|
| 45 |
relieving: str | None = None
|
| 46 |
ros: dict[str, list[str]] = {}
|
| 47 |
-
emergency: bool = False
|
| 48 |
reply: str = ""
|
| 49 |
|
| 50 |
|
|
|
|
| 14 |
- Do NOT diagnose or give medical advice.
|
| 15 |
- Do NOT ask more than one question.
|
| 16 |
- If all fields are complete, set reply to "Thank you β I have everything I need."
|
|
|
|
| 17 |
|
| 18 |
OUTPUT FORMAT (strictly follow this, no extra text):
|
| 19 |
{
|
|
|
|
| 26 |
"aggravating": "...",
|
| 27 |
"relieving": "...",
|
| 28 |
"ros": {"system_name": ["finding1", "finding2"]},
|
|
|
|
| 29 |
"reply": "The single question to ask the patient next"
|
| 30 |
}
|
| 31 |
|
|
|
|
| 42 |
aggravating: str | None = None
|
| 43 |
relieving: str | None = None
|
| 44 |
ros: dict[str, list[str]] = {}
|
|
|
|
| 45 |
reply: str = ""
|
| 46 |
|
| 47 |
|