highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-04 01:48:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

address JR comments

Browse Source
This commit is contained in:
Evan Lohn 2025-01-24 13:09:08 -08:00
parent db2004542e
commit 2bbe20edc3
6 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/onyx/agents/agent_search/basic/graph_builder.py
View File

@ -5,12 +5,14 @@ from langgraph.graph import StateGraph
from onyx.agents.agent_search.basic.states import BasicInput
from onyx.agents.agent_search.basic.states import BasicOutput
from onyx.agents.agent_search.basic.states import BasicState
from onyx.agents.agent_search.orchestration.basic_use_tool_response import (
from onyx.agents.agent_search.orchestration.nodes.basic_use_tool_response import (
basic_use_tool_response,
)
from onyx.agents.agent_search.orchestration.llm_tool_choice import llm_tool_choice
from onyx.agents.agent_search.orchestration.prepare_tool_input import prepare_tool_input
from onyx.agents.agent_search.orchestration.tool_call import tool_call
from onyx.agents.agent_search.orchestration.nodes.llm_tool_choice import llm_tool_choice
from onyx.agents.agent_search.orchestration.nodes.prepare_tool_input import (
prepare_tool_input,
)
from onyx.agents.agent_search.orchestration.nodes.tool_call import tool_call
from onyx.utils.logger import setup_logger
logger = setup_logger()

10
backend/onyx/agents/agent_search/deep_search_a/main/graph_builder.py
View File

@ -67,12 +67,14 @@ from onyx.agents.agent_search.deep_search_a.main.nodes.retrieval_consolidation i
)
from onyx.agents.agent_search.deep_search_a.main.states import MainInput
from onyx.agents.agent_search.deep_search_a.main.states import MainState
from onyx.agents.agent_search.orchestration.basic_use_tool_response import (
from onyx.agents.agent_search.orchestration.nodes.basic_use_tool_response import (
basic_use_tool_response,
)
from onyx.agents.agent_search.orchestration.llm_tool_choice import llm_tool_choice
from onyx.agents.agent_search.orchestration.prepare_tool_input import prepare_tool_input
from onyx.agents.agent_search.orchestration.tool_call import tool_call
from onyx.agents.agent_search.orchestration.nodes.llm_tool_choice import llm_tool_choice
from onyx.agents.agent_search.orchestration.nodes.prepare_tool_input import (
prepare_tool_input,
)
from onyx.agents.agent_search.orchestration.nodes.tool_call import tool_call
from onyx.agents.agent_search.shared_graph_utils.utils import get_test_config
from onyx.utils.logger import setup_logger

0
backend/onyx/agents/agent_search/orchestration/basic_use_tool_response.py → backend/onyx/agents/agent_search/orchestration/nodes/basic_use_tool_response.py
View File

0
backend/onyx/agents/agent_search/orchestration/llm_tool_choice.py → backend/onyx/agents/agent_search/orchestration/nodes/llm_tool_choice.py
View File

1
backend/onyx/agents/agent_search/orchestration/prepare_tool_input.py → backend/onyx/agents/agent_search/orchestration/nodes/prepare_tool_input.py
View File

@ -10,6 +10,7 @@ from onyx.agents.agent_search.orchestration.states import ToolChoiceInput
def prepare_tool_input(state: Any, config: RunnableConfig) -> ToolChoiceInput:
agent_config = cast(AgentSearchConfig, config["metadata"]["config"])
return ToolChoiceInput(
# NOTE: this node is used at the top level of the agent, so we always stream
should_stream_answer=True,
prompt_snapshot=None, # uses default prompt builder
tools=[tool.name for tool in (agent_config.tools or [])],

0
backend/onyx/agents/agent_search/orchestration/tool_call.py → backend/onyx/agents/agent_search/orchestration/nodes/tool_call.py
View File