Update extract_date.py

extract_date.py

@@ -50,6 +50,11 @@ class ExtractDateAndDuration:
             str: Extracted dates and durations.
         """
         try:
+            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"""Your task is Identify Dates and Durations Mentioned in the contract and extract that date and duration in key-value pair.
@@ -61,16 +66,12 @@ class ExtractDateAndDuration:
                         ]
 
             # 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