| from typing import List, Optional
|
| from tavily import TavilyClient
|
| from langchain_core.tools import tool
|
| import time
|
|
|
|
|
| client =TavilyClient('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')
|
|
|
|
|
| def search(query: str, topic: Optional[str] = None, max_results: int = 5) -> List[str]:
|
| """
|
| Perform a Tavily search and return only the extracted content from results.
|
|
|
| Args:
|
| query (str): The search query string.
|
| topic (Optional[str]): (Optional) Topic/domain to refine the search.
|
| max_results (int): Maximum number of search results to retrieve.
|
|
|
| Returns:
|
| List[str]: A list of content snippets extracted from the search results.
|
| """
|
| try:
|
| response = client.search(
|
| query=query,
|
| topic=topic,
|
| max_results=max_results,
|
| search_depth="advanced",
|
| )
|
|
|
|
|
| contents = [
|
| result.get("content", "").strip()
|
| for result in response.get("results", [])
|
| if result.get("content")
|
| ]
|
|
|
| return contents if contents else ["No content found."]
|
|
|
| except Exception as e:
|
| return [f"Search failed: {str(e)}"]
|
|
|
| def get_time():
|
| return time.time()
|
|
|
| print(get_time())
|
|
|
|
|
|
|
|
|
