AI Agents and Function Calling

tool = {
    "type": "function",
    "function": {
        "name": "query_economic_data",
        "description": "Get macro-economic data (GDP, CPI, unemployment, etc). "
                       "Use as_of to get data as it was known on a specific date.",
        "parameters": {
            "type": "object",
            "properties": {
                "series_id": {
                    "type": "string",
                    "description": "GDPC1=GDP, CPIAUCSL=CPI, UNRATE=unemployment, "
                                   "PAYEMS=payrolls, FEDFUNDS=fed funds rate"
                },
                "as_of": {
                    "type": "string",
                    "description": "Date (YYYY-MM-DD). Returns data as known on this date. "
                                   "Omit for latest values."
                },
                "limit": {
                    "type": "integer",
                    "description": "Number of observations (default 5, max 1000)"
                }
            },
            "required": ["series_id"]
        }
    }
}

import os
import requests

def query_economic_data(series_id: str, as_of: str = None, limit: int = 5) -> dict:
    params = {"limit": limit}
    if as_of:
        params["as_of"] = as_of

    resp = requests.get(
        f"https://api.vintl.io/v1/series/{series_id}/observations",
        headers={"X-API-Key": os.environ["VINTL_API_KEY"]},
        params=params
    )
    return resp.json()

def query_yield_curve(date: str = None) -> dict:
    params = {}
    if date:
        params["date"] = date

    resp = requests.get(
        "https://api.vintl.io/v1/treasury/yields/curve",
        headers={"X-API-Key": os.environ["VINTL_API_KEY"]},
        params=params
    )
    return resp.json()