openai[patch] get output_type when using with_structured_output

libs/partners/openai/langchain_openai/chat_models/base.py

@@ -1417,7 +1417,10 @@ class AnswerWithJustification(BaseModel):
             response_format = _convert_to_openai_response_format(schema, strict=strict)
             llm = self.bind(response_format=response_format)
             output_parser = (
-                cast(Runnable, _oai_structured_outputs_parser)
+                cast(
+                    Runnable,
+                    _oai_structured_outputs_parser.with_types(output_type=schema),
+                )
                 if is_pydantic_schema
                 else JsonOutputParser()
             )

libs/partners/openai/tests/unit_tests/chat_models/test_base.py

@@ -694,3 +694,20 @@ def test_get_num_tokens_from_messages() -> None:
     expected = 176
     actual = llm.get_num_tokens_from_messages(messages)
     assert expected == actual
+
+
+def test_schema_from_with_structured_output() -> None:
+    """Test schema from with_structured_output."""
+
+    class Foo(BaseModel):
+        bar: Any
+
+    model_chat = ChatOpenAI(api_key="foo")
+
+    model = model_chat.with_structured_output(Foo, method="json_schema", strict=True)
+
+    assert model.get_output_schema().schema() == {
+        "properties": {"bar": {"title": "Bar"}},
+        "title": "Foo",
+        "type": "object",
+    }