update: ollama model test

modules/llm_in_use.py

@@ -7,19 +7,14 @@ from langchain_ollama import ChatOllama
 def get_llm():
     # Load environment variables from .env file
     load_dotenv()
+
     # Get the API key
     # google_api_key = os.getenv("GOOGLE_API_KEY")
 
     # llm = ChatOpenAI(temperature=0)
 
-    # llm = ChatOllama(model="llama3.2:latest", temperature=0)
-
-    """llm = ChatOpenAI(
-        api_key="ollama",
-        model="llama3.2:latest",
-        base_url="http://141.211.127.171/",
-    )"""
+    llm = ChatOllama(model="llama3.2:latest", temperature=0, base_url="http://141.211.127.171")
 
-    llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash-latest")
+    # llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash-latest")
 
     return llm

paintrek-chat-v2.ipynb

@@ -14,7 +14,16 @@
     "id": "xaiioUQni_ga",
     "trusted": true
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "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": [
     "from modules.data_class import DataState\n",
     "from modules.tools import data_node\n",
@@ -105,33 +114,55 @@
      "text": [
       "Executing the chatbot graph...\n",
       "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",
-      "User: Let's start\n",
-      "Model: First, let's collect some basic information for identification.  What is your full name, date of birth (YYYY-MM-DD), gender, and contact number?\n",
-      "User: I am Frank, DOB 1986-01-01, male, phone number is 12345, emergency name is Zoe, number is 67890.\n",
-      "Model: Thank you. Now, can you describe your main symptom and how long you've been experiencing it?\n",
-      "User: I had a headache, and it started two days ago.\n",
-      "Model: Okay. Let's move on to your pain.  Can you tell me about the location, side (left, right, or both), and intensity (on a scale of 0-10, with 0 being no pain and 10 being the worst pain imaginable) of your headache?  Also, please describe the characteristics of your pain (e.g., sharp, dull, throbbing), when it started, if it radiates to other areas, what triggers it, and if there are any associated symptoms.\n",
-      "User: My pain is located on the right head, and the scale is 2, the pain is sharp, the pain started 2 days ago, and the pain is not radiating. Cold will trigger the pain, and no other symptoms. \n",
-      "Model: To continue, could you provide information on your medical history, including any existing medical conditions, when they first appeared, any surgeries you've had and when, your current medications, and any allergies you have?\n",
-      "User: I had sinus, diagnosed in 2001-01-01, I did not have any surgery. I am taking ibprofin, no allergies.\n",
-      "Model: Thank you.  Now, let's discuss your family history.  Are there any significant medical conditions that run in your family?\n",
-      "User: My mom had a lung cancer at 45.\n",
-      "Model: Could you share some information about your social history, including your occupation, whether you smoke or drink alcohol, your living situation, and your support system?\n",
-      "User: I am a data scientist, I am a viggie. I don't use tobbaco or alcohol. My stress level is midium. \n",
-      "Model: Let's proceed with a systems review.  Have you experienced any recent weight changes, fever, chills, night sweats, changes in sleep patterns, gastrointestinal issues, or urinary problems?\n",
-      "User: I sleep well, and no other symptoms.\n",
-      "Model: Now, let's discuss your pain management history. What pain medications are you currently using, have you consulted any specialists for your pain, have you tried any alternative therapies, and how effective have these treatments been?\n",
-      "User: No, I am not taking any other medication, and I haven't tried any alternative therapies. \n",
-      "Model: Finally, let's assess your functional status. How has this pain affected your quality of life, what activities have you been limited in, and how would you describe your current mood and emotional state?\n",
-      "User: I am okay on my mood, and my quality of life is also good. I want to fix my headache.\n",
-      "Model: Before we conclude, could you please confirm that all the information I have gathered is accurate?  If there are any corrections or additions, please let me know now.\n",
-      "User: Can you print the data you have already collected?\n",
-      "Model: Okay, everything seems correct.\n",
+      "User: Hello\n",
+      "Model: I'm MedAssist, your intelligent medical intake system. I'll guide you through a structured data collection process to gather comprehensive patient information.\n",
+      "\n",
+      "To begin, can you please tell me your name and date of birth? This will help me identify you in our records and ensure accurate communication throughout the process.\n",
       "User: \n",
-      "Your input data:\n",
-      "{'ID': {'name': 'Frank', 'DOB': '1986-01-01', 'gender': 'male', 'contact': '12345', 'emergency_contact': '67890'}, 'symptom': {'main_symptom': 'headache', 'symptom_length': 'two days'}, 'pain': {'pain_location': 'right head', 'pain_side': 'right', 'pain_intensity': 2.0, 'pain_description': 'sharp', 'start_time': '2 days ago', 'radiation': False, 'triggers': 'cold', 'symptom': 'headache'}, 'medical_hist': {'medical_condition': 'sinus', 'first_time': '2001-01-01', 'surgery_history': 'None', 'medication': 'ibprofin', 'allergy': 'None'}, 'family_hist': {'family_history': 'Mother had lung cancer at 45'}, 'social_hist': {'occupation': 'data scientist', 'smoke': False, 'alcohol': False, 'drug': False, 'support_system': 'unknown', 'living_condition': 'medium stress'}, 'review_system': {'weight_change': 'None', 'fever': False, 'chill': False, 'night_sweats': False, 'sleep': 'well', 'gastrointestinal': 'None', 'urinary': 'None'}, 'pain_manage': {'pain_medication': 'None', 'specialist': False, 'other_therapy': 'None', 'effectiveness': False}, 'functional': {'life_quality': 'good', 'limit_activity': 'None', 'mood': 'okay'}, 'plan': {'goal': '', 'expectation': '', 'alternative_treatment': ''}}\n",
-      "Saving the data!\n",
-      "{'ID': {'name': 'Frank', 'DOB': '1986-01-01', 'gender': 'male', 'contact': '12345', 'emergency_contact': '67890'}, 'symptom': {'main_symptom': 'headache', 'symptom_length': 'two days'}, 'pain': {'pain_location': 'right head', 'pain_side': 'right', 'pain_intensity': 2.0, 'pain_description': 'sharp', 'start_time': '2 days ago', 'radiation': False, 'triggers': 'cold', 'symptom': 'headache'}, 'medical_hist': {'medical_condition': 'sinus', 'first_time': '2001-01-01', 'surgery_history': 'None', 'medication': 'ibprofin', 'allergy': 'None'}, 'family_hist': {'family_history': 'Mother had lung cancer at 45'}, 'social_hist': {'occupation': 'data scientist', 'smoke': False, 'alcohol': False, 'drug': False, 'support_system': 'unknown', 'living_condition': 'medium stress'}, 'review_system': {'weight_change': 'None', 'fever': False, 'chill': False, 'night_sweats': False, 'sleep': 'well', 'gastrointestinal': 'None', 'urinary': 'None'}, 'pain_manage': {'pain_medication': 'None', 'specialist': False, 'other_therapy': 'None', 'effectiveness': False}, 'functional': {'life_quality': 'good', 'limit_activity': 'None', 'mood': 'okay'}, 'plan': {'goal': '', 'expectation': '', 'alternative_treatment': ''}}\n"
+      "Model: Here is the JSON payload that can be sent to the server to collect all the necessary information:\n",
+      "\n",
+      "```json\n",
+      "{\n",
+      "  \"name\": \"\",\n",
+      "  \"family_hist\": \"\",\n",
+      "  \"social_hist\": {\n",
+      "    \"occupation\": \"\",\n",
+      "    \"smoke\": false,\n",
+      "    \"alcohol\": false,\n",
+      "    \"drug\": false,\n",
+      "    \"support_system\": \"\",\n",
+      "    \"living_condition\": \"\"\n",
+      "  },\n",
+      "  \"pain_manage\": {\n",
+      "    \"pain_medication\": \"\",\n",
+      "    \"specialist\": false,\n",
+      "    \"other_therapy\": \"\",\n",
+      "    \"effectiveness\": false\n",
+      "  },\n",
+      "  \"functional\": {\n",
+      "    \"life_quality\": \"\",\n",
+      "    \"limit_activity\": \"\",\n",
+      "    \"mood\": \"\"\n",
+      "  },\n",
+      "  \"plan\": {\n",
+      "    \"goal\": \"\",\n",
+      "    \"expectation\": \"\",\n",
+      "    \"alternative_treatment\": \"\"\n",
+      "  },\n",
+      "  \"review_system\": {\n",
+      "    \"weight_change\": \"\",\n",
+      "    \"fever\": false,\n",
+      "    \"chill\": false,\n",
+      "    \"night_sweats\": false,\n",
+      "    \"sleep\": \"\",\n",
+      "    \"gastrointestinal\": \"\",\n",
+      "    \"urinary\": \"\"\n",
+      "  }\n",
+      "}\n",
+      "```\n",
+      "\n",
+      "This payload includes all the necessary fields to collect information about the patient's family history, social history, pain management, functional assessment, and review system. Note that some of these fields are optional (e.g., `smoke`, `alcohol`, etc.) and can be set to `false` if not applicable.\n",
+      "User: q\n"
      ]
     }
    ],
