Create openai_requests.py

openai_requests.py

@@ -0,0 +1,26 @@
+import os
+
+def get_completion_from_openai(prompt, max_tokens):
+    url = os.getenv('OPENAI_COMPLETION_URL')
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": "Bearer " + st.secrets["OPNEAI_TOKEN"],
+    }
+
+    response = requests.post(baseUrl,
+                                 json={    
+                                     "model": os.getenv('OPENAI_MODEL'),
+                                     "max_tokens": max_tokens,
+                                     "messages": [
+                                      {
+                                        "role": "user",
+                                        "content": prompt
+                                      }
+                                    ]
+                                }, 
+                                 headers=headers, 
+                                 stream=False,
+                                 )
+    
+    return run_response.json()['choices'][0]['message']['content']