utils.py 454 B

123456789101112131415
  1. import openai
  2. import requests
  3. openai.api_key = "YOUR_OPEN_AI_KEY"
  4. model_engine = "text-davinci-003"
  5. # Get the text input from the js server
  6. def get_answer(prompt):
  7. prompt = prompt + " could you write a text about this using 10000 words, do not copy it?"
  8. completions = openai.Completion.create(engine=model_engine, prompt=prompt, max_tokens=2000, n=1,stop=None,temperature=0.5)
  9. response = completions.choices[0].text
  10. return response