KarthickAdopleAI commited on
Commit
e0e4187
·
verified ·
1 Parent(s): 1a26d38

Update Clauses_Extractor.py

Browse files
Files changed (1) hide show
  1. Clauses_Extractor.py +10 -10
Clauses_Extractor.py CHANGED
@@ -23,6 +23,10 @@ class Clauses:
23
  str: Extracted clauses from GPT-3 response.
24
  """
25
  try:
 
 
 
 
26
  conversation = [
27
  {"role": "system", "content": "You are a helpful Cluases and SubCluases Extracter From Given Content."},
28
  {"role": "user", "content": f"""Extract clauses and sub-clauses from the provided contract PDF
@@ -30,16 +34,12 @@ class Clauses:
30
  ]
31
 
32
  # Call OpenAI GPT-3.5-turbo
33
- chat_completion = openai.ChatCompletion.create(
34
- engine="ChatGPT",
35
- messages = conversation,
36
- temperature=0.7,
37
- max_tokens=800,
38
- top_p=0.95,
39
- frequency_penalty=0,
40
- presence_penalty=0,
41
- stop=None
42
- )
43
  response = chat_completion.choices[0].message.content
44
  return response
45
 
 
23
  str: Extracted clauses from GPT-3 response.
24
  """
25
  try:
26
+ client = AzureOpenAI(api_key=os.getenv("AZURE_OPENAI_KEY"),
27
+ api_version="2023-07-01-preview",
28
+ azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
29
+ )
30
  conversation = [
31
  {"role": "system", "content": "You are a helpful Cluases and SubCluases Extracter From Given Content."},
32
  {"role": "user", "content": f"""Extract clauses and sub-clauses from the provided contract PDF
 
34
  ]
35
 
36
  # Call OpenAI GPT-3.5-turbo
37
+ chat_completion = client.chat.completions.create(
38
+ model = "ChatGPT",
39
+ messages = conversation,
40
+ max_tokens=1000,
41
+ temperature=0
42
+ )
 
 
 
 
43
  response = chat_completion.choices[0].message.content
44
  return response
45