Merge pull request #113 from UiPath/fix/required-state-schema

fix: specify state_schema for every StateGraph

Author: radu-mocanu

samples/RAG-quiz-generator/pyproject.toml

@@ -9,6 +9,6 @@ dependencies = [
     "langchain-anthropic>=0.3.8",
     "langchain-experimental>=0.3.4",
     "tavily-python>=0.5.0",
-    "uipath-langchain==0.0.105",
+    "uipath-langchain==0.0.113",
 ]
 requires-python = ">=3.10"

samples/RAG-quiz-generator/src/agents/quiz-generator-RAG-agent.py

@@ -167,7 +167,7 @@ def return_quiz(state: GraphState) -> GraphOutput:
     return GraphOutput(quiz=state["quiz"])
 
 # Build the state graph
-builder = StateGraph(input=GraphInput, output=GraphOutput)
+builder = StateGraph(GraphState, input=GraphInput, output=GraphOutput)
 builder.add_node("invoke_researcher", invoke_researcher)
 builder.add_node("create_quiz", create_quiz)
 builder.add_node("return_quiz", return_quiz)
@@ -178,7 +178,6 @@ def return_quiz(state: GraphState) -> GraphOutput:
 builder.add_conditional_edges("create_quiz", check_quiz_creation)
 builder.add_edge("invoke_researcher", "create_quiz")
 builder.add_edge("return_quiz", END)
-import os
 
 # Compile the graph
 graph = builder.compile()

samples/RAG-quiz-generator/src/agents/researcher-RAG-agent.py

@@ -90,7 +90,7 @@ async def add_data_to_context_grounding_index(state: GraphState) -> MessagesStat
 
 
 # Build the state graph
-builder = StateGraph(input=GraphInput, output=MessagesState)
+builder = StateGraph(GraphState ,input=GraphInput, output=MessagesState)
 builder.add_node("research", research_node)
 builder.add_node("add_data_to_context_grounding_index", add_data_to_context_grounding_index)
 builder.add_node("prepare_input", prepare_input)

samples/RAG-quiz-generator/uv.lock

@@ -35,15 +35,15 @@
 research_agent = create_react_agent(llm, tools=[tavily_tool], prompt=system_prompt)
 
 
-class GraphInput(BaseModel):
+class GraphState(BaseModel):
     company_name: str
 
 
 class GraphOutput(BaseModel):
     response: str
 
 
-async def research_node(state: GraphInput) -> GraphOutput:
+async def research_node(state: GraphState) -> GraphOutput:
     # Format the user message with the company name
     user_message = f"""Please provide a comprehensive analysis and outreach strategy for the company: {state.company_name}. Use the TavilySearchResults tool to gather information. Include detailed research on the company's background, organizational structure, key decision-makers, and a tailored outreach strategy. Format your response using the following section headers:
 
@@ -64,7 +64,7 @@ async def research_node(state: GraphInput) -> GraphOutput:
 
 
 # Build the state graph
-builder = StateGraph(input=GraphInput, output=GraphOutput)
+builder = StateGraph(GraphState, output=GraphOutput)
 builder.add_node("researcher", research_node)
 
 builder.add_edge(START, "researcher")

samples/company-research-agent/graph.py

@@ -8,7 +8,7 @@ dependencies = [
     "langgraph>=0.2.55",
     "langchain-anthropic>=0.3.8",
     "tavily-python>=0.5.0",
-    "uipath-langchain==0.0.100"
+    "uipath-langchain==0.0.113"
 ]
 
 [project.optional-dependencies]

samples/company-research-agent/pyproject.toml

@@ -34,4 +34,4 @@
     "version": "2.0",
     "resources": []
   }
-}
+}

samples/company-research-agent/uipath.json

@@ -9,7 +9,7 @@ dependencies = [
     "langchain-anthropic>=0.3.8",
     "langchain-experimental>=0.3.4",
     "tavily-python>=0.5.0",
-    "uipath-langchain==0.0.100"
+    "uipath-langchain==0.0.113"
 ]
 
 [project.optional-dependencies]

samples/company-research-agent/uv.lock

@@ -24,7 +24,7 @@ async def research_node(state: MessagesState) -> GraphOutput:
 
 
 # Build the state graph
-builder = StateGraph(input=MessagesState, output=GraphOutput)
+builder = StateGraph(MessagesState, output=GraphOutput)
 builder.add_node("researcher", research_node)
 
 builder.add_edge(START, "researcher")