update tools functions

command_hist.txt

@@ -1,3 +1 @@
-I am Frank, DOB 1986-01-01, male, number 12345, emergency contact is Zoe, 67890.
-
 curl -X POST http://141.211.127.171/api/generate -H "Content-Type: application/json" -d '{"model":"llama3.2:latest","prompt": "Hello"}'

modules/instrctions.py

@@ -60,10 +60,9 @@ MEDICAL_INTAKE_SYSINT = (
 
    Data Management Commands:
    - Use get_data to review current information
-   - Use add_to_data to append new information
    - Use clear_data to reset the current session
    - Use confirm_data to verify information with the patient
-   - Use insert_data to finalize the record
+   - Use save_data to save the record to the database
 
    Guidelines:
    1. Always introduce yourself and explain the intake process

modules/nodes.py

@@ -81,7 +81,7 @@ def chatbot_with_tools(state: DataState) -> DataState:
                             "support_system": "",
                             "living_condition": "",
                         },
-                        "reveiw_system": {
+                        "review_system": {
                             "weight_change": "",
                             "fever": False,
                             "chill": False,

modules/tools.py

@@ -1,3 +1,5 @@
+import pprint
+
 from langchain_core.tools import tool
 from modules.data_class import DataState
 from langgraph.prebuilt import InjectedState
@@ -20,7 +22,7 @@ def patient_id(name: str, DOB: str, gender: str, contact: str, emergency_contact
     """
 
 @tool
-def symptom(main_symptom: str, length: str) -> str:
+def symptom(main_symptom: str, symptom_length: str) -> str:
     """Collecting patient's main symptom assessment including:
        - Primary symptoms
        - Duration of the symptoms
@@ -29,6 +31,68 @@ def symptom(main_symptom: str, length: str) -> str:
       The updated data with the patient's symptom information added.
     """
 
+@tool
+def pain(pain_location: str, pain_side: str, pain_intensity: int, pain_description: str, start_time: str, radiation: bool, triggers: str, symptom: str) -> str:
+    """Collecting patient's pain status including:
+        - Pain location using body mapping (head, arms, hands, trunk, legs, feet)
+        - Pain side (left or right)
+        - Pain intensity (0-10 scale for each location)
+        - Pain characteristics and patterns
+        - Onset time
+        - Radiation patterns
+        - Triggering factors
+        - Associated symptoms
+
+    Returns:
+      The updated data with the patient's pain status added.
+    """
+
+@tool
+def medical_hist(medical_condition: str, first_time: str, surgery_history: list, medication: str, allergy: str) -> str:
+    """Collecting patient's medical history including:
+        - Existing medical conditions
+        - First occurrence date
+        - Surgical history with dates
+        - Current medications
+        - Allergies
+
+    Returns:
+      The updated data with the patient's medical history added.
+    """
+
+@tool
+def family_hist(family_history: str) -> str:
+    """Collecting patient's family history
+
+    Returns:
+      The updated data with the patient's family history added.
+    """
+
+@tool
+def social_hist(occupation: str, smoking: bool, alcohol: bool, drug: bool, support_system: str, living_condition: str) -> str:
+    """Collecting patient's social history including:
+        - Occupation
+        - smoking or not
+        - alcohol use or not
+        - drug use or not
+        - living conditions 
+        - support system
+
+    Returns:
+      The updated data with the patient's social history added.
+    """
+
+@tool
+def review_system(weight_change: str, fever: bool, chill: bool, night_sweats: bool, sleep: str, gastrointestinal: str, urinary: str) -> str:
+    """Collecting patient's review information including:
+        - Recent weight changes
+        - Constitutional symptoms (fever, chills, night sweats)
+        - Sleep patterns
+        - Gastrointestinal and urinary function
+        
+    Returns:
+      The updated data with the patient's review.
+    """
 
 @tool
 def confirm_data() -> str:
@@ -68,30 +132,38 @@ def data_node(state: DataState) -> DataState:
     for tool_call in tool_msg.tool_calls:
 
         if tool_call["name"] == "patient_id":
-
             # Each order item is just a string. This is where it assembled as "drink (modifiers, ...)".
-            patient_name = tool_call["args"]["name"]
-            patient_DOB = tool_call["args"]["DOB"]
-            patient_gender = tool_call["args"]["gender"]
-            patient_contact = tool_call["args"]["contact"]
-            patient_emergency_contact = tool_call["args"]["emergency_contact"]
-
-            data["ID"]["name"]=patient_name
-            data["ID"]["DOB"]=patient_DOB
-            data["ID"]["gender"]=patient_gender
-            data["ID"]["contact"]=patient_contact
-            data["ID"]["emergency_contact"]=patient_emergency_contact
+            data["ID"]["name"]=tool_call["args"]["name"]
+            data["ID"]["DOB"]=tool_call["args"]["DOB"]
+            data["ID"]["gender"]=tool_call["args"]["gender"]
+            data["ID"]["contact"]=tool_call["args"]["contact"]
+            data["ID"]["emergency_contact"]=tool_call["args"]["emergency_contact"]
             
             response = "\n".join(data)
 
         elif tool_call["name"] == "symptom":
-
             # Each order item is just a string. This is where it assembled as "drink (modifiers, ...)".
-            main_symptom = tool_call["args"]["main_symptom"]
-            symptom_length = tool_call["args"]["length"]
+            data["symptom"]["main_symptom"]=tool_call["args"]["main_symptom"]
+            data["symptom"]["symptom_length"]=tool_call["args"]["length"]
+            response = "\n".join(data)
+
+        elif tool_call["name"] == "pain":
+            data["pain"]["pain_location"] = tool_call["args"]["pain_location"]
+            data["pain"]["pain_side"] = tool_call["args"]["pain_side"]
+            data["pain"]["pain_intensity"] = tool_call["args"]["pain_intensity"]
+            data["pain"]["pain_description"] = tool_call["args"]["pain_description"]
+            data["pain"]["start_time"] = tool_call["args"]["start_time"]
+            data["pain"]["radiation"] = tool_call["args"]["radiation"]
+            data["pain"]["triggers"] = tool_call["args"]["triggers"]
+            data["pain"]["symptom"] = tool_call["args"]["symptom"]
+            response = "\n".join(data)
 
-            data["symptom"]["main_symptom"]=main_symptom
-            data["symptom"]["symptom_length"]=symptom_length
+        elif tool_call["name"] == "medical_hist":
+            data["medical_hist"]["medical_condition"] = tool_call["args"]["medical_condition"]
+            data["medical_hist"]["first_time"] = tool_call["args"]["first_time"]
+            data["medical_hist"]["surgery_history"] = tool_call["args"]["surgery_history"]
+            data["medical_hist"]["medication"] = tool_call["args"]["medication"]
+            data["medical_hist"]["allergy"] = tool_call["args"]["allergy"]
             response = "\n".join(data)
 
         elif tool_call["name"] == "confirm_data":
@@ -126,7 +198,7 @@ def data_node(state: DataState) -> DataState:
 
             #order_text = "\n".join(order)
             print("Saving the data!")
-            #print(order_text)
+            pprint(data)
 
             # TODO(you!): Implement cafe.
             data_saved = True