minko186 commited on
Commit
4df8ced
·
verified ·
1 Parent(s): 237e24d

Update ai_generate.py

Browse files
Files changed (1) hide show
  1. ai_generate.py +6 -15
ai_generate.py CHANGED
@@ -54,20 +54,11 @@ def generate(text, model, api):
54
  elif model == "Gemma":
55
  return generate_groq(text, "gemma2-9b-it")
56
  elif model == "OpenAI GPT 3.5":
57
- try:
58
- openai_client = OpenAI(api_key=api)
59
- return generate_openai(text, "gpt-3.5-turbo", openai_client)
60
- except:
61
- return "Please add a valid API key"
62
  elif model == "OpenAI GPT 4":
63
- try:
64
- openai_client = OpenAI(api_key=api)
65
- return generate_openai(text, "gpt-4-turbo", openai_client)
66
- except:
67
- return "Please add a valid API key"
68
  elif model == "OpenAI GPT 4o":
69
- try:
70
- openai_client = OpenAI(api_key=api)
71
- return generate_openai(text, "gpt-4o", openai_client)
72
- except:
73
- return "Please add a valid API key"
 
54
  elif model == "Gemma":
55
  return generate_groq(text, "gemma2-9b-it")
56
  elif model == "OpenAI GPT 3.5":
57
+ openai_client = OpenAI(api_key=api)
58
+ return generate_openai(text, "gpt-3.5-turbo", openai_client)
 
 
 
59
  elif model == "OpenAI GPT 4":
60
+ openai_client = OpenAI(api_key=api)
61
+ return generate_openai(text, "gpt-4-turbo", openai_client)
 
 
 
62
  elif model == "OpenAI GPT 4o":
63
+ openai_client = OpenAI(api_key=api)
64
+ return generate_openai(text, "gpt-4o", openai_client)