class Assistant:
    def __init__(self, client_id, llm=call_llm):
        self.client_id = client_id          # vient de la session, jamais du modèle
        self.llm = llm
        self.messages = [{"role": "system", "content": SYSTEM}]
        self.tools_py = {
            "chercher_produit": chercher_produit,
            "consulter_stock": consulter_stock,
            "creer_commande": lambda **a: creer_commande(client_id, **a),
            "transferer_a_un_humain": lambda **a: transferer_a_un_humain(client_id, **a),
        }

    def run_tool(self, call):
        name = call["function"]["name"]
        if name not in self.tools_py:
            result = {"erreur": f"Outil inconnu : {name}"}
        else:
            try:
                args = json.loads(call["function"]["arguments"] or "{}")
                result = self.tools_py[name](**args)
            except Exception as e:
                result = {"erreur": f"{type(e).__name__}: {e}"}
        return {"role": "tool", "tool_call_id": call["id"],
                "content": json.dumps(result, ensure_ascii=False)}

    def demander(self, question, max_steps=8):
        self.messages.append({"role": "user", "content": question})
        for _ in range(max_steps):
            msg = self.llm(self.messages, TOOLS)
            self.messages.append(msg)
            calls = msg.get("tool_calls") or []
            if not calls:
                return msg.get("content") or ""
            for call in calls:
                self.messages.append(self.run_tool(call))
        transferer_a_un_humain(self.client_id, "max_steps atteint")
        return ESCALADE


if __name__ == "__main__":
    bot = Assistant(client_id="demo")
    while (q := input("vous> ").strip()) not in ("", "quit"):
        print("agent>", bot.demander(q))