feat: Add WebSearchTool to the chat service tools
This commit is contained in:
parent
64cdac963b
commit
c314dd6199
1 changed files with 2 additions and 0 deletions
|
|
@ -15,6 +15,7 @@ from llmai.shared import ( # type: ignore[import-not-found]
|
|||
TextContentPart,
|
||||
ToolResponseMessage,
|
||||
UserMessage,
|
||||
WebSearchTool
|
||||
)
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
|
|
@ -81,6 +82,7 @@ class PresentationChatService:
|
|||
client = get_client(config=get_llm_config())
|
||||
model = get_model()
|
||||
tools = self._tools.get_tool_definitions()
|
||||
tools.append(WebSearchTool())
|
||||
|
||||
called_tools: list[str] = []
|
||||
last_tool_results: list[dict[str, Any]] = []
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue