소스 검색

go: support openai route

Frank 3 달 전
부모
커밋
6a288426b8
1개의 변경된 파일과 14개의 추가작업 그리고 0개의 파일을 삭제
  1. 14 0
      packages/console/app/src/routes/zen/go/v1/responses.ts

+ 14 - 0
packages/console/app/src/routes/zen/go/v1/responses.ts

@@ -0,0 +1,14 @@
+import type { APIEvent } from "@solidjs/start/server"
+import { handler } from "~/routes/zen/util/handler"
+import { parseOpenAiVariant } from "~/routes/zen/util/variant"
+
+export function POST(input: APIEvent) {
+  return handler(input, {
+    format: "openai",
+    modelList: "lite",
+    parseApiKey: (headers: Headers) => headers.get("authorization")?.split(" ")[1],
+    parseModel: (url: string, body: any) => body.model,
+    parseVariant: (url: string, body: any) => parseOpenAiVariant(body),
+    parseIsStream: (url: string, body: any) => !!body.stream,
+  })
+}