Upload 8 files

.gitignore

@@ -0,0 +1,5 @@
+node_modules
+package-lock.json
+.vscode
+config
+test.js

README.md

@@ -1,10 +1,11 @@
----
-title: T1
-emoji: 👀
-colorFrom: green
-colorTo: indigo
-sdk: docker
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# PPLX-Proxy
+
+A proxy for Perplexity AI
+
+把 Perplexity 转换为 Anthropic 格式的通用代理。
+
+[**Usage 使用方法**](usage.md)
+
+**It is forbidden to use this project for profit.**
+
+**仅供个人部署用于访问自己合法取得的订阅，严禁用于转售或其他商业用途。不提供任何技术支持、不为任何违规使用导致的封号负责。**

docker-build.sh

@@ -0,0 +1 @@
+docker build . -t 'archeb/pplx-proxy'

index.js

@@ -0,0 +1,257 @@
+const express = require("express");
+const { io } = require("socket.io-client");
+const { v4: uuidv4 } = require("uuid");
+const { ProxyAgent } = require("proxy-agent");
+const agent = new ProxyAgent();
+
+const app = express();
+const port = process.env.PORT || 8081;
+
+var opts = {
+	agent: agent,
+	auth: {
+		jwt: "anonymous-ask-user",
+	},
+	reconnection: false,
+	transports: ["websocket"],
+	path: "/socket.io",
+	hostname: "www.perplexity.ai",
+	secure: true,
+	port: "443",
+	extraHeaders: {
+		Cookie: process.env.PPLX_COOKIE,
+		"User-Agent": process.env.USER_AGENT,
+		Accept: "*/*",
+		priority: "u=1, i",
+		Referer: "https://www.perplexity.ai/",
+	},
+};
+
+app.post("/v1/messages", (req, res) => {
+	req.rawBody = "";
+	req.setEncoding("utf8");
+
+	req.on("data", function (chunk) {
+		req.rawBody += chunk;
+	});
+
+	req.on("end", async () => {
+		res.setHeader("Content-Type", "text/event-stream;charset=utf-8");
+		try {
+			let jsonBody = JSON.parse(req.rawBody);
+			if (jsonBody.stream == false) {
+				res.send(
+					JSON.stringify({
+						id: uuidv4(),
+						content: [
+							{
+								text: "Please turn on streaming.",
+							},
+							{
+								id: "string",
+								name: "string",
+								input: {},
+							},
+						],
+						model: "string",
+						stop_reason: "end_turn",
+						stop_sequence: "string",
+						usage: {
+							input_tokens: 0,
+							output_tokens: 0,
+						},
+					})
+				);
+			} else if (jsonBody.stream == true) {
+				// 计算用户消息长度
+				let userMessage = [{ question: "", answer: "" }];
+				let userQuery = "";
+				let lastUpdate = true;
+				if (jsonBody.system) {
+					// 把系统消息加入messages的首条
+					jsonBody.messages.unshift({ role: "system", content: jsonBody.system });
+				}
+				console.log(jsonBody.messages);
+				jsonBody.messages.forEach((msg) => {
+					if (msg.role == "system" || msg.role == "user") {
+						if (lastUpdate) {
+							userMessage[userMessage.length - 1].question += msg.content + "\n";
+						} else if (userMessage[userMessage.length - 1].question == "") {
+							userMessage[userMessage.length - 1].question += msg.content + "\n";
+						} else {
+							userMessage.push({ question: msg.content + "\n", answer: "" });
+						}
+						lastUpdate = true;
+					} else if (msg.role == "assistant") {
+						if (!lastUpdate) {
+							userMessage[userMessage.length - 1].answer += msg.content + "\n";
+						} else if (userMessage[userMessage.length - 1].answer == "") {
+							userMessage[userMessage.length - 1].answer += msg.content + "\n";
+						} else {
+							userMessage.push({ question: "", answer: msg.content + "\n" });
+						}
+						lastUpdate = false;
+					}
+				});
+                // user message to plaintext
+                let previousMessages = jsonBody.messages
+                    .map((msg) => {
+                        return msg.content
+                    })
+                    .join("\n\n");
+
+                let msgid = uuidv4();
+				// send message start
+				res.write(
+					createEvent("message_start", {
+						type: "message_start",
+						message: {
+							id: msgid,
+							type: "message",
+							role: "assistant",
+							content: [],
+							model: "claude-3-opus-20240229",
+							stop_reason: null,
+							stop_sequence: null,
+							usage: { input_tokens: 8, output_tokens: 1 },
+						},
+					})
+				);
+				res.write(createEvent("content_block_start", { type: "content_block_start", index: 0, content_block: { type: "text", text: "" } }));
+				res.write(createEvent("ping", { type: "ping" }));
+
+				// proxy response
+                var socket = io("wss://www.perplexity.ai/", opts);
+
+                socket.on("connect", function () {
+                    console.log(" > [Connected]");
+                    socket
+                        .emitWithAck("perplexity_ask", previousMessages, {
+                            "version": "2.9",
+                            "source": "default",
+                            "attachments": [],
+                            "language": "en-GB",
+                            "timezone": "Europe/London",
+                            "search_focus": "writing",
+                            "frontend_uuid": uuidv4(),
+                            "mode": "concise",
+                            "is_related_query": false,
+                            "is_default_related_query": false,
+                            "visitor_id": uuidv4(),
+                            "frontend_context_uuid": uuidv4(),
+                            "prompt_source": "user",
+                            "query_source": "home"
+                        })
+                        .then((response) => {
+                            
+                            console.log(response);
+                            res.write(createEvent("content_block_stop", { type: "content_block_stop", index: 0 }));
+                            res.write(
+                                createEvent("message_delta", {
+                                    type: "message_delta",
+                                    delta: { stop_reason: "end_turn", stop_sequence: null },
+                                    usage: { output_tokens: 12 },
+                                })
+                            );
+                            res.write(createEvent("message_stop", { type: "message_stop" }));
+
+                            res.end();
+                        }).catch((error) => {
+							if(error.message != "socket has been disconnected"){
+								console.log(error);
+							}
+						});
+                });
+                socket.onAny((event, ...args) => {
+                    console.log(`> [got ${event}]`);
+                });
+                socket.on("query_progress", (data) => {
+                    if(data.text){
+                        var text = JSON.parse(data.text)
+                        var chunk = text.chunks[text.chunks.length - 1];
+                        if(chunk){
+                            chunkJSON = JSON.stringify({
+                                type: "content_block_delta",
+                                index: 0,
+                                delta: { type: "text_delta", text: chunk },
+                            });
+                            res.write(createEvent("content_block_delta", chunkJSON));
+                        }
+                    }
+                });
+                socket.on("disconnect", function () {
+                    console.log(" > [Disconnected]");
+                });
+                socket.on("error", (error) => {
+					chunkJSON = JSON.stringify({
+						type: "content_block_delta",
+						index: 0,
+						delta: { type: "text_delta", text: "Error occured while fetching output 输出时出现错误\nPlease refer to the log for more information 请查看日志以获取更多信息" },
+					});
+					res.write(createEvent("content_block_delta", chunkJSON));
+					res.write(createEvent("content_block_stop", { type: "content_block_stop", index: 0 }));
+					res.write(
+						createEvent("message_delta", {
+							type: "message_delta",
+							delta: { stop_reason: "end_turn", stop_sequence: null },
+							usage: { output_tokens: 12 },
+						})
+					);
+					res.write(createEvent("message_stop", { type: "message_stop" }));
+
+					res.end();
+                    console.log(error);
+                });
+                socket.on("connect_error", function (error) {
+					chunkJSON = JSON.stringify({
+						type: "content_block_delta",
+						index: 0,
+						delta: { type: "text_delta", text: "Failed to connect to the Perplexity.ai 连接到Perplexity失败\nPlease refer to the log for more information 请查看日志以获取更多信息" },
+					});
+					res.write(createEvent("content_block_delta", chunkJSON));
+					res.write(createEvent("content_block_stop", { type: "content_block_stop", index: 0 }));
+					res.write(
+						createEvent("message_delta", {
+							type: "message_delta",
+							delta: { stop_reason: "end_turn", stop_sequence: null },
+							usage: { output_tokens: 12 },
+						})
+					);
+					res.write(createEvent("message_stop", { type: "message_stop" }));
+
+					res.end();
+                    console.log(error);
+                });
+				res.on("close", function () {
+					console.log(" > [Client closed]");
+					socket.disconnect();
+				});
+			} else {
+				throw new Error("Invalid request");
+			}
+		} catch (e) {
+			console.log(e);
+			res.write(JSON.stringify({ error: e.message }));
+			res.end();
+			return;
+		}
+	});
+});
+
+// handle other
+app.use((req, res, next) => {
+	res.status(404).send("Not Found");
+});
+
+app.listen(port, () => {
+	console.log(`Perplexity proxy listening on port ${port}`);
+});
+
+// eventStream util
+function createEvent(event, data) {
+	// if data is object, stringify it
+	if (typeof data === "object") {
+		data = JSON.stringify(data);
+	}
+	return `event: ${event}\ndata: ${data}\n\n`;
+}

