zzstoatzz commited on
Commit
903487a
·
1 Parent(s): b23a78c

update references to old class names

Browse files
Files changed (1) hide show
  1. tests/test_func_metadata.py +8 -8
tests/test_func_metadata.py CHANGED
@@ -241,12 +241,12 @@ def test_complex_function_json_schema():
241
  "title": "InnerModel",
242
  "type": "object",
243
  },
244
- "TestInputModelA": {
245
  "properties": {},
246
- "title": "TestInputModelA",
247
  "type": "object",
248
  },
249
- "TestInputModelB": {
250
  "properties": {
251
  "how_many_shrimp": {
252
  "description": "How many shrimp in the tank???",
@@ -257,7 +257,7 @@ def test_complex_function_json_schema():
257
  "y": {"title": "Y", "type": "null"},
258
  },
259
  "required": ["how_many_shrimp", "ok", "y"],
260
- "title": "TestInputModelB",
261
  "type": "object",
262
  },
263
  },
@@ -301,9 +301,9 @@ def test_complex_function_json_schema():
301
  "type": "integer",
302
  },
303
  "unannotated": {"title": "unannotated", "type": "string"},
304
- "my_model_a": {"$ref": "#/$defs/TestInputModelA"},
305
- "my_model_a_forward_ref": {"$ref": "#/$defs/TestInputModelA"},
306
- "my_model_b": {"$ref": "#/$defs/TestInputModelB"},
307
  "an_int_annotated_with_field_default": {
308
  "default": 1,
309
  "description": "An int with a field",
@@ -316,7 +316,7 @@ def test_complex_function_json_schema():
316
  "type": "string",
317
  },
318
  "my_model_a_with_default": {
319
- "$ref": "#/$defs/TestInputModelA",
320
  "default": {},
321
  },
322
  "an_int_with_default": {
 
241
  "title": "InnerModel",
242
  "type": "object",
243
  },
244
+ "SomeInputModelA": {
245
  "properties": {},
246
+ "title": "SomeInputModelA",
247
  "type": "object",
248
  },
249
+ "SomeInputModelB": {
250
  "properties": {
251
  "how_many_shrimp": {
252
  "description": "How many shrimp in the tank???",
 
257
  "y": {"title": "Y", "type": "null"},
258
  },
259
  "required": ["how_many_shrimp", "ok", "y"],
260
+ "title": "SomeInputModelB",
261
  "type": "object",
262
  },
263
  },
 
301
  "type": "integer",
302
  },
303
  "unannotated": {"title": "unannotated", "type": "string"},
304
+ "my_model_a": {"$ref": "#/$defs/SomeInputModelA"},
305
+ "my_model_a_forward_ref": {"$ref": "#/$defs/SomeInputModelA"},
306
+ "my_model_b": {"$ref": "#/$defs/SomeInputModelB"},
307
  "an_int_annotated_with_field_default": {
308
  "default": 1,
309
  "description": "An int with a field",
 
316
  "type": "string",
317
  },
318
  "my_model_a_with_default": {
319
+ "$ref": "#/$defs/SomeInputModelA",
320
  "default": {},
321
  },
322
  "an_int_with_default": {