Skip to content

Commit

Permalink
feat: use the built in list instead of List
Browse files Browse the repository at this point in the history
Signed-off-by: Palaniappan R <[email protected]>
  • Loading branch information
palaniappan-r committed Jul 14, 2024
1 parent 999e757 commit b5d6335
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions backend/src/agents/retriever_graph.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from .retriever_agent import RetrieverAgent

from typing import TypedDict, Annotated, Union, Optional, List
from typing import TypedDict, Annotated, Union, Optional
from langchain_core.messages import AnyMessage

from langgraph.graph import START, StateGraph, END
Expand Down Expand Up @@ -83,7 +83,7 @@ def generate(self, state: AgentState) -> dict:

return {"messages": []}

def route(self, state: AgentState) -> List[str]:
def route(self, state: AgentState) -> list[str]:
tools = state["tools"]
tool_names = []

Expand Down

0 comments on commit b5d6335

Please sign in to comment.