@@ -155,72 +186,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {
     "trusted": true
    },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "{'ID': {'name': 'Frank',\n",
-       "  'DOB': '1986-01-01',\n",
-       "  'gender': 'male',\n",
-       "  'contact': '12345',\n",
-       "  'emergency_contact': '67890'},\n",
-       " 'symptom': {'main_symptom': 'headache', 'symptom_length': 'two days'},\n",
-       " 'pain': {'pain_location': 'right head',\n",
-       "  'pain_side': 'right',\n",
-       "  'pain_intensity': 2.0,\n",
-       "  'pain_description': 'sharp',\n",
-       "  'start_time': '2 days ago',\n",
-       "  'radiation': False,\n",
-       "  'triggers': 'cold',\n",
-       "  'symptom': 'headache'},\n",
-       " 'medical_hist': {'medical_condition': 'sinus',\n",
-       "  'first_time': '2001-01-01',\n",
-       "  'surgery_history': 'None',\n",
-       "  'medication': 'ibprofin',\n",
-       "  'allergy': 'None'},\n",
-       " 'family_hist': {'family_history': 'Mother had lung cancer at 45'},\n",
-       " 'social_hist': {'occupation': 'data scientist',\n",
-       "  'smoke': False,\n",
-       "  'alcohol': False,\n",
-       "  'drug': False,\n",
-       "  'support_system': 'unknown',\n",
-       "  'living_condition': 'medium stress'},\n",
-       " 'review_system': {'weight_change': 'None',\n",
-       "  'fever': False,\n",
-       "  'chill': False,\n",
-       "  'night_sweats': False,\n",
-       "  'sleep': 'well',\n",
-       "  'gastrointestinal': 'None',\n",
-       "  'urinary': 'None'},\n",
-       " 'pain_manage': {'pain_medication': 'None',\n",
-       "  'specialist': False,\n",
-       "  'other_therapy': 'None',\n",
-       "  'effectiveness': False},\n",
-       " 'functional': {'life_quality': 'good',\n",
-       "  'limit_activity': 'None',\n",
-       "  'mood': 'okay'},\n",
-       " 'plan': {'goal': '', 'expectation': '', 'alternative_treatment': ''}}"
-      ]
-     },
-     "execution_count": 4,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "state[\"data\"]"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {
@@ -238,7 +211,7 @@
    "sourceType": "notebook"
   },
   "kernelspec": {
-   "display_name": "paintrek",
+   "display_name": "paintrekbot",
    "language": "python",
    "name": "python3"
   },
@@ -252,7 +225,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.13"
+   "version": "3.12.8"
   }
  },
  "nbformat": 4,