[fix] convert key-type error in test script

Author: Irfnfnkemed

tests/python/serve/server/test_server_structural_tag.py

@@ -11,7 +11,7 @@
 import json
 import os
 import re
-from typing import Dict, List, Optional, Tuple
+from typing import Dict, List, Optional
 
 import pytest
 import requests
@@ -20,13 +20,13 @@
 
 
 def check_openai_nonstream_response(
-    response: Dict,
-    *,
-    model: str,
-    object_str: str,
-    num_choices: int,
-    finish_reason: List[str],
-    completion_tokens: Optional[int] = None,
+        response: Dict,
+        *,
+        model: str,
+        object_str: str,
+        num_choices: int,
+        finish_reason: List[str],
+        completion_tokens: Optional[int] = None,
 ):
     assert response["model"] == model
     assert response["object"] == object_str
@@ -68,16 +68,16 @@ def check_openai_nonstream_response(
 
 
 def check_openai_stream_response(
-    responses: List[Dict],
-    *,
-    model: str,
-    object_str: str,
-    num_choices: int,
-    finish_reason: str,
-    echo_prompt: Optional[str] = None,
-    suffix: Optional[str] = None,
-    stop: Optional[List[str]] = None,
-    require_substr: Optional[List[str]] = None,
+        responses: List[Dict],
+        *,
+        model: str,
+        object_str: str,
+        num_choices: int,
+        finish_reason: str,
+        echo_prompt: Optional[str] = None,
+        suffix: Optional[str] = None,
+        stop: Optional[List[str]] = None,
+        require_substr: Optional[List[str]] = None,
 ):
     assert len(responses) > 0
 
@@ -219,7 +219,6 @@ def check_format(name_beg: str, name_end: str, beg_tag: str, schema: str):
 You are a helpful assistant.""",
 }
 
-
 STRUCTURAL_TAGS = {
     "triggers": ["<CALL--->", "<call--->"],
     "tags": [
@@ -236,14 +235,14 @@ def check_format(name_beg: str, name_end: str, beg_tag: str, schema: str):
                         "state": {
                             "type": "string",
                             "description": "the two-letter abbreviation for the state that the city is"
-                            " in, e.g. 'CA' which would mean 'California'",
+                                           " in, e.g. 'CA' which would mean 'California'",
                         },
                         "unit": {
                             "type": "string",
                             "description": "The unit to fetch the temperature in",
                             "enum": ["celsius", "fahrenheit"],
                         },
-                        "hash_code": {"const": 1234},
+                        "hash_code": {"const": "1234"},
                     },
                     "required": ["city", "state", "unit", "hash_code"],
                 }
@@ -260,7 +259,7 @@ def check_format(name_beg: str, name_end: str, beg_tag: str, schema: str):
                             "type": "string",
                             "description": "The timezone to fetch the current date and time for, e.g. 'America/New_York'",
                         },
-                        "hash_code": {"const": 2345},
+                        "hash_code": {"const": "2345"},
                     },
                     "required": ["timezone", "hash_code"],
                 }
@@ -280,14 +279,14 @@ def check_format(name_beg: str, name_end: str, beg_tag: str, schema: str):
                         "state": {
                             "type": "string",
                             "description": "the two-letter abbreviation for the state that the city is"
-                            " in, e.g. 'CA' which would mean 'California'",
+                                           " in, e.g. 'CA' which would mean 'California'",
                         },
                         "unit": {
                             "type": "string",
                             "description": "The unit to fetch the temperature in",
                             "enum": ["celsius", "fahrenheit"],
                         },
-                        "hash_code": {"const": 3456},
+                        "hash_code": {"const": "3456"},
                     },
                     "required": ["city", "state", "unit", "hash_code"],
                 }
@@ -304,7 +303,7 @@ def check_format(name_beg: str, name_end: str, beg_tag: str, schema: str):
                             "type": "string",
                             "description": "The timezone to fetch the current date and time for, e.g. 'America/New_York'",
                         },
-                        "hash_code": {"const": 4567},
+                        "hash_code": {"const": "4567"},
                     },
                     "required": ["timezone", "hash_code"],
                 }
@@ -315,29 +314,28 @@ def check_format(name_beg: str, name_end: str, beg_tag: str, schema: str):
 }
 
 CHECK_INFO = {
-    1234: {
+    "1234": {
         "name": "get_current_weather",
         "beg_tag": "CALL",
         "required": ["city", "state", "unit", "hash_code"],
     },
-    2345: {
+    "2345": {
         "name": "get_current_date",
         "beg_tag": "CALL",
         "required": ["timezone", "hash_code"],
     },
-    3456: {
+    "3456": {
         "name": "get_current_weather",
         "beg_tag": "call",
         "required": ["city", "state", "unit", "hash_code"],
     },
-    4567: {
+    "4567": {
         "name": "get_current_date",
         "beg_tag": "call",
         "required": ["timezone", "hash_code"],
     },
 }
 
-
 CHAT_COMPLETION_MESSAGES = [
     # messages #0
     [
@@ -369,10 +367,10 @@ def check_format(name_beg: str, name_end: str, beg_tag: str, schema: str):
 @pytest.mark.parametrize("stream", [False, True])
 @pytest.mark.parametrize("messages", CHAT_COMPLETION_MESSAGES)
 def test_openai_v1_chat_completion_structural_tag(
-    served_model: str,
-    launch_server,  # pylint: disable=unused-argument
-    stream: bool,
-    messages: List[Dict[str, str]],
+        served_model: str,
+        launch_server,  # pylint: disable=unused-argument
+        stream: bool,
+        messages: List[Dict[str, str]],
 ):
     # `served_model` and `launch_server` are pytest fixtures
     # defined in conftest.py.