Update ContractGenerator.py

ContractGenerator.py

@@ -33,22 +33,23 @@ class ContractGenerator:
         """
         
         # Define a prompt
+        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 Contract Generator."},
                         {"role": "user", "content": f"""Your task is to generate a contract form based on user instructions. ***Instructions:{instructions}***"""}
                       ]
                     
         # 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