import openai

class CTRLChat():
    '''
    OpenAI api调用控制类
    '''
    def chat(self, key:str, q:str, tokens_limit:int) -> str:
      openai.api_key = key
      completion = openai.ChatCompletion.create(
          model="gpt-3.5-turbo",
          messages=[
            {"role": "user", "content": q}
          ],
          max_tokens=tokens_limit
      )

      return completion.choices[0].message.content