RUN YOUR AGENT AS THE MAIN MODEL IN HERMES

OUMI_API_KEY=your-oumi-api-key

name: stock-tools
version: "1.0"
description: The tools of the stock lookup agent

import json
import urllib.error
import urllib.parse
import urllib.request

API = "https://api.example-stocks.com"

SEARCH_COMPANY = {
    "name": "search_company",
    "description": "Find a company's stock ticker symbol by company name. Call this before get_stock_price.",
    "parameters": {
        "type": "object",
        "properties": {"name": {"type": "string", "description": "Company name, e.g. 'Apple'."}},
        "required": ["name"],
    },
}

GET_STOCK_PRICE = {
    "name": "get_stock_price",
    "description": "Get the latest stock price for a ticker symbol returned by search_company.",
    "parameters": {
        "type": "object",
        "properties": {"ticker": {"type": "string", "description": "Stock ticker symbol, e.g. 'AAPL'."}},
        "required": ["ticker"],
    },
}


def search_company(args: dict, **kwargs) -> str:
    found = get_json(f"{API}/search?q={urllib.parse.quote(args['name'])}")
    if found is None:
        return json.dumps({"found": False, "message": f"No company found matching \"{args['name']}\"."})
    return json.dumps({"found": True, "ticker": found["ticker"], "name": found["name"]})


def get_stock_price(args: dict, **kwargs) -> str:
    quote = get_json(f"{API}/quote/{urllib.parse.quote(args['ticker'])}")
    if quote is None:
        return json.dumps({"found": False, "message": f"No price found for \"{args['ticker']}\"."})
    return json.dumps({"found": True, "price": quote["price"], "currency": quote["currency"]})


# A 404 means "no match" and returns None. Any other failure raises, and
# Hermes sends the error to the model as the tool result.
def get_json(url: str) -> dict | None:
    try:
        with urllib.request.urlopen(url, timeout=30) as res:
            return json.load(res)
    except urllib.error.HTTPError as error:
        if error.code == 404:
            return None
        raise


def register(ctx):
    ctx.register_tool(name="search_company", toolset="stock", schema=SEARCH_COMPANY, handler=search_company)
    ctx.register_tool(name="get_stock_price", toolset="stock", schema=GET_STOCK_PRICE, handler=get_stock_price)

providers:
  oumi:
    api: https://inference.oumi.ai/v1
    key_env: OUMI_API_KEY
    transport: chat_completions
    discover_models: false
    models:
      projects/{project_id}/deployments/{deployment_id}:
        context_length: 32768

model:
  provider: custom:oumi
  default: projects/{project_id}/deployments/{deployment_id}