Weird result with openAI config

[pipilurj]

Dear authors,
Thank you so much for this in interesting work. I am trying to run vision agent locally with openai config the the following example:
from vision_agent.agent import VisionAgentCoderV2
from vision_agent.models import AgentMessage

agent = VisionAgentCoderV2(verbose=True)
code_context = agent.generate_code(
[
AgentMessage(
role="user",
content="Count the number of apples in this image",
media=["apples.png"]
)
]
)

with open("generated_code.py", "w") as f:
f.write(code_context.code + "\n" + code_context.test)

But I got the following outputs:

The resulting code also contains bugs.

I wonder if this is because I am using the OpenAI config rather than Anthropic config?
Thanks!

[pipilurj]

The above is the bug I often encounter