xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-11-13 12:43:42 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

build: update langchain

Browse Source
This commit is contained in:
Hk-Gosuto
2024-01-24 20:51:46 +08:00
parent 30f9dc756a
commit 16e82afaad
16 changed files with 161 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/api/langchain/tool/agent/edge/route.ts
View File

@@ -2,9 +2,8 @@ import { NextRequest, NextResponse } from "next/server";
import { AgentApi, RequestBody, ResponseBody } from "../agentapi";
import { auth } from "@/app/api/auth";
import { EdgeTool } from "../../../../langchain-tools/edge_tools";
import { OpenAI } from "langchain/llms/openai";
import { OpenAIEmbeddings } from "langchain/embeddings/openai";
import { ModelProvider } from "@/app/constant";
import { OpenAI, OpenAIEmbeddings } from "@langchain/openai";
async function handle(req: NextRequest) {
if (req.method === "OPTIONS") {
@@ -21,7 +20,8 @@ async function handle(req: NextRequest) {
const encoder = new TextEncoder();
const transformStream = new TransformStream();
const writer = transformStream.writable.getWriter();
const agentApi = new AgentApi(encoder, transformStream, writer);
const controller = new AbortController();
const agentApi = new AgentApi(encoder, transformStream, writer, controller);
const reqBody: RequestBody = await req.json();
const authToken = req.headers.get("Authorization") ?? "";
@@ -52,6 +52,9 @@ async function handle(req: NextRequest) {
await writer.write(
encoder.encode(`data: ${JSON.stringify(response)}\n\n`),
);
controller.abort({
reason: "dall-e tool abort",
});
};
var edgeTool = new EdgeTool(