start of no sub-answer gen search

Author: joachim-danswer

backend/onyx/agent_search/pro_search_b/answer_initial_sub_question/nodes/empty_answer_check.py

@@ -1,33 +1,31 @@
-from langchain_core.messages import HumanMessage
-from langchain_core.messages import merge_message_runs
-
 from onyx.agent_search.pro_search_b.answer_initial_sub_question.states import (
     AnswerQuestionState,
 )
 from onyx.agent_search.pro_search_b.answer_initial_sub_question.states import (
     QACheckUpdate,
 )
-from onyx.agent_search.shared_graph_utils.prompts import SUB_CHECK_PROMPT
 
 
 def answer_check(state: AnswerQuestionState) -> QACheckUpdate:
-    msg = [
-        HumanMessage(
-            content=SUB_CHECK_PROMPT.format(
-                question=state["question"],
-                base_answer=state["answer"],
-            )
-        )
-    ]
+    # msg = [
+    #     HumanMessage(
+    #         content=SUB_CHECK_PROMPT.format(
+    #             question=state["question"],
+    #             base_answer=state["answer"],
+    #         )
+    #     )
+    # ]
 
-    fast_llm = state["subgraph_fast_llm"]
-    response = list(
-        fast_llm.stream(
-            prompt=msg,
-        )
-    )
+    # fast_llm = state["subgraph_fast_llm"]
+    # response = list(
+    #     fast_llm.stream(
+    #         prompt=msg,
+    #     )
+    # )
+
+    # quality_str = merge_message_runs(response, chunk_separator="")[0].content
 
-    quality_str = merge_message_runs(response, chunk_separator="")[0].content
+    quality_str = "yes"
 
     return QACheckUpdate(
         answer_quality=quality_str,

backend/onyx/agent_search/run_graph.py

@@ -12,8 +12,8 @@
 from onyx.agent_search.basic.graph_builder import basic_graph_builder
 from onyx.agent_search.basic.states import BasicInput
 from onyx.agent_search.models import AgentDocumentCitations
-from onyx.agent_search.pro_search_a.main.graph_builder import main_graph_builder
-from onyx.agent_search.pro_search_a.main.states import MainInput
+from onyx.agent_search.pro_search_b.main.graph_builder import main_graph_builder
+from onyx.agent_search.pro_search_b.main.states import MainInput
 from onyx.agent_search.shared_graph_utils.utils import get_test_config
 from onyx.chat.llm_response_handler import LLMResponseHandlerManager
 from onyx.chat.models import AgentAnswerPiece
@@ -34,6 +34,9 @@
 from onyx.tools.tool_runner import ToolCallKickoff
 from onyx.utils.logger import setup_logger
 
+# from onyx.agent_search.pro_search_a.main.graph_builder import main_graph_builder
+# from onyx.agent_search.pro_search_a.main.states import MainInput
+
 logger = setup_logger()
 
 _COMPILED_GRAPH: CompiledStateGraph | None = None