import gradio as gr import requests import time import json # 设置 FastGPT API 的基本信息 #API_URL = "https://api.fastgpt.in/api/v1/chat/completions" # 请根据您的部署地址修改 #API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF" # 请替换为您的实际 API 密钥 # 设置 FastGPT API 的基本信息 #API_URL = "https://api.fastgpt.com/generate" # 请根据您的实际部署地址修改 #API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF" # 请替换为您的实际 API 密钥 #import gradio as gr #import requests #API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF" # 请替换为您的实际 API 密钥 #API_URL = "https://api.fastgpt.com/v1/chat" # 替换成你的 FastGPT 的 BaseURL 和 应用特定 API Key #BASE_URL = "https://api.fastgpt.in/api/v1" #API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF" # 应用特定 key # 替换成你的 FastGPT 的 BaseURL 和 应用特定 API Key BASE_URL = "https://api.fastgpt.in/api/v1" API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF" # 应用特定 key import ijson # Install with: pip install ijson def chat_with_fastgpt(message, history): """ 与 FastGPT 进行对话的函数 Args: message: 用户输入的消息 history: 聊天历史记录 Returns: 生成器的响应内容 """ headers = { "Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json", } # 构建消息历史 messages = [] for human, assistant in history: messages.append({"role": "user", "content": human}) messages.append({"role": "assistant", "content": assistant}) messages.append({"role": "user", "content": message}) data = { "chatId": "your_chat_id", # 如果需要,替换成你的 chatId "stream": True, # 仍然使用流式请求 "detail": False, "messages": messages, } try: response = requests.post( f"{BASE_URL}/chat/completions", headers=headers, json=data, stream=True ) response.raise_for_status() buffer = "" for chunk in response.iter_content(chunk_size=8192): if chunk: buffer += chunk.decode('utf-8') # 尝试解析 JSON 对象 while True: # Continue processing the buffer until no more complete objects are found try: first_brace_index = buffer.find('{') if first_brace_index == -1: break obj, end_index = json.JSONDecoder().raw_decode(buffer[first_brace_index:]) # 解析成功,处理 JSON 对象 if obj.get("choices"): delta = obj["choices"][0].get("delta") if delta and delta.get("content"): yield delta["content"] buffer = buffer[first_brace_index + end_index:] except json.JSONDecodeError as e: break except requests.exceptions.RequestException as e: print(f"请求错误: {e}") yield "请求 FastGPT API 出错,请检查网络或 API Key。" except Exception as e: print(f"未知错误: {e}") print(f"错误信息: {e}") yield "发生未知错误。" # Gradio 界面 iface = gr.ChatInterface( fn=chat_with_fastgpt, title="FastGPT Chatbot", description="与 FastGPT 支持的聊天机器人对话", ) iface.launch()