change import

posthog/hogql/ai.py

@@ -1,6 +1,5 @@
 from typing import TYPE_CHECKING, Optional
 import openai
-from openai.types.chat import ChatCompletionMessageParam
 from posthog.event_usage import report_user_action
 from posthog.hogql.context import HogQLContext
 from posthog.hogql.errors import HogQLException
@@ -69,7 +68,7 @@ def write_sql_from_prompt(prompt: str, *, current_query: Optional[str] = None, t
         )
     )
     instance_region = get_instance_region() or "HOBBY"
-    messages: list[ChatCompletionMessageParam] = [
+    messages: list[openai.types.chat.ChatCompletionMessageParam] = [
         {"role": "system", "content": IDENTITY_MESSAGE},
         {
             "role": "system",