package.json

@@ -0,0 +1,17 @@
+{
+  "name": "pplx-proxy",
+  "version": "1.0.0",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "author": "",
+  "license": "ISC",
+  "description": "",
+  "dependencies": {
+    "express": "^4.19.2",
+    "proxy-agent": "^6.4.0",
+    "socket.io-client": "^4.7.5",
+    "uuid": "^9.0.1"
+  }
+}

start.bat

@@ -0,0 +1,6 @@
+call npm install
+set PPLX_COOKIE=Your Cookie
+set USER_AGENT=Your User-Agent
+set all_proxy=
+node index
+pause

start.sh

@@ -0,0 +1,6 @@
+#!/bin/sh
+npm install
+export PPLX_COOKIE="Your Cookie"
+export USER_AGENT="Your User-Agent"
+export all_proxy=""
+node index

usage.md

@@ -0,0 +1,57 @@
+# 使用方法
+
+1. 获得一个 Perplexity 账户并且订阅，登录。
+
+2. 打开 F12（DevTools），找到 “Network（网络）”、刷新一下页面，找到“www.perplexity.ai”这个项目
+
+3. 点进去，往下滑找到 "Cookie"，完整的复制后面的内容。
+
+4. 用同样的方法找到 "User-Agent"，完整的复制后面的内容。
+
+5. 下载或Clone本项目代码，解压
+
+6. 编辑 `start.bat` 文件，把上面的 Cookie 和 User Agent 粘贴进去
+
+7. 启动 start.bat （Linux/macOS用户使用 start.sh）
+
+8. 酒馆中选择 Claude，反向代理地址填 http://127.0.0.1:8081/v1。反代密码必须填，随便什么都可以。
+
+9. 开始使用。如果失败了/没有结果/403/Warning 就多重试几次。
+
+# 使用代理
+
+可以使用本地的socks5或http(s)代理。只需在 start.bat 中设置 `all_proxy` 环境变量。
+
+比如，如要使用 Clash 的默认本地SOCKS5代理，则应设置为 `set all_proxy=socks5://127.0.0.1:7890`
+
+## 注意事项
+
+出现 403 错误请重新抓 COOKIE 或者更换代理出口 IP。
+
+# Usage
+
+1. Get a Perplexity account and subscribe, log in.
+
+2. Open F12 (DevTools), find “Network”, refresh the page, and find “www.perplexity.ai”.
+
+3. Click on it, scroll down and find “Cookie:”, and copy the entire contents.
+
+4. Find the "user-agnet" in the same way.
+
+5. Download or Clone the code of this project and unzip it.
+
+6. Edit the `start.bat` file and paste the cookie and User Agent into it.
+
+7. Start start.bat
+
+8. Select Claude in the Tavern and put http://127.0.0.1:8081/v1 as the address of the reverse proxy. Use any random string for password.
+
+9. Enjoy it. If it fails/no result/403/Warning, try again.
+
+# Use custom proxy
+
+Use the `all_proxy` env to set custom proxy. Refer to https://www.npmjs.com/package/proxy-from-env for detail.
+
+## Caution
+
+If you get 403 errors, consider getting the cookie again or changing your IP.