kkAIGC/SQLBot

1import openai2 3#SQL产生函数4def sql_generation(5    user_input,6    model="gpt-3.5-turbo-16k",7    temperature=0,8    max_tokens=3000,9    sql_type = 'hive'10    ):11 12    system_message = f"""13    根据用户的描述写一段{sql_type} SQL代码，仅提供{sql_type} SQL代码：14    """15 16    messages =  [17        {'role':'system', 'content': system_message},18        {'role':'user', 'content': user_input},19        ]20 21    response = openai.ChatCompletion.create(22        model=model,23        messages=messages,24        temperature=temperature,25        max_tokens=max_tokens,26    )27    return response.choices[0].message["content"]28 29# 定义合并函数30def merge_textbox(textbox1='', textbox2='', textbox3='', textbox4=''):31    merged_text = f"{textbox1}\n{textbox2}\n{textbox3}\n{textbox4}"32    return merged_text33