adding tools

app.py

@@ -548,6 +548,48 @@ def sms_send_test(prompt):
     else:
         return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
 
+def service_mode_tests(prompt):
+    test_name = "service_mode_tests"
+    content_one_time_test = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt the Start_Date is present, No if is missing. IF 'null' is in the input , the answer is NO. ACCEPTED ANSWER : YES / NO."
+    content_multiple_tests = "you only answer in shematic format,possibile asnwer are YES, NO. YES if in the prompt, both  Number_of_tests, Start_Date are present, No if even only one is missing.IF 'null' is in the input , the answer is NO. THERE CAN BE MORE IN THE PROMPT, YOU DON'T HAVE TO CAARE ABOUT IT. ACCEPTED ANSWER : YES / NO."
+
+    if "One time test" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        response = get_chat_completion(prompt, content_one_time_test)
+        if "yes" in str(response).lower():
+            print(prompt)
+            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test, Start_Date: selected_start_date}}"
+        else:
+            return "There are some infomration needed missing, information needed: Start_Date "
+
+    elif "Periodic test" in prompt or "Driving Mode" in prompt:
+        if "now" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("now", formatted_date)
+        elif "today" in prompt:
+            now = datetime.now()
+            formatted_date = str(now.strftime("%d/%m/%Y"))
+            prompt = prompt.replace("today", formatted_date)
+
+        response = get_chat_completion(prompt, content_multiple_tests)
+        if "yes" in str(response).lower():
+            print(prompt)
+            return f"report to the user the following json file , test started successfully: {{test_name: {test_name}, type of test : selected_type_of_test,  Number_of_tests: selected_number_of_tests, Start_Date: selected_start_date}}"
+        else:
+            return "There are some infomration needed missing, information needed: Number_of_tests, Start_Date "
+    else:
+        return "seems like the user did not specify the type of test: Possible choiches: 1.One time test, 2.Periodic test, 3.Driving Mode "
+
+
 
 tools = [
     Tool(
@@ -684,11 +726,17 @@ tools = [
                         3: Driving Mode. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy], Host_list: list, phone_number: int, sms_message: str, recover_email: str[optional]
                         So first of all you have to understand which tetst does the user wants to do. ask DIRECTLLY the user which test options they want to perform, don't try to guess it. Then you have to find the informaton needed for that test.Users alway forget something, if you are missing some info, just pass none.In the field date,IF AND ONLY IF users says now or today, just pass now or today. NEVER CREATE INPUT IF NOT WRITTEN BY THE USER. IN THE INPUT PASS THE INFORMATION IN JSON FORMAT. IN THE INPUR PASS THE SPECIFIED TET TYPE, ask if missing test_type:(One time test/Periodic test/Driving Mode)"""
     ),
-
-
-
-
-
+        Tool(
+        name="service_mode_tests",
+        func=lambda prompt: service_mode_tests(prompt),
+        description="""Use when the user wants to do a service_mode_tests, 
+                        start the tool IF AND ONLY IF  you have ALL the information needed.
+                        There are three options for this test, 
+                        1: One time test. Information Needed : (Start_Date : str[dd/mm/yy].)
+                        2: Periodic test. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy].)
+                        3: Driving Mode. Information Needed : (Number_of_tests : int, Start_Date : str[dd/mm/yy].)
+                        So first of all you have to understand which tetst does the user wants to do.ask DIRECTLLY the user which test options they want to perform, don't try to guess it. Then you have to find the informaton needed for that test.Users alway forget something, if you are missing some info, just pass none.In the field date,IF AND ONLY IF users says now or today, just pass now or today. NEVER CREATE INPUT IF NOT WRITTEN BY THE USER. IN THE INPUT PASS THE INFORMATION IN JSON FORMAT. IN THE INPUR PASS THE SPECIFIED TET TYPE, ask if missing test_type:(One time test/Periodic test/Driving Mode)"""
+    ),
 ]
 
 # Set Streamlit page configuration