Contract_Management / ContractGenerator.py
naveenvenkatesh's picture
Update ContractGenerator.py
9bab356
raw
history blame
1.31 kB
import openai
class ContractGenerator:
"""
A class for generating contract forms based on user instructions using the OpenAI GPT-3.5 model.
"""
def __init__(self, api_key: str):
"""
Initialize the ContractGenerator.
Args:
api_key (str): Your OpenAI API key.
"""
pass
def generate_contract(self, instructions: str) -> None:
"""
Generate a contract form based on user instructions.
Args:
instructions (str): User-provided instructions for the contract form.
Raises:
openai.error.OpenAIError: If there is an error with the OpenAI API request.
"""
# Define a prompt
prompt = f"Your task is to generate a contract form based on user instructions. ***Instructions:{instructions}***"
try:
# Generate text using the GPT-3.5 model
response = openai.Completion.create(
engine="text-davinci-003",
prompt=prompt,
max_tokens=500 # You can adjust the length of the generated text
)
# Print the generated text
return response.choices[0].text
except openai.error.OpenAIError as e:
print(f"Error generating the contract: {str(e)}")