types of embedding

Author: cuipengfei

src/routes/embeddings/route.ts

@@ -2,17 +2,15 @@ import consola from "consola"
 import { Hono } from "hono"
 import { FetchError } from "ofetch"
 
-import type { ChatCompletionsPayload } from "~/services/copilot/chat-completions/types.ts"
+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<ChatCompletionsPayload>(),
-    )
+    const embeddings = await embedding(await c.req.json<EmbeddingRequest>())
     return c.json(embeddings)
   } catch (error) {
     if (error instanceof FetchError) {

src/services/copilot/embedding/service.ts

@@ -1,10 +1,9 @@
-import type { ChatCompletionsPayload } from "~/services/copilot/chat-completions/types.ts"
-import type { GetModelsResponse } from "~/services/copilot/get-models/types.ts"
+import type { EmbeddingRequest, EmbeddingResponse } from "./types"
 
 import { copilot } from "../../api-instance"
 
-export const embedding = (payload: ChatCompletionsPayload) =>
-  copilot<GetModelsResponse>("/embeddings", {
+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
+  }
+}