Merge pull request ericc-ch#10 from cuipengfei/master

add the embeddings api

Author: ericc-ch

src/routes/embeddings/route.ts

@@ -0,0 +1,21 @@
+import consola from "consola"
+import { Hono } from "hono"
+import { FetchError } from "ofetch"
+
+import type { EmbeddingRequest } from "~/services/copilot/embedding/types"
+
+import { embedding } from "~/services/copilot/embedding/service"
+
+export const embeddingRoutes = new Hono()
+
+embeddingRoutes.post("/", async (c) => {
+  try {
+    const embeddings = await embedding(await c.req.json<EmbeddingRequest>())
+    return c.json(embeddings)
+  } catch (error) {
+    if (error instanceof FetchError) {
+      consola.error(`Request failed: ${error.message}`, error.response?._data)
+    }
+    throw error
+  }
+})

src/server.ts

@@ -3,6 +3,7 @@ import { cors } from "hono/cors"
 import { logger } from "hono/logger"
 
 import { completionRoutes } from "./routes/chat-completions/route"
+import { embeddingRoutes } from "./routes/embeddings/route"
 import { modelRoutes } from "./routes/models/route"
 
 export const server = new Hono()
@@ -14,7 +15,9 @@ server.get("/", (c) => c.text("Server running"))
 
 server.route("/chat/completions", completionRoutes)
 server.route("/models", modelRoutes)
+server.route("/embeddings", embeddingRoutes)
 
 // Compatibility with tools that expect v1/ prefix
 server.route("/v1/chat/completions", completionRoutes)
 server.route("/v1/models", modelRoutes)
+server.route("/v1/embeddings", embeddingRoutes)

src/services/copilot/embedding/service.ts

@@ -0,0 +1,11 @@
+import type { EmbeddingRequest, EmbeddingResponse } from "./types"
+
+import { copilot } from "../../api-instance"
+
+export const embedding = (payload: EmbeddingRequest) =>
+  copilot<EmbeddingResponse>("/embeddings", {
+    method: "POST",
+    body: {
+      ...payload,
+    },
+  })

src/services/copilot/embedding/types.ts

@@ -0,0 +1,20 @@
+export interface EmbeddingRequest {
+  input: string | Array<string>
+  model: string
+}
+
+export interface Embedding {
+  object: string
+  embedding: Array<number>
+  index: number
+}
+
+export interface EmbeddingResponse {
+  object: string
+  data: Array<Embedding>
+  model: string
+  usage: {
+    prompt_tokens: number
+    total_tokens: number
+  }
+}