Module:04-Subgraphs: Pass state_schema in StateGraph of subgraphs instead of input, to remove deprecation warning.

[rehan-ul-haq]

Problem

Currently, input and output schema is passed in subgraphs which raises LangGraphDeprecationWarning as mentioned below.
"LangGraphDeprecationWarning: Initializing StateGraph without state_schema is deprecated. Please pass in an explicit state_schema instead of just an input and output schema.
fa_builder = StateGraph(input=FailureAnalysisState,output=FailureAnalysisOutputState)"

Solution

The same schema is passed as state_schema rather than input and it removed the deprecation warning. Moreover, passing FailureAnalysisState and QuestionSummarizationState as state_schema (overall state) rather than input schema, seems more intuitive.

Testing

• Ran the notebook and received the response as expected