Merge pull request #61 from chatchat-space/dev

Dev

Author: yuehua-s

libs/chatchat-server/chatchat/server/agent/graphs_factory/base_rag.py

@@ -172,6 +172,10 @@ class Grade(BaseModel):
         scored_result = await referee.ainvoke(state)
         score = scored_result.binary_score
 
+        if score is None or score == "":
+            logger.warning(f"The score is not provided. Defaulting to 'yes'. Question: {state['question']}")
+            score = "yes"
+
         if score == "yes":
             return "generate"
         else:

libs/chatchat-server/chatchat/webui_pages/utils.py

@@ -822,7 +822,9 @@ async def process_graph(graph_class: Graph, graph: CompiledStateGraph, graph_inp
             response_last = ""
             async for event in events:
                 node, response = extract_node_and_response(event)
+
                 # debug
+                # import rich
                 # print(f"--- node: {node} ---")
                 # rich.print(response)
 
@@ -833,7 +835,11 @@ async def process_graph(graph_class: Graph, graph: CompiledStateGraph, graph_inp
                 response = graph_class.handle_event(node=node, event=response)
                 # 将 event 转化为 json
                 response = serialize_content_to_json(response)
-                # rich.print(response)
+
+                # debug
+                import rich
+                print(f"--- node: {node} ---")
+                rich.print(response)
 
                 # 检查 'content' 是否在响应中(因为我们只需要 AIMessage 的内容)
                 if "content" in response: