Update: full intake tool set

modules/nodes.py

@@ -104,7 +104,7 @@ def chatbot_with_tools(state: DataState) -> DataState:
                         "plan": {
                             "goal": "",
                             "expectation": "",
-                            "alternative_treatment_illness": "",
+                            "alternative_treatment": "",
                         }
     }, "finished": False}
 

modules/tools.py

@@ -46,7 +46,7 @@ def pain(pain_location: str, pain_side: str, pain_intensity: int, pain_descripti
     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:
@@ -94,6 +94,40 @@ def review_system(weight_change: str, fever: bool, chill: bool, night_sweats: bo
       The updated data with the patient's review.
     """
 
+@tool
+def pain_manage(pain_medication: str, specialist: bool, other_therapy: str, effectiveness: bool) -> str:
+    """Collecting patient's pain management including:
+        - Current pain medications
+        - Specialist consultations
+        - Alternative therapies
+        - Treatment effectiveness
+        
+    Returns:
+      The updated data with the patient's pain management.
+    """
+
+@tool
+def functional(life_quality: str, limit_activity: str, mood: str) -> str:
+    """Collecting patient's functional assement information including:
+        - Impact on quality of life
+        - Activity limitations
+        - Mood and emotional state
+        
+    Returns:
+      The updated data with the patient's functional assessment information.
+    """
+
+@tool
+def plan(goal: str, expectation: str, alternative_treatment: str) -> str:
+    """Collecting patient's future treatment plan information including:
+        - Treatment goals
+        - Patient expectations
+        - Alternative treatment considerations
+        
+    Returns:
+      The updated data with the patient's future treatment plan information.
+    """
+
 @tool
 def confirm_data() -> str:
     """Asks the patient if the data intake is correct.
@@ -102,17 +136,14 @@ def confirm_data() -> str:
       The user's free-text response.
     """
 
-
 @tool
 def get_data() -> str:
     """Returns the users data so far. One item per line."""
 
-
 @tool
 def clear_data():
     """Removes all items from the user's order."""
 
-
 @tool
 def save_data() -> int:
     """Send the data into database.
@@ -121,7 +152,6 @@ def save_data() -> int:
       The status of data saving, finished.
     """
 
-
 def data_node(state: DataState) -> DataState:
     """The ordering node. This is where the dataintake is manipulated."""
     tool_msg = state.get("messages", [])[-1]
@@ -166,6 +196,48 @@ def data_node(state: DataState) -> DataState:
             data["medical_hist"]["allergy"] = tool_call["args"]["allergy"]
             response = "\n".join(data)
 
+        elif tool_call["name"] == "family_hist":
+            data["family_hist"]["family_history"] = tool_call["args"]["family_history"]
+            response = "\n".join(data)
+
+        elif tool_call["name"] == "social_hist":
+            data["social_hist"]["occupation"] = tool_call["args"]["occupation"]
+            data["social_hist"]["smoke"] = tool_call["args"]["smoke"]
+            data["social_hist"]["alcohol"] = tool_call["args"]["alcohol"]
+            data["social_hist"]["drug"] = tool_call["args"]["drug"]
+            data["social_hist"]["support_system"] = tool_call["args"]["support_system"]
+            data["social_hist"]["living_condition"] = tool_call["args"]["living_condition"]
+            response = "\n".join(data)
+
+        elif tool_call["name"] == "review_system":
+            data["review_system"]["weight_change"] = tool_call["args"]["weight_change"]
+            data["review_system"]["fever"] = tool_call["args"]["fever"]
+            data["review_system"]["chill"] = tool_call["args"]["chill"]
+            data["review_system"]["night_sweats"] = tool_call["args"]["night_sweats"]
+            data["review_system"]["sleep"] = tool_call["args"]["sleep"]
+            data["review_system"]["gastrointestinal"] = tool_call["args"]["gastrointestinal"]
+            data["review_system"]["urinary"] = tool_call["args"]["urinary"]            
+            response = "\n".join(data)
+        
+        elif tool_call["name"] == "pain_manage":
+            data["pain_manage"]["pain_medication"] = tool_call["args"]["pain_medication"]
+            data["pain_manage"]["specialist"] = tool_call["args"]["specialist"]
+            data["pain_manage"]["other_therapy"] = tool_call["args"]["other_therapy"]
+            data["pain_manage"]["effectiveness"] = tool_call["args"]["effectiveness"]         
+            response = "\n".join(data)
+
+        elif tool_call["name"] == "functional":
+            data["functional"]["life_quality"] = tool_call["args"]["life_quality"]
+            data["functional"]["limit_activity"] = tool_call["args"]["limit_activity"]
+            data["functional"]["mood"] = tool_call["args"]["mood"]
+            response = "\n".join(data)
+        
+        elif tool_call["name"] == "plan":
+            data["plan"]["goal"] = tool_call["args"]["goal"]
+            data["plan"]["expectation"] = tool_call["args"]["expectation"]
+            data["plan"]["alternative_treatment"] = tool_call["args"]["alternative_treatment"]
+            response = "\n".join(data)
+
         elif tool_call["name"] == "confirm_data":
 
             # We could entrust the LLM to do order confirmation, but it is a good practice to

requirements.txt

@@ -3,22 +3,12 @@ aiohttp==3.11.11
 aiosignal==1.3.2
 annotated-types==0.7.0
 anyio==4.8.0
-asttokens @ file:///opt/conda/conda-bld/asttokens_1646925590279/work
 attrs==25.1.0
-Bottleneck @ file:///croot/bottleneck_1731058641041/work
-Brotli @ file:///croot/brotli-split_1736182456865/work
 cachetools==5.5.1
 certifi==2024.12.14
 charset-normalizer==3.4.1
-comm @ file:///croot/comm_1709322850197/work
-contourpy @ file:///croot/contourpy_1732540045555/work
-cycler @ file:///tmp/build/80754af9/cycler_1637851556182/work
-debugpy @ file:///croot/debugpy_1736267418885/work
-decorator @ file:///opt/conda/conda-bld/decorator_1643638310831/work
 distro==1.9.0
-executing @ file:///opt/conda/conda-bld/executing_1646925071911/work
 filetype==1.2.0
-fonttools @ file:///croot/fonttools_1737039080035/work
 frozenlist==1.5.0
 google-ai-generativelanguage==0.6.15
 google-api-core==2.24.1
@@ -35,15 +25,13 @@ httpcore==1.0.7
 httplib2==0.22.0
 httpx==0.28.1
 idna==3.10
-ipykernel @ file:///croot/ipykernel_1737660677549/work
-ipython @ file:///croot/ipython_1734548052611/work
-jedi @ file:///croot/jedi_1733987392413/work
+ipykernel
+ipython
 jiter==0.8.2
 jsonpatch==1.33
 jsonpointer==3.0.0
-jupyter_client @ file:///croot/jupyter_client_1737570961872/work
-jupyter_core @ file:///croot/jupyter_core_1718818295206/work
-kiwisolver @ file:///croot/kiwisolver_1737039087198/work
+jupyter_client
+jupyter_core
 langchain==0.3.17
 langchain-core==0.3.33
 langchain-google-genai==2.0.9
@@ -55,63 +43,38 @@ langgraph-checkpoint==2.0.10
 langgraph-sdk==0.1.51
 langsmith==0.3.3
 matplotlib==3.10.0
-matplotlib-inline @ file:///work/perseverance-python-buildout/croot/matplotlib-inline_1698864771271/work
+matplotlib-inline
 mkl-service==2.4.0
-mkl_fft @ file:///io/mkl313/mkl_fft_1730824109137/work
-mkl_random @ file:///io/mkl313/mkl_random_1730823916628/work
 msgpack==1.1.0
 multidict==6.1.0
-nest-asyncio @ file:///croot/nest-asyncio_1708532673751/work
-numexpr @ file:///croot/numexpr_1730215937391/work
-numpy @ file:///croot/numpy_and_numpy_base_1708638617955/work/dist/numpy-1.26.4-cp312-cp312-linux_x86_64.whl#sha256=1d700f51d8b4fa684d858c9e3b56b1656bc5c82b6b79ff08d4e3b491c430059f
+numpy==1.26.4
 ollama==0.4.7
 openai==1.60.2
 orjson==3.10.15
-packaging @ file:///croot/packaging_1734472117206/work
-pandas @ file:///croot/pandas_1732735089971/work/dist/pandas-2.2.3-cp312-cp312-linux_x86_64.whl#sha256=57b66702d418720ec8483f7c4ec7c08d41815316ad7ce09d5b7bbc34eefcfdfd
-parso @ file:///croot/parso_1733963305961/work
-pexpect @ file:///tmp/build/80754af9/pexpect_1605563209008/work
-pillow @ file:///croot/pillow_1738010226202/work
-platformdirs @ file:///work/perseverance-python-buildout/croot/platformdirs_1701732573265/work
-prompt-toolkit @ file:///croot/prompt-toolkit_1704404351921/work
+pandas
 propcache==0.2.1
 proto-plus==1.26.0
 protobuf==5.29.3
-psutil @ file:///croot/psutil_1736367091698/work
-ptyprocess @ file:///tmp/build/80754af9/ptyprocess_1609355006118/work/dist/ptyprocess-0.7.0-py2.py3-none-any.whl
-pure-eval @ file:///opt/conda/conda-bld/pure_eval_1646925070566/work
 pyasn1==0.6.1
 pyasn1_modules==0.4.1
 pydantic==2.10.6
 pydantic_core==2.27.2
-Pygments @ file:///work/perseverance-python-buildout/croot/pygments_1698846270603/work
 pyparsing==3.2.1
-python-dateutil @ file:///croot/python-dateutil_1716495738603/work
 python-dotenv==1.0.1
-pytz @ file:///croot/pytz_1713974312559/work
 PyYAML==6.0.2
-pyzmq @ file:///croot/pyzmq_1734687138743/work
 regex==2024.11.6
 requests==2.32.3
 requests-toolbelt==1.0.0
 rsa==4.9
-seaborn @ file:///croot/seaborn_1718302919398/work
 setuptools==75.8.0
-six @ file:///tmp/build/80754af9/six_1644875935023/work
 sniffio==1.3.1
 SQLAlchemy==2.0.37
-stack-data @ file:///opt/conda/conda-bld/stack_data_1646927590127/work
 tenacity==9.0.0
 tiktoken==0.8.0
-tornado @ file:///croot/tornado_1733960490606/work
 tqdm==4.67.1
-traitlets @ file:///croot/traitlets_1718227057033/work
 typing_extensions==4.12.2
-tzdata @ file:///croot/python-tzdata_1690578112552/work
-unicodedata2 @ file:///croot/unicodedata2_1736541023050/work
 uritemplate==4.1.1
 urllib3==2.3.0
-wcwidth @ file:///Users/ktietz/demo/mc3/conda-bld/wcwidth_1629357192024/work
 wheel==0.45.1
 yarl==1.18.3
 zstandard==0.23.0