[PromptFlow] Move import to the head of the file to avoid package references.#349