Browse Source

Add missing embed response properties (#215)

Pieter Ouwerkerk 3 months ago
parent
commit
b24558558f
1 changed files with 3 additions and 0 deletions
  1. 3 0
      src/interfaces.ts

+ 3 - 0
src/interfaces.ts

@@ -198,6 +198,9 @@ export interface ChatResponse {
 export interface EmbedResponse {
 export interface EmbedResponse {
   model: string
   model: string
   embeddings: number[][]
   embeddings: number[][]
+  total_duration: number
+  load_duration: number
+  prompt_eval_count: number
 }
 }
 
 
 export interface EmbeddingsResponse {
 export interface EmbeddingsResponse {