Agente de web search construido e implementado(Tavily ejemplo de .env en readme)
This commit is contained in:
29
backend/app/agents/web_search/models.py
Normal file
29
backend/app/agents/web_search/models.py
Normal file
@@ -0,0 +1,29 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
|
||||
class SearchResult(BaseModel):
|
||||
"""Individual search result from web search"""
|
||||
|
||||
title: str
|
||||
url: str
|
||||
content: str
|
||||
score: float | None = None
|
||||
|
||||
|
||||
class WebSearchState(BaseModel):
|
||||
"""State passed to agent tools via deps"""
|
||||
|
||||
user_query: str
|
||||
max_results: int = Field(default=5, ge=1, le=10)
|
||||
include_raw_content: bool = False
|
||||
|
||||
|
||||
class WebSearchResponse(BaseModel):
|
||||
"""Structured response from the web search agent"""
|
||||
|
||||
query: str
|
||||
results: list[SearchResult]
|
||||
summary: str
|
||||
total_results: int
|
||||
Reference in New Issue
Block a user