mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-05-05 09:20:26 +02:00
8a4d762798
* Fix follow ups in thread + fix user name * Add back single history str * Remove newline
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
from typing import cast
|
|
from typing import TYPE_CHECKING
|
|
|
|
from langchain_core.messages import HumanMessage
|
|
|
|
from onyx.llm.utils import message_to_prompt_and_imgs
|
|
from onyx.tools.tool import Tool
|
|
|
|
if TYPE_CHECKING:
|
|
from onyx.chat.prompt_builder.answer_prompt_builder import AnswerPromptBuilder
|
|
from onyx.tools.tool_implementations.custom.custom_tool import (
|
|
CustomToolCallSummary,
|
|
)
|
|
from onyx.tools.message import ToolCallSummary
|
|
from onyx.tools.models import ToolResponse
|
|
|
|
|
|
def build_user_message_for_non_tool_calling_llm(
|
|
message: HumanMessage,
|
|
tool_name: str,
|
|
*args: "ToolResponse",
|
|
) -> str:
|
|
query, _ = message_to_prompt_and_imgs(message)
|
|
|
|
tool_run_summary = cast("CustomToolCallSummary", args[0].response).tool_result
|
|
return f"""
|
|
Here's the result from the {tool_name} tool:
|
|
|
|
{tool_run_summary}
|
|
|
|
Now respond to the following:
|
|
|
|
{query}
|
|
""".strip()
|
|
|
|
|
|
class BaseTool(Tool):
|
|
def build_next_prompt(
|
|
self,
|
|
prompt_builder: "AnswerPromptBuilder",
|
|
tool_call_summary: "ToolCallSummary",
|
|
tool_responses: list["ToolResponse"],
|
|
using_tool_calling_llm: bool,
|
|
) -> "AnswerPromptBuilder":
|
|
if using_tool_calling_llm:
|
|
prompt_builder.append_message(tool_call_summary.tool_call_request)
|
|
prompt_builder.append_message(tool_call_summary.tool_call_result)
|
|
else:
|
|
prompt_builder.update_user_prompt(
|
|
HumanMessage(
|
|
content=build_user_message_for_non_tool_calling_llm(
|
|
prompt_builder.user_message_and_token_cnt[0],
|
|
self.name,
|
|
*tool_responses,
|
|
)
|
|
)
|
|
)
|
|
|
|
return prompt_builder
|