update: minor package path update

modules/data_class.py

@@ -1,6 +1,8 @@
 from typing_extensions import TypedDict
 from datetime import date
 from typing import Annotated, Dict, Any
+from langgraph.graph.message import add_messages
+
 
 
 class PainLevels(TypedDict):

modules/nodes.py

@@ -1,11 +1,12 @@
 from modules.data_class import DataState
 from modules.instrctions import MEDICAL_INTAKE_SYSINT, WELCOME_MSG
+from modules.llm_in_use import get_llm
+from modules.tools import patient_id, symptom, confirm_data, get_data, clear_data, save_data
+
 from datetime import date
 from typing import Literal
 from langgraph.graph import StateGraph, START, END
 from langchain_core.messages.ai import AIMessage
-from llm_in_use import get_llm
-from tools import patient_id, symptom, confirm_data, get_data, clear_data, save_data, data_node
 
 
 llm = get_llm()

paintrek-chat-v1.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 21,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2025-01-29T20:09:11.440091Z",

paintrek-chat-v2.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 1,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2025-01-29T20:09:11.440091Z",
@@ -16,27 +16,19 @@
    },
    "outputs": [
     {
-     "ename": "NameError",
-     "evalue": "name 'add_messages' is not defined",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mNameError\u001b[0m                                 Traceback (most recent call last)",
-      "Cell \u001b[0;32mIn[12], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mmodules\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mdata_class\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m DataState\n\u001b[1;32m      2\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mtools\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m data_node\n\u001b[1;32m      3\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mnodes\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m chatbot_with_tools, human_node, maybe_exit_human_node, maybe_route_to_tools\n",
-      "File \u001b[0;32m/media/frank-elite/Application/PythonProj/LangGraphExampleResume/modules/data_class.py:98\u001b[0m\n\u001b[1;32m     88\u001b[0m     main: MainSymptom\n\u001b[1;32m     89\u001b[0m \u001b[38;5;250m    \u001b[39m\u001b[38;5;124;03m\"\"\"pain: Pain\u001b[39;00m\n\u001b[1;32m     90\u001b[0m \u001b[38;5;124;03m    medical_hist: MedicalHistory\u001b[39;00m\n\u001b[1;32m     91\u001b[0m \u001b[38;5;124;03m    family_hist: FamilyHistory\u001b[39;00m\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m     95\u001b[0m \u001b[38;5;124;03m    functional: Functional\u001b[39;00m\n\u001b[1;32m     96\u001b[0m \u001b[38;5;124;03m    plan: Plan\"\"\"\u001b[39;00m\n\u001b[0;32m---> 98\u001b[0m \u001b[38;5;28;01mclass\u001b[39;00m \u001b[38;5;21;01mDataState\u001b[39;00m(TypedDict):\n\u001b[1;32m     99\u001b[0m \u001b[38;5;250m    \u001b[39m\u001b[38;5;124;03m\"\"\"State representing the patient's data status and conversation.\"\"\"\u001b[39;00m\n\u001b[1;32m    100\u001b[0m     messages: Annotated[\u001b[38;5;28mlist\u001b[39m, add_messages]\n",
-      "File \u001b[0;32m/media/frank-elite/Application/PythonProj/LangGraphExampleResume/modules/data_class.py:100\u001b[0m, in \u001b[0;36mDataState\u001b[0;34m()\u001b[0m\n\u001b[1;32m     98\u001b[0m \u001b[38;5;28;01mclass\u001b[39;00m \u001b[38;5;21;01mDataState\u001b[39;00m(TypedDict):\n\u001b[1;32m     99\u001b[0m \u001b[38;5;250m    \u001b[39m\u001b[38;5;124;03m\"\"\"State representing the patient's data status and conversation.\"\"\"\u001b[39;00m\n\u001b[0;32m--> 100\u001b[0m     messages: Annotated[\u001b[38;5;28mlist\u001b[39m, add_messages]\n\u001b[1;32m    101\u001b[0m     data: Dict[\u001b[38;5;28mstr\u001b[39m, PatientData]\n\u001b[1;32m    102\u001b[0m     finished: \u001b[38;5;28mbool\u001b[39m\n",
-      "\u001b[0;31mNameError\u001b[0m: name 'add_messages' is not defined"
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/frank-elite/miniconda3/envs/paintrekbot/lib/python3.12/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
+      "  from .autonotebook import tqdm as notebook_tqdm\n"
      ]
     }
    ],
    "source": [
-    "\n",
-    "\n",
     "from modules.data_class import DataState\n",
-    "from tools import data_node\n",
-    "from nodes import chatbot_with_tools, human_node, maybe_exit_human_node, maybe_route_to_tools\n",
+    "from modules.tools import data_node\n",
+    "from modules.nodes import chatbot_with_tools, human_node, maybe_exit_human_node, maybe_route_to_tools\n",
     "\n",
-    "from langgraph.graph.message import add_messages\n",
     "from langgraph.graph import StateGraph, START, END\n",
     "\n",
     "from IPython.display import Image, display\n",
@@ -50,7 +42,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2025-01-29T20:09:11.906458Z",
@@ -70,7 +62,7 @@
        "<IPython.core.display.Image object>"
       ]
      },
-     "execution_count": 13,
+     "execution_count": 2,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -99,7 +91,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 3,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2025-01-29T20:09:38.185616Z",
@@ -118,33 +110,6 @@
      "text": [
       "Model: Welcome to the Paintrek world. I am a health assistant, an interactive clinical recording system. I will ask you questions about your pain and related symptoms and record your responses.  I will then store this information securely. At any time, you can type `q` to quit.\n"
      ]
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Model: I apologize for the format. Let's start again.\n",
-      "\n",
-      "Hello, my name is MedAssist, and I'll be guiding you through a structured data collection process to gather comprehensive patient information. Please feel free to share any concerns or symptoms you're experiencing.\n",
-      "\n",
-      "To begin, can you please tell me your full name?\n",
-      "Model: I think there's been a mistake! Let's start again from the beginning.\n",
-      "\n",
-      "Hello, my name is MedAssist, and I'll be guiding you through a structured data collection process to gather comprehensive patient information. Please feel free to share any concerns or symptoms you're experiencing.\n",
-      "\n",
-      "To confirm, your full name is indeed John Doe, as per our initial introduction. Is that correct?\n",
-      "Model: Thank you for correcting me, Frank. I've updated your patient information.\n",
-      "\n",
-      "Now, let's move on to the next step. Can you please tell me what brings you here today? What seems to be the main reason for your visit? Is it related to any specific symptoms or concerns you're experiencing?\n",
-      "Model: Thank you for sharing that with me, Frank. A headache can be quite uncomfortable.\n",
-      "\n",
-      "Can you please tell me more about your headache? Where exactly is the pain located on a scale of 0 to 10, with 0 being no pain at all and 10 being the worst possible pain?\n",
-      "\n",
-      "Also, how long have you been experiencing this headache? Was it sudden or did it come on gradually?\n",
-      "Model: I apologize if I lost you for a moment. Can you please tell me more about your headache? Where exactly is the pain located on a scale of 0 to 10, with 0 being no pain at all and 10 being the worst possible pain?\n",
-      "\n",
-      "Also, how long have you been experiencing this headache? Was it sudden or did it come on gradually?\n"
-     ]
     }
    ],
    "source": [