कोड में एजेंट लूप

def get_weather(city: str) -> str:
    fake = {"Chennai": "34 C and humid", "Shimla": "12 C and clear"}
    return fake.get(city, "error: unknown city " + city)

TOOLS = {"get_weather": get_weather}

def call_model(messages: list) -> dict:
    # Placeholder model. Plug in your provider here and convert its reply to
    # {"tool": name, "args": {...}} or {"answer": text}.
    last = messages[-1]
    if last["role"] == "user":
        return {"tool": "get_weather", "args": {"city": "Chennai"}}
    return {"answer": "It is " + last["content"] + " in Chennai. Carry water."}

def run_agent(task: str, max_steps: int = 5) -> str:
    messages = [{"role": "user", "content": task}]
    for step in range(max_steps):
        reply = call_model(messages)                # think
        if "answer" in reply:                       # stop: model is done
            return reply["answer"]
        name, args = reply.get("tool"), reply.get("args", {})
        if name not in TOOLS:
            result = "error: there is no tool called " + str(name)
        else:
            try:
                result = TOOLS[name](**args)        # act
            except Exception as err:
                result = "error: " + str(err)
        messages.append({"role": "tool", "content": result})   # observe
    return "Stopped: reached the step limit without an answer."

print(run_agent("Do I need to carry water in Chennai today?"))