major Agent Search Updates (#3994)

This commit is contained in:
joachim-danswer
2025-02-14 11:40:21 -08:00
committed by GitHub
parent ec78f78f3c
commit 6687d5d499
36 changed files with 2115 additions and 431 deletions

View File

@@ -81,6 +81,7 @@ class CustomModelServer(LLM):
tools: list[dict] | None = None,
tool_choice: ToolChoiceOptions | None = None,
structured_response_format: dict | None = None,
timeout_override: int | None = None,
) -> BaseMessage:
return self._execute(prompt)
@@ -90,5 +91,6 @@ class CustomModelServer(LLM):
tools: list[dict] | None = None,
tool_choice: ToolChoiceOptions | None = None,
structured_response_format: dict | None = None,
timeout_override: int | None = None,
) -> Iterator[BaseMessage]:
yield self._execute(prompt)