single_round_conversation
single_round_conversation (api_key, base_url, model, user_prompt, system_prompt=None, image_path=None, temperature=None, max_tokens=None, timeout=200, kwargs)
方法描述
使用OpenAI,调用文本生成类AI大模型,完成单轮对话任务
参数说明
api_key<str>模型调用的密钥,可以在对应的模型平台中获取
base_url<str>模型服务地址,如通义系列大模型的地址为:'https://dashscope.aliyuncs.com/compatible-mode/v1'
model<str>模型的英文名称,用于指定具体需要使用的模型,如,希望调用通义千问-plus,则填写'qwen-plus'
user_prompt<str>代表用户输入的内容,即,用户向模型提出的具体问题或指令,如“计算1+1的值”
system_prompt<str> 用于设定模型的整体行为或背景信息,赋予模型角色、任务、输出格式等,如“你是一个数学专家,解决用户的数学问题”
image_path<str>传给模型的图片路径,支持url(以http或https开头)和本地文件路径,支持PNG、JPEG、BMP、WEBP、TIFF、HEIC图像类型, 请确保调用的模型支持图片传入
temperature<float>采样温度,介于 0 和 2 之间,步长0.01。温度值越高,生成的文本更多样,反之,生成的文本更确定。默认不指定具体值,以所选模型默认值为准
max_tokens<int>介于 1 到 8192 间的整数,限制一次请求中模型生成 completion 的最大 token 数。默认不指定具体值,以所选模型默认值为准
timeout<int>默认值为200秒,超过超时时间则抛出异常
kwargs<str>
返回值说明
OpenAI ChatCompletion对象<ChatCompletion>
调用样例- rpa.ai.llm.single_round_conversation-
# 服务获取可参考大模型服务平台官网:https://bailian.console.aliyun.com/?tab=api#/api/?type=model&url=2712576
# 代码调用样例如下:
api_key=os.getenv("DASHSCOPE_API_KEY")
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
model="qwen-plus"
user_prompt = '计算1+1的值'
response = rpa.ai.llm.single_round_conversation(api_key, base_url, model, user_prompt)
print(response)
print(response.choices[0].message.content)