from openai import OpenAI
 
client = OpenAI(
    base_url="https://api.unieai.com/",
    api_key="<YOUR_API_KEY>"
)
 
completion = client.chat.completions.create(
    model="UnieAI.u1-2503",
    messages=[
        {
            "role": "user",
            "content": "The efficiency of the following code is very low, and it doesn’t handle edge cases. Please first explain the problems with this code and the solutions, then optimize it:\n\n
python\ndef fib(n):\n    if n <= 2:\n        return n\n    return fib(n-1) + fib(n-2)\n
"
        }
    ]
)
 
print(completion.choices[0].message.content)