Update key_value_extractor.py

key_value_extractor.py

@@ -1,4 +1,4 @@
-import openai
+from openai import OpenAI
 import os
 
 
@@ -21,17 +21,21 @@ class KeyValue:
         try:
 
 
-            # Use OpenAI's Completion API to analyze the text and extract key-value pairs
-            response = openai.Completion.create(
-                engine="text-davinci-003",
-                temperature=0,
-                prompt=f"analyze the given contract and get meaningful key value pairs in given content.contract in backticks.```{extracted_summary}```.",
-                max_tokens=1000
+            conversation = [
+                            {"role": "system", "content": "You are a helpful Keywords Extracter."},
+                            {"role": "user", "content": f"""analyze the given contract and Extract Keywords for following contract in triple backticks. tags should be bullet points.contract :```{extracted_summary}```."""}
+                          ]
+                        
+            # Call OpenAI GPT-3.5-turbo
+            chat_completion = self.client.chat.completions.create(
+                model = "gpt-3.5-turbo",
+                messages = conversation,
+                max_tokens=500,
+                temperature=0
             )
-
-            # Extract and return the chatbot's reply
-            result = response['choices'][0]['text'].strip()
-            return result
+            response = chat_completion.choices[0].message.content
+            return response
+            
         except Exception as e:
             # If an error occurs during the key-value extraction process, log the error
             print(f"Error occurred while extracting key-value pairs: {str(e)}")