Update extract_date.py

extract_date.py

@@ -33,6 +33,7 @@ class ExtractDateAndDuration:
         """
         Initialize the ExtractDateAndDuration class.
         """
+        self.client=OpenAI()
         pass
 
     def get_date_and_duration(self, contract_text: str) -> str:
@@ -46,16 +47,25 @@ class ExtractDateAndDuration:
             str: Extracted dates and durations.
         """
         try:
-            response = openai.Completion.create(
-                engine="text-davinci-003",
-                prompt=f"""Your task is Identify Dates and Durations Mentioned in the contract and extract that date and duration in key-value pair.
-                ```contract: {contract_text}```
-                """,
-                max_tokens=300,
+            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.
+                        ```contract: {contract_text}```
+                        format:
+                          date:extracted date
+                          Durations:extracted Durations           
+                        """}
+                        ]
+
+            # Call OpenAI GPT-3.5-turbo
+            chat_completion = client.chat.completions.create(
+                model = "gpt-3.5-turbo",
+                messages = conversation,
+                max_tokens=1000,
                 temperature=0
             )
-            extracted_date_duration = response.choices[0].text.strip()
-            return extracted_date_duration
+            response = chat_completion.choices[0].message.content
+            return response
 
         except Exception as e:
             LOGGER.error(f"An error occurred during text analysis: {str(e)}")