Update Clauses_Extractor.py

Clauses_Extractor.py

@@ -1,12 +1,15 @@
 import openai
 import os
-from openai import OpenAI
 
 # Define the Clauses class
 class Clauses:
+    
     def __init__(self):
-
-           self.client = OpenAI()
+            
+           openai.api_type = os.getenviron['api_type']
+           openai.api_base = os.getenviron['api_base']
+           openai.api_version = os.getenviron['api_version']
+           openai.api_key = os.getenviron['api_key']
 
     def get_extracted_clauses(self,extracted_summary):
         """
@@ -26,12 +29,16 @@ class Clauses:
                           ]
                         
             # 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
-            )
+            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