商汤日日新5.0 python对话助手程序

1、导入Python库

pip install sensenova

2、复制代码，填入ID、秘钥，运行程序。商量大模型API：https://console.sensecore.cn/home

import sensenova
import sys

def chat_with_assistant(user_input):
    # 填写真实的模型ID
    model_id = "SenseChat-5"

    # 设置 API 密钥
    sensenova.access_key_id = "你的ID"
    sensenova.secret_access_key = "你的密钥"

    resp = sensenova.ChatCompletion.create(
        messages=[{"role": "user", "content": user_input}],
        model=model_id,
        stream=False,
        max_new_tokens=1024,
        n=1,
        repetition_penalty=1.05,
        temperature=0.8,
        top_p=0.7,
        know_ids=[],
        user="sensenova-python-test-user",
        knowledge_config={
            "control_level": "normal",
            "knowledge_base_result": True,
            "knowledge_base_configs":[]
        },
        plugins={
            "associated_knowledge": {
                "content": "需要注入给模型的知识",
                "mode": "concatenate"
            },
            "web_search": {
                "search_enable": True,
                "result_enable": True
            },
        }
    )

    resp = [resp]  # 将输出结果转换为列表以方便处理
    assistant_response = ""

    for part in resp:
        choices = part['data']["choices"]
        for c_idx, c in enumerate(choices):
            if len(choices) > 1:
                assistant_response += "===== Chat Completion {} =====\n".format(c_idx)
            assistant_response += c["message"]
            if len(choices) > 1:
                assistant_response += "\n"

    return assistant_response

def main():
    print("欢迎使用对话助手程序！")
    print("输入 '退出' 来结束对话。")

    while True:
        user_input = input("你: ")
        
        if user_input == '退出':
            print("对话结束。")
            break
        
        assistant_response = chat_with_assistant(user_input)
        print("对话助手:", assistant_response)

if __name__ == "__main__":
    main()

3、演示