Agent Lore

import requests, json

def ask_llm(messages, tools=None):
    payload = {"model": "local-model", "messages": messages}
    if tools:
        payload["tools"] = tools
    r = requests.post("http://localhost:1234/v1/chat/completions", json=payload)
    return r.json()["choices"][0]["message"]

def read_file(path):
    try:
        with open(path) as f:
            return f.read()
    except FileNotFoundError:
        return "File not found"

tools = [{
    "type": "function",
    "function": {
        "name": "read_file",
        "parameters": {
            "type": "object",
            "properties": {"path": {"type": "string"}},
            "required": ["path"]
        }
    }
}]

messages = [{"role": "user", "content": "What does main.py do?"}]
msg = ask_llm(messages, tools)

if msg.get("tool_calls"):
    args = json.loads(msg["tool_calls"][0]["function"]["arguments"])
    result = read_file(args["path"])
    messages.append(msg)
    messages.append({"role": "tool", "content": result, "tool_call_id": msg["tool_calls"][0]["id"]})
    final = ask_llm(messages)
    print(final["content"])