cogcorp commited on
Commit
769cc8c
·
1 Parent(s): 074e59b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -13
app.py CHANGED
@@ -7,19 +7,23 @@ from datetime import datetime
7
  # Put your OpenAI API key here
8
  openai.api_key = os.getenv('OpenAPI')
9
 
10
- def gpt3_query(prompt, engine='gpt-3.5-turbo-16k', max_tokens=15050, temperature=0.5):
11
- try:
12
- response = openai.ChatCompletion.create(
13
- model=engine,
14
- messages=[{"role": "system", "content": "You are a creative writer who specializes in crafting engaging stories to sell products or services"},
15
- {"role": "user", "content": prompt}],
16
- max_tokens=max_tokens,
17
- n=1,
18
- temperature=temperature
19
- )
20
- return response.choices[0].message['content'].strip()
21
- except Exception as e:
22
- return f"Error in gpt3_query: {str(e)}", None
 
 
 
 
23
 
24
  def read_file(file_path):
25
  with open(file_path, 'r') as file:
 
7
  # Put your OpenAI API key here
8
  openai.api_key = os.getenv('OpenAPI')
9
 
10
+ def gpt3_query(prompt, engine='gpt-3.5-turbo-16k', max_tokens=15050, temperature=0.5, retries=3):
11
+ for i in range(retries):
12
+ try:
13
+ response = openai.ChatCompletion.create(
14
+ model=engine,
15
+ messages=[{"role": "system", "content": "You are a creative writer who specializes in crafting engaging stories to sell products or services"},
16
+ {"role": "user", "content": prompt}],
17
+ max_tokens=max_tokens,
18
+ n=1,
19
+ temperature=temperature
20
+ )
21
+ return response.choices[0].message['content'].strip()
22
+ except Exception as e:
23
+ if i < retries - 1: # i is zero indexed
24
+ continue
25
+ else:
26
+ return f"Error in gpt3_query after {retries} attempts: {str(e)}", None
27
 
28
  def read_file(file_path):
29
  with open(file_path, 'r') as file: