Update _retrieve_ref inside json_schema.py to include an isdigit() check (#14745)

- **Description:** Update _retrieve_ref inside json_schema.py to include
an isdigit() check
- **Issue:** This library is used inside dereference_refs inside
langchain_community.agent_toolkits.openapi.spec. When I read in a yaml
file which has references for "400", "401" etc; the line "out =
out[component]" causes a KeyError. The isdigit() check ensures that if
it is an integer like "400" or "401"; it converts it into integer before
using it as a key to prevent the error.
  - **Dependencies:** No dependencies
  - **Tag maintainer:** @baskaryan

---------

Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>

libs/core/langchain_core/utils/json_schema.py

@@ -13,7 +13,10 @@ def _retrieve_ref(path: str, schema: dict) -> dict:
         )
     out = schema
     for component in components[1:]:
-        out = out[component]
+        if component.isdigit():
+            out = out[int(component)]
+        else:
+            out = out[component]
     return deepcopy(out)
 
 

libs/core/tests/unit_tests/utils/test_json_schema.py

@@ -149,3 +149,35 @@ def test_dereference_refs_remote_ref() -> None:
     }
     with pytest.raises(ValueError):
         dereference_refs(schema)
+
+
+def test_dereference_refs_integer_ref() -> None:
+    schema = {
+        "type": "object",
+        "properties": {
+            "error_400": {"$ref": "#/$defs/400"},
+        },
+        "$defs": {
+            400: {
+                "type": "object",
+                "properties": {"description": "Bad Request"},
+            },
+        },
+    }
+    expected = {
+        "type": "object",
+        "properties": {
+            "error_400": {
+                "type": "object",
+                "properties": {"description": "Bad Request"},
+            },
+        },
+        "$defs": {
+            400: {
+                "type": "object",
+                "properties": {"description": "Bad Request"},
+            },
+        },
+    }
+    actual = dereference_refs(schema)
+    assert actual == expected