title List -> Array

guardrails-ai · Mar 13, 2024 · 82500da · 82500da
1 parent cf3952e
commit 82500da
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/guardrails/utils/pydantic_utils/v1.py b/guardrails/utils/pydantic_utils/v1.py
@@ -340,7 +340,7 @@ def convert_pydantic_model_to_openai_fn(
 
     # if type_origin == list:
     #     json_schema = {
-    #         "title": f"List<{json_schema.get('title')}>",
+    #         "title": f"Array<{json_schema.get('title')}>",
     #         "type": "array",
     #         "items": json_schema,
     #     }

diff --git a/guardrails/utils/pydantic_utils/v2.py b/guardrails/utils/pydantic_utils/v2.py
@@ -157,7 +157,7 @@ def convert_pydantic_model_to_openai_fn(
 
     if type_origin == list:
         json_schema = {
-            "title": f"List<{json_schema.get('title')}>",
+            "title": f"Array<{json_schema.get('title')}>",
             "type": "array",
             "items": json_schema,
         }

diff --git a/tests/unit_tests/utils/pydantic_utils/v1.py b/tests/unit_tests/utils/pydantic_utils/v1.py
@@ -64,15 +64,15 @@ def test_list_schema(self):
 
         # fmt: off
         expected_schema = {
-            "title": f"List<{expected_schema.get('title')}>",
+            "title": f"Array<{expected_schema.get('title')}>",
             "type": "array",
             "items": expected_schema
         }
         # fmt: on
 
         # fmt: off
         expected_fn_params = {  # noqa
-            "name": "List<Foo>",
+            "name": "Array<Foo>",
             "parameters": expected_schema
         }
         # fmt: on

diff --git a/tests/unit_tests/utils/pydantic_utils/v2.py b/tests/unit_tests/utils/pydantic_utils/v2.py
@@ -57,15 +57,15 @@ def test_list_schema(self):
 
         # fmt: off
         expected_schema = {
-            "title": f"List<{expected_schema.get('title')}>",
+            "title": f"Array<{expected_schema.get('title')}>",
             "type": "array",
             "items": expected_schema
         }
         # fmt: on
 
         # fmt: off
         expected_fn_params = {
-            "name": "List<Foo>",
+            "name": "Array<Foo>",
             "parameters": expected_schema
         }
         # fmt: on