update: some tests were not but not working

modules/llm_in_use.py

@@ -14,13 +14,14 @@ def get_llm():
     # llm = ChatOpenAI(temperature=0)
 
     # llm = ChatOllama(model="llama3.2:3b", temperature=0, base_url="http://141.211.127.171")
-    # llm = ChatOllama(model="qwen2.5:14b", temperature=0, base_url="http://141.211.127.171")
+    llm = ChatOllama(model="qwen2.5:14b", temperature=0, base_url="http://141.211.127.171")
     # llm = ChatOllama(model="nexusraven:13b", temperature=0, base_url="http://141.211.127.171")
     
     """llm = ChatOpenAI(
         api_key="ollama",
-        model="llama3.2:latest",
+        model="qwen2.5:14b",
         base_url="http://141.211.127.171/",
+        temperature=0
     )"""
 
 

paintrek-chat-v2.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2025-01-29T20:09:11.440091Z",
@@ -36,7 +36,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2025-01-29T20:09:11.906458Z",
@@ -69,12 +69,12 @@
     "graph_builder.add_edge(START, \"chatbot_healthassistant\")\n",
     "graph_with_order_tools = graph_builder.compile(checkpointer=memory)\n",
     "\n",
-    "Image(graph_with_order_tools.get_graph().draw_mermaid_png())"
+    "# Image(graph_with_order_tools.get_graph().draw_mermaid_png())"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 3,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2025-01-29T20:09:38.185616Z",
@@ -86,7 +86,32 @@
     "id": "NCRSgaBUfIHF",
     "trusted": true
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "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: Great! Let’s begin by gathering some basic information. Could you please provide your full name, date of birth, and contact number? This will help us proceed with the medical record setup.\n",
+      "User: My name is Frank my DOB is 19860214, my contact number is 123456\n",
+      "Model: Thank you for providing your information, Frank. Could you please confirm your full date of birth and provide any additional contact details such as an email address or another phone number? Also, could you tell me about your emergency contacts if applicable?\n",
+      "If you're ready to proceed with the rest of the intake process now, let me know!\n",
+      "User: my date of birth is Feb 14 1986, emergency contact is Zoe and her phone number is 568790\n",
+      "Your input data:\n",
+      "{'ID': {'name': '', 'DOB': datetime.date(1900, 1, 1), 'gender': '', 'contact': '', 'emergency_contact': ''}, 'symptom': {'main_symptom': '', 'symptom_length': ''}, 'pain': {'pain_location': '', 'pain_side': '', 'pain_intensity': 0, 'pain_description': '', 'start_time': datetime.date(1900, 1, 1), 'radiation': False, 'triggers': '', 'symptom': ''}, 'medical_hist': {'medical_condition': '', 'first_time': datetime.date(1900, 1, 1), 'surgery_history': '', 'medication': '', 'allergy': ''}, 'family_hist': {'family_history': ''}, 'social_hist': {'occupation': '', 'smoke': False, 'alcohol': False, 'drug': False, 'support_system': '', 'living_condition': ''}, 'review_system': {'weight_change': '', 'fever': False, 'chill': False, 'night_sweats': False, 'sleep': '', 'gastrointestinal': '', 'urinary': ''}, 'pain_manage': {'pain_medication': '', 'specialist': False, 'other_therapy': '', 'effectiveness': False}, 'functional': {'life_quality': '', 'limit_activity': '', 'mood': ''}, 'plan': {'goal': '', 'expectation': '', 'alternative_treatment': ''}}\n",
+      "Model: Let's start by gathering your personal information and medical history. Could you please provide me with the following details:\n",
+      "\n",
+      "1. Your full name, date of birth, and contact information.\n",
+      "2. The reason for your visit today (symptoms or concerns).\n",
+      "3. Any current pain issues you are experiencing.\n",
+      "\n",
+      "Once I have this initial information, we can proceed to gather more detailed medical history and other relevant data. Please start with the first two points if possible. For the third point, could you describe any areas where you're feeling pain, how long it has been going on, and what makes it better or worse?\n",
+      "User: q\n"
+     ]
+    }
+   ],
    "source": [
     "# This is for the checkpointer to have a thread id to remember for each user, and I don't know if letter mix with number will work here \n",
     "# The default recursion limit for traversing nodes is 25 - setting it higher means you can try a more complex order with multiple steps and round-trips.\n",

test_ollama.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -12,7 +12,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -21,20 +21,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "AIMessage(content='Je aime le programmation.', additional_kwargs={}, response_metadata={'model': 'llama3.2:3b', 'created_at': '2025-03-03T20:22:10.986078Z', 'done': True, 'done_reason': 'stop', 'total_duration': 564767250, 'load_duration': 36439292, 'prompt_eval_count': 45, 'prompt_eval_duration': 207000000, 'eval_count': 7, 'eval_duration': 45000000, 'message': Message(role='assistant', content='', images=None, tool_calls=None)}, id='run-b473d2e4-5ea8-49c1-880c-57529bab3be3-0', usage_metadata={'input_tokens': 45, 'output_tokens': 7, 'total_tokens': 52})"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "from langchain_core.messages import AIMessage\n",
     "\n",
@@ -51,20 +40,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "AIMessage(content='Ich liebe Programmierung.', additional_kwargs={}, response_metadata={'model': 'llama3.2:3b', 'created_at': '2025-03-03T20:27:19.219749Z', 'done': True, 'done_reason': 'stop', 'total_duration': 1047082250, 'load_duration': 580482958, 'prompt_eval_count': 40, 'prompt_eval_duration': 161000000, 'eval_count': 6, 'eval_duration': 38000000, 'message': Message(role='assistant', content='', images=None, tool_calls=None)}, id='run-bbaf365e-81ba-4928-b6a3-8578efe20cc5-0', usage_metadata={'input_tokens': 40, 'output_tokens': 6, 'total_tokens': 46})"
-      ]
-     },
-     "execution_count": 4,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "from langchain_core.prompts import ChatPromptTemplate\n",
     "\n",
@@ -90,24 +68,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[{'name': 'validate_user',\n",
-       "  'args': {'addresses': ['123 Fake St', '234 Pretend Boulevard'],\n",
-       "   'user_id': 123},\n",
-       "  'id': '0401ef59-16fb-446c-8fa2-11c2ad671d44',\n",
-       "  'type': 'tool_call'}]"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "from typing import List\n",
     "\n",