Update extract_date.py

extract_date.py

@@ -1,6 +1,5 @@
 from PyPDF2 import PdfReader
-from openai import OpenAI
-# import fitz  # PyMuPDF
+import openai
 import logging
 
 # Configure logging
@@ -33,7 +32,10 @@ class ExtractDateAndDuration:
         """
         Initialize the ExtractDateAndDuration class.
         """
-        self.client=OpenAI()
+        openai.api_type = os.getenv['api_type']
+        openai.api_base = os.getenv['api_base']
+        openai.api_version = os.getenv['api_version']
+        openai.api_key = os.getenv['api_key']
         
 
     def get_date_and_duration(self, contract_text: str) -> str:
@@ -58,12 +60,16 @@ class ExtractDateAndDuration:
                         ]
 
             # Call OpenAI GPT-3.5-turbo
-            chat_completion = client.chat.completions.create(
-                model = "gpt-3.5-turbo",
-                messages = conversation,
-                max_tokens=1000,
-                temperature=0
-            )
+            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
+                                )
             response = chat_completion.choices[0].message.content
             return response