@@ -28,7 +28,10 @@ def mock_field_blueprint_data():
2828 "updatedBy" : 1 ,
2929 "tags" : ["test" ],
3030 "fieldConfig" : {"some" : "config" },
31- "promptConfig" : {"localization" : "SEARCH" },
31+ "promptConfig" : {
32+ "description" : "Find organization names in the document" ,
33+ "localization" : "SEARCH" ,
34+ },
3235 }
3336
3437
@@ -47,6 +50,36 @@ def test_create_field_blueprint(mock_field_blueprint_data):
4750 assert result [0 ].uid == "test_uid"
4851 assert result [0 ].field_config ["some" ] == "config"
4952 assert result [0 ].prompt_config ["localization" ] == "SEARCH"
53+ assert (
54+ result [0 ].prompt_config ["description" ]
55+ == "Find organization names in the document"
56+ )
57+
58+
59+ def test_field_blueprint_queries_use_description_in_prompt_config ():
60+ for query in (
61+ CreateFieldBlueprint .query ,
62+ GetFieldBlueprints .query ,
63+ ListFieldBlueprints .query ,
64+ ):
65+ assert "description" in query
66+ assert "\n prompt\n " not in query
67+ assert "\n prompt\n " not in query
68+ assert "\n prompt\n " not in query
69+
70+
71+ def test_field_blueprint_queries_use_scalar_validation_config ():
72+ for query in (
73+ CreateFieldBlueprint .query ,
74+ GetFieldBlueprints .query ,
75+ ListFieldBlueprints .query ,
76+ ):
77+ assert "validationConfig" in query
78+ assert "formatConfig" in query
79+ assert "inputConfig" in query
80+ assert "validationConfig {" not in query
81+ assert "formatConfig {" not in query
82+ assert "inputConfig {" not in query
5083
5184
5285def test_get_field_blueprints (mock_field_blueprint_data ):
0 commit comments