Spaces:
Sleeping
Sleeping
| import os | |
| import requests | |
| from langchain_core.tools import tool | |
| from tavily import TavilyClient | |
| def web_search(query: str) -> str: | |
| """ | |
| Search the web for information using Tavily API. | |
| Great for finding competitors, market sizing, and general industry trends. | |
| """ | |
| api_key = os.getenv("TAVILY_API_KEY") | |
| if not api_key: | |
| return "Error: TAVILY_API_KEY not set in environment." | |
| try: | |
| client = TavilyClient(api_key=api_key) | |
| # We perform a basic search and ask for answer and raw content | |
| response = client.search(query=query, search_depth="basic", max_results=3) | |
| # Format the response back to a string so the LLM can read it | |
| results_str = [] | |
| for result in response.get("results", []): | |
| results_str.append(f"Title: {result.get('title')}\nURL: {result.get('url')}\nContent: {result.get('content')}\n") | |
| return "\n---\n".join(results_str) | |
| except Exception as e: | |
| return f"Error during web search: {e}" | |
| def hn_search(query: str) -> str: | |
| """ | |
| Search HackerNews to gauge community sentiment, find past discussions, or see similar product launches. | |
| """ | |
| url = f"http://hn.algolia.com/api/v1/search?query={query}&tags=story" | |
| try: | |
| response = requests.get(url) | |
| response.raise_for_status() | |
| data = response.json() | |
| results_str = [] | |
| for item in data.get("hits", [])[:5]: # Get top 5 discussions | |
| results_str.append(f"Title: {item.get('title')}\nURL: {item.get('url')}\nPoints: {item.get('points')}\nComments: {item.get('num_comments')}\n") | |
| return "\n---\n".join(results_str) if results_str else "No HackerNews discussions found." | |
| except Exception as e: | |
| return f"Error during HackerNews search: {e}" | |
| def github_search(query: str) -> str: | |
| """ | |
| Search GitHub repositories to identify open-source alternatives or technical risks. | |
| """ | |
| # Using the GitHub API to search for repositories | |
| url = f"https://api.github.com/search/repositories?q={query}&sort=stars&order=desc" | |
| headers = {"Accept": "application/vnd.github.v3+json"} | |
| token = os.getenv("GITHUB_TOKEN") | |
| if token: | |
| headers["Authorization"] = f"token {token}" | |
| try: | |
| response = requests.get(url, headers=headers) | |
| response.raise_for_status() | |
| data = response.json() | |
| results_str = [] | |
| for item in data.get("items", [])[:3]: # Get top 3 repos | |
| results_str.append(f"Repo: {item.get('full_name')}\nDescription: {item.get('description')}\nStars: {item.get('stargazers_count')}\nURL: {item.get('html_url')}\n") | |
| return "\n---\n".join(results_str) if results_str else "No open-source alternatives found on GitHub." | |
| except Exception as e: | |
| return f"Error during GitHub search: {e}" | |