使用大语言模型生成关键词

最新推荐文章于 2024-10-02 05:39:15 发布

無敗の草

最新推荐文章于 2024-10-02 05:39:15 发布

阅读量276

点赞数 4

文章标签：语言模型 javascript 服务器

本文链接：https://blog.csdn.net/weixin_52717285/article/details/142259386

版权

使用大语言模型生成关键词

使用OpenAI API将文本文件中的每一行内容转换为与主商品关联的三个印尼语关键字。首先，它定义了一个名为openai_join的函数，用于创建一个OpenAI客户端对象。然后，定义了一个名为openai的函数，该函数接受一个OpenAI客户端对象和一个字符串作为参数，并使用OpenAI的聊天模型生成与输入字符串相关的印尼语关键字。

接下来，代码读取名为"a.txt"的文件，并将其内容存储在变量data中。然后，它创建一个名为"b.txt"的新文件，用于存储生成的印尼语关键字。对于data中的每一行，代码调用openai函数并将结果写入"b.txt"文件。最后，关闭文件。

from openai import OpenAI


def openai_join():
    client = OpenAI(
        api_key="6dcc6a1f95b97e3be804c24fee176bce:ZDY5YTYxY2YyZmI2NThkZTU3NjdhMTVl",
        base_url='https://spark-api-open.xf-yun.com/v1'
    )
    return client
def openai(join, string=""):
    
    completion = join.chat.completions.create(
        model='generalv3.5',
        messages=[{"role": "user", "content": string, }]
    )
    return completion.choices[0].message.content


join = openai_join()
f = open("a.txt",mode="r", encoding="utf-8")
data = f.read()
f.close()
f1 = open("b.txt", mode="w", encoding="utf-8")
for i in data.split("\n"):
    tag3 = openai(join, i + " 给出与主商品关联的三个印尼语关键字，关键字需要描述出是什么商品，只要关键字不要描述")
    f1.write('"%s"\n' %(tag3))
    print(i)

f1.close()