From 1efa6ca3f233b5b84a1450ee2133cbe1806d55b9 Mon Sep 17 00:00:00 2001
From: zachary62 <zhuang333@wisc.edu>
Date: Thu, 2 Jan 2025 19:47:10 +0000
Subject: [PATCH] refine answer node

---
 docs/agent.md | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docs/agent.md b/docs/agent.md
index 6694862..d4e2aff 100644
--- a/docs/agent.md
+++ b/docs/agent.md
@@ -77,6 +77,10 @@ class DirectAnswer(Node):
         query, context = inputs
         return call_llm(f"Context: {context}\nAnswer: {query}")
 
+    def post(self, shared, prep_res, exec_res):
+       print(f"Answer: {exec_res}")
+       shared["answer"] = exec_res
+
 # Connect nodes
 decide = DecideAction()
 search = SearchWeb()