Update key_value_extractor.py

key_value_extractor.py

@@ -22,24 +22,23 @@ class KeyValue:
         Prints the extracted key-value pairs.
         """
         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 Keywords Extracter."},
                             {"role": "user", "content": f"""analyze the given contract and Extract Keywords for following contract in triple backticks. tags should be bullet points.contract :```{extracted_summary}```."""}
                           ]
                         
             # 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