@@ -384,8 +384,7 @@ def main():
384384 "intent" : "normal_search" ,
385385 "slot" : []
386386 }
387- # 通过标志位控制后续的逻辑
388- # 主要的意图有4个, 拒绝, 查询, 思维链, 知识问答
387+
389388 if intent_ner_recognition_flag :
390389 with st .status ("Performing intent recognition..." ) as status_text :
391390 intent_response = get_query_intent (model_type , search_box , prompt_map )
@@ -413,12 +412,10 @@ def main():
413412 else :
414413 search_intent_flag = True
415414
416- # 主要的逻辑部分,调用LLM
417415 if reject_intent_flag :
418416 st .write ("Your query statement is currently not supported by the system" )
419417
420418 elif search_intent_flag :
421- # 执行普通的查询,并可视化结果
422419 normal_search_result = normal_text_search_streamlit (search_box , model_type ,
423420 database_profile ,
424421 entity_slot , opensearch_info ,
@@ -456,11 +453,9 @@ def main():
456453 else :
457454 st .error ("Intent recognition error" )
458455
459- # 前端结果显示agent cot任务拆分信息, normal_text_search 的显示,做了拆分,为了方便跟API逻辑一致
460456 if search_intent_flag :
461457 if normal_search_result .sql != "" :
462458 current_nlq_chain .set_generated_sql (normal_search_result .sql )
463- # st.code(normal_search_result.sql, language="sql")
464459
465460 current_nlq_chain .set_generated_sql_response (normal_search_result .response )
466461
@@ -478,7 +473,6 @@ def main():
478473 with st .expander (f'Agent Task Result: { len (agent_search_result )} ' ):
479474 st .write (agent_search_result )
480475
481- # 连接数据库,执行SQL, 记录历史记录并展示
482476 if search_intent_flag :
483477 with st .spinner ('Executing query...' ):
484478 search_intent_result = get_sql_result_tool (
@@ -544,18 +538,16 @@ def main():
544538 # do something here
545539 pass
546540
547- # 数据可视化展示
548541 if visualize_results_flag and search_intent_flag :
549542 current_search_sql_result = st .session_state .current_sql_result [selected_profile ]
550543 if current_search_sql_result is not None and len (current_search_sql_result ) > 0 :
551544 st .session_state .messages [selected_profile ].append (
552545 {"role" : "assistant" , "content" : current_search_sql_result , "type" : "pandas" })
553- # Reset change flag to False
546+
554547 do_visualize_results (current_nlq_chain , st .session_state .current_sql_result [selected_profile ])
555548 else :
556549 st .markdown ("No relevant data found" )
557550
558- # 生成推荐问题
559551 if gen_suggested_question_flag and (search_intent_flag or agent_intent_flag ):
560552 st .markdown ('You might want to further ask:' )
561553 with st .spinner ('Generating suggested questions...' ):
@@ -576,7 +568,7 @@ def main():
576568 on_click = sample_question_clicked ,
577569 args = [gen_sq_list [2 ]])
578570 else :
579- # st.error("Please enter a valid query.")
571+
580572 if current_nlq_chain .is_visualization_config_changed ():
581573 if visualize_results_flag :
582574 do_visualize_results (current_nlq_chain , st .session_state .current_sql_result [selected_profile ])
0 commit comments