Update aggressive_content_finder.py

aggressive_content_finder.py

@@ -14,8 +14,7 @@ class AggressiveContentFinder:
         """
         Initialize the AggressiveContentFinder with your OpenAI API key.
         """
-        # openai.api_key = openai_api_key
-        pass
+        self.client = OpenAI()
 
     def _extract_aggressive_content(self, contract_text: str) -> str:
         """
@@ -28,17 +27,22 @@ class AggressiveContentFinder:
             str: Identified aggressive terms.
         """
         try:
-            response = openai.Completion.create(
-                engine="text-davinci-003",
-                prompt=f"""This is a contract document content. Your task is to identify aggressive terms like warning terms, penalties in the given contract:
-                (Example: "The bank may take possession of the property.")
-                ```contract: {contract_text}```
-                """,
-                max_tokens=70,
+            conversation = [
+                            {"role": "system", "content": "You are a helpful Aggressive Terms Finder in Given Contract."},
+                            {"role": "user", "content": f"""This is a contract document content. Your task is to identify aggressive terms like warning terms, penalties in the given contract:
+                              ```contract: {contract_text}```"""}
+                            ]
+                        
+            # 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
             )
-            aggressive_terms = response.choices[0].text.strip()
-            return aggressive_terms
+            response = chat_completion.choices[0].message.content
+            return response
+            
         except Exception as e:
             print(f"An error occurred during text analysis: {str(e)}")