fahmiaziz98 commited on
Commit
0f2feed
·
1 Parent(s): 80962c2

[FIX] Correct response structure in create_query_embedding to return all embeddings

Browse files
Files changed (1) hide show
  1. src/api/routers/embedding.py +1 -1
src/api/routers/embedding.py CHANGED
@@ -250,7 +250,7 @@ async def create_query_embedding(
250
  processing_time = time.time() - start_time
251
 
252
  response = DenseEmbedResponse(
253
- embeddings=embeddings[0],
254
  dimension=len(embeddings[0]),
255
  model_id=request.model_id,
256
  processing_time=processing_time,
 
250
  processing_time = time.time() - start_time
251
 
252
  response = DenseEmbedResponse(
253
+ embeddings=embeddings,
254
  dimension=len(embeddings[0]),
255
  model_id=request.model_id,
256
  processing_time=processing_time,