from omnicoreagent.community import TavilySearch, CalculatorTool, HackerNewsGetTopStories

from omnicoreagent import OmniCoreAgent
from omnicoreagent.community import TavilySearch, CalculatorTool

agent = OmniCoreAgent(
    name="assistant",
    system_instruction="You have search and math.",
    model_config={"provider": "openai", "model": "gpt-4o"},
    local_tools=[TavilySearch(), CalculatorTool()],
)

from omnicoreagent import OmniCoreAgent, ToolRegistry, Tool
from omnicoreagent.community import TavilySearch, HackerNewsGetTopStories

# Your custom tool
class GreetTool:
    def get_tool(self) -> Tool:
        return Tool(
            name="greet",
            description="Greet a user.",
            inputSchema={"type": "object", "properties": {"name": {"type": "string"}}, "required": ["name"]},
            function=self._greet,
        )
    async def _greet(self, name: str):
        return {"status": "success", "data": f"Hello, {name}!", "message": "Greeted"}

registry = ToolRegistry()
registry.register(GreetTool())           # Custom tool
registry.register(TavilySearch())        # Community tool
registry.register(HackerNewsGetTopStories())

agent = OmniCoreAgent(
    name="hybrid_agent",
    system_instruction="You can greet users, search the web, and fetch HN stories.",
    model_config={"provider": "openai", "model": "gpt-4o"},
    local_tools=registry,
)

export TAVILY_API_KEY=tvly-...
export GOOGLE_API_KEY=...
export GOOGLE_CSE_ID=...

# Tools auto-read from env vars when no api_key is passed
agent = OmniCoreAgent(
    local_tools=[TavilySearch()],  # Reads TAVILY_API_KEY automatically
    ...
)

{
    "status": "success" | "error",
    "data": Any,       # The actual result
    "message": str     # Human-readable summary or error details
}

from omnicoreagent.community import SpotifySearch

try:
    tool = SpotifySearch(access_token="...")
except ImportError as e:
    print(e)
    # → "Could not import `httpx` python package. Please install it with `pip install httpx`."