chatui node scaffolding

app/main.py

@@ -9,7 +9,20 @@ class GraphState(TypedDict):
     context: str
     result: str
 
-# node 1: retriever
+
+### OPEN QUESTION: SHOULD WE PASS ALL PARAMS FROM THE ORCHESTRATOR TO THE NODES INSTEAD OF SETTING IN EACH MODULE?
+
+# node 1: chatUI
+# To be finalized
+# def chatui_node(state: GraphState) -> GraphState:
+#     client = Client("giz/chatfed_chatui") 
+#     context = client.predict(
+#         query=state["query"], # not sure if we need to pass the query here
+#         api_name="/chat"
+#     )
+#     return {"query": query}
+
+# node 2: retriever
 def retrieve_node(state: GraphState) -> GraphState:
     client = Client("giz/chatfed_retriever")  # HF repo name
     context = client.predict(
@@ -22,7 +35,7 @@ def retrieve_node(state: GraphState) -> GraphState:
     )
     return {"context": context}
 
-# node 2: generator
+# node 3: generator
 def generate_node(state: GraphState) -> GraphState:
     client = Client("giz/chatfed_generator")
     result = client.predict(
@@ -36,10 +49,13 @@ def generate_node(state: GraphState) -> GraphState:
 workflow = StateGraph(GraphState)
 
 # Add nodes
+# workflow.add_node("chatui", chatui_node)
 workflow.add_node("retrieve", retrieve_node)
 workflow.add_node("generate", generate_node)
 
 # Add edges
+# workflow.add_edge(START, "chatui")
+# workflow.add_edge("chatui", "retrieve")
 workflow.add_edge(START, "retrieve")
 workflow.add_edge("retrieve", "generate")
 workflow.add_edge("generate", END)