How can I use openai package in Flyte? I get some errors caused by the runtime env not have openai.

Build a custom image with the packages you want in, and use that

also check out the new imagespec stuff (docs to be merged soon) to help with image building