better parser for step call

[TonioSnowden]

Works best for :

from notte.env import NotteEnv
import litellm

goal = "Find best flights from Boston to Tokyo on google flights"
messages = [{"role": "system", "content": goal}]
max_steps = 10

def llm_agent(messages):
    response = litellm.completion(
        model="cerebras/llama3.1-70b",
        messages=messages,
    )
    return response.choices[0].message.content

async def main(max_steps=10):
    async with NotteEnv(headless=False) as env:
        while '<done>' not in messages[-1]['content'] and max_steps > 0:
            resp = llm_agent(messages) # query your llm policy.
            obs = await env.chat(resp) # query notte with llm response.
            messages.append({"role": "assistant", "content": resp})
            messages.append({"role": "user", "content": obs})
            max_steps -= 1

await main()

[andreakiro]

+Precommits are failing, install and fix them before we merge

pre-commit install