refine answer node

2025-01-02 19:47:10 +00:00 · 2025-01-02 19:47:10 +00:00 · 1efa6ca3f2
parent aaadf064e7
commit 1efa6ca3f2
1 changed files with 4 additions and 0 deletions
--- 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()