Update aggressive_content_finder.py

aggressive_content_finder.py

@@ -30,6 +30,11 @@ class AggressiveContentFinder:
             str: Identified aggressive terms.
         """
         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 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:
@@ -37,16 +42,12 @@ class AggressiveContentFinder:
                             ]
                         
             # 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