Spaces:
Sleeping
Sleeping
KarthickAdopleAI
commited on
Update Clauses_Extractor.py
Browse files- 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 =
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
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 |
|