如何调用百度文心一言API实现智能问答 - qidao123.com技术社区-IT企服评测·应用市场

import openai
import backoff
client = openai.OpenAI(
api_key=OPENAI_API_KEY,
base_url="https://aistudio.baidu.com/llm/lmapi/v3"
)
LLM_MODEL="ernie-3.5-8k"
@backoff.on_exception(backoff.expo, (openai.RateLimitError, openai.APITimeoutError))
def llm_check_search(query, msg_history=None, llm_model=LLM_MODEL):
msg_history = msg_history or []
new_msg_history = msg_history + [{"role": "user", "content": query}]
response = client.chat.completions.create(
model=llm_model,
messages=[*new_msg_history],
max_tokens=30
).choices[0].message.content
# 返回 LLM 的响应
return response
print(llm_check_search("你是谁？"))

复制代码

我是文心一言，百度研发的知识增强大语言模型。我能够回答各种问题，提供相关的知识和信息，并且可以进行文本创
作和知识推理等多种任务。如果您有任何问题或需要帮助，请随时告诉我。

复制代码

import openai
import backoff
client = openai.OpenAI(
api_key=OPENAI_API_KEY,
base_url="https://aistudio.baidu.com/llm/lmapi/v3"
)
LLM_MODEL="ernie-3.5-8k"
@backoff.on_exception(backoff.expo, (openai.RateLimitError, openai.APITimeoutError))
def llm_check_search(query, msg_history=None, llm_model=LLM_MODEL):
msg_history = msg_history or []
new_msg_history = msg_history + [{"role": "user", "content": query}]
response = client.chat.completions.create(
model=llm_model,
messages=[*new_msg_history],
max_tokens=30,
stream=True
)
for chunk in response:
chunk_content = chunk.choices[0].delta.content
if chunk_content:
print(chunk_content)
llm_check_search("你是谁？")

复制代码

我是百度
公司研发
的知识增强
大语言
模型，
我的中文名
是文心
一言，
英文名是
ERNIE
Bot。
我可以完成的任务
包括知识
问答，
文本创作
，知识
推理，
数学计算
，代码
理解与编写
，作画
，翻译
等。
如果您有任何
问题，
请随时
向我提问
。

复制代码

import gradio as gr
from try_wenxinyiyan1 import llm_check_search
demo = gr.Interface(
fn=llm_check_search,
inputs=["text"],
outputs=["text"],
)
demo.launch()

复制代码

import gradio as gr
from try_wenxinyiyan1 import llm_check_search
# 修改自定义 CSS
custom_css = """
body {
background-color: #f0f8ff; /* 淡蓝色背景 */
}
.container {max-width: 730px; margin: auto; padding-top: 0;}
.title-container {
display: flex;
flex-direction: column;
align-items: center;
justify-content: center;
margin-top: 5rem;
margin-bottom: 2rem;
}
.main-title {
font-size: 2.5em;
font-weight: bold;
color: #1e90ff; /* 深蓝色标题 */
display: flex;
align-items: center;
justify-content: center;
}
.main-title img {
width: 40px;
height: 40px;
margin-right: 10px;
filter: invert(39%) sepia(95%) saturate(1095%) hue-rotate(194deg) brightness(103%) contrast(101%); /* 将图标改为蓝色 */
}
.powered-by {
font-size: 0.8em;
color: #4682b4; /* 钢蓝色 */
margin-top: 0.5rem;
}
#component-1 {
border: none !important;
box-shadow: none !important;
background: none !important;
}
#component-1 .form {
border: 1px solid #87cefa; /* 淡蓝色边框 */
border-radius: 4px;
padding: 10px;
background-color: white;
}
.button-container {
display: flex;
flex-direction: column;
align-items: center;
}
.search-button.primary {
min-width: 120px !important;
justify-content: center !important;
background-color: #1e90ff !important; /* 深蓝色按钮 */
color: white !important;
border: none !important;
}
.search-button.primary:hover {
background-color: #4169e1 !important; /* 悬停时稍深的蓝色 */
}
.search-button.primary span {
margin: 0 auto !important;
}
"""
# 修改自定义 HTML 标题
custom_title = """
<div class="title-container">
<div class="main-title">
<img src="https://img.icons8.com/ios-filled/50/000000/search--v1.png" alt="Search Icon"/>
<span>AI问答机</span>
</div>
<div class="powered-by">Powered by 文心大模型</div>
</div>
"""
with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as iface:
gr.HTML(custom_title)
with gr.Column(elem_classes="container"):
query_input = gr.Textbox(
lines=1,
placeholder="想对文心一言问什么？",
label=None,
container=False,
)
gr.Examples(
examples=[["火鸡好吃吗？"], ["感恩节的来源是什么？"], ["哲学专业学什么？"]],
inputs=query_input,
)
with gr.Column(elem_classes="button-container"):
submit_button = gr.Button("搜索一下", elem_classes="search-button primary")
output = gr.Markdown()
submit_button.click(fn=llm_check_search, inputs=query_input, outputs=output)
if __name__ == "__main__":
iface.launch()

复制代码