File size: 1,117 Bytes
771fc51 fe33690 771fc51 fe33690 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import os
import gradio as gr
# 手动设置 API 金钥
os.environ["groq_key"] = "your_groq_api_key"
# 确保 groq 套件已安装
try:
from groq import Groq
except ImportError:
os.system('pip install groq')
from groq import Groq
# 从环境变量中获取 API 金钥
api_key = os.getenv("groq_key")
if api_key is None:
raise ValueError("API key is not set. Please set it in your environment variables with the name 'groq_key'.")
# 初始化 Groq 客户端
client = Groq(api_key=api_key)
# 定义聊天机器人的响应函数
def chatbot_response(messages):
completion = client.chat.completions.create(
model="llama3-8b-8192",
messages=[
{
"role": "system",
"content": "You are a corporate secretary who is skilled at drafting business emails. The prompt will feed you addressee, main message, and final greetings."
}
] + messages,
temperature=1,
max_tokens=1024,
top_p=1,
stream=True,
stop=None,
)
response = ""
for chunk in completion:
respons
|