Update ContractGenerator.py

ContractGenerator.py

@@ -1,4 +1,4 @@
-from openai import OpenAI
+from openai
 
 class ContractGenerator:
     """
@@ -6,15 +6,21 @@ class ContractGenerator:
     """
 
     def __init__(self):
+        
         """
         Initialize the ContractGenerator.
 
         Args:
             api_key (str): Your OpenAI API key.
         """
-        self.client = OpenAI()
-
+        
+           openai.api_type = os.getenv['api_type']
+           openai.api_base = os.getenv['api_base']
+           openai.api_version = os.getenv['api_version']
+           openai.api_key = os.getenv['api_key']
+        
     def generate_contract(self, instructions: str) -> None:
+        
         """
         Generate a contract form based on user instructions.
 
@@ -24,22 +30,24 @@ class ContractGenerator:
         Raises:
             openai.error.OpenAIError: If there is an error with the OpenAI API request.
         """
+        
         # Define a prompt
-        client = OpenAI()
-        os.environ["OPENAI_API_KEY"]="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"
-
         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 = 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