Update headings_extractor.py

headings_extractor.py

@@ -38,7 +38,11 @@ class HeadingsExtractor:
         """
         try:
             #get response from openai api
-            
+            client = AzureOpenAI(api_key=os.getenv("AZURE_OPENAI_KEY"),  
+                                api_version="2023-07-01-preview",
+                                azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
+                                 )
+                        
             conversation = [
                             {"role": "system", "content": "You are a helpful assistant."},
                             {"role": "user", "content": f"""Extract Headings from given paragraph do not generate jsu extract the headings from paragraph.
@@ -46,18 +50,14 @@ class HeadingsExtractor:
                         ]
             
             # Call OpenAI GPT-3.5-turbo
-            chat_completion =openai.ChatCompletion.create(
-                                  engine="ChatGPT",
-                                  messages = conversation,
-                                  temperature=0.7,
-                                  max_tokens=800,
-                                  top_p=0.95,
-                                  frequency_penalty=0,
-                                  presence_penalty=0,
-                                  stop=None
-                                )
+            chat_completion = client.chat.completions.create(
+                model = "ChatGPT",
+                messages = conversation,
+                max_tokens=1000,
+                temperature=0
+            )
             response = chat_completion.choices[0].message.content
-            return response 
+            return response
 
         except Exception as e:
             # If an error occurs during the key-value extraction process, log the error