Create openai_client.py

openai_client.py

@@ -0,0 +1,12 @@
+import os
+import openai
+
+openai.api_key = os.environ["OPENAI_API_KEY"]
+
+def ask_gpt(messages, model="gpt-4"):
+    response = openai.ChatCompletion.create(
+        model=model,
+        messages=messages,
+        temperature=0.3,
+    )
+    return response.choices[0].message['content']