fahmiaziz98 commited on
Commit
af36df4
·
1 Parent(s): fa16bad
Files changed (2) hide show
  1. core/embedding.py +2 -2
  2. core/model_manager.py +0 -1
core/embedding.py CHANGED
@@ -48,8 +48,8 @@ class EmbeddingModel:
48
  self.load()
49
 
50
  try:
51
- embedding = self.model.encode_query(text, prompt=prompt)
52
- return embedding[0].tolist()
53
  except Exception as e:
54
  logger.error(f"Embedding generation failed: {e}")
55
  raise
 
48
  self.load()
49
 
50
  try:
51
+ embeddings = self.model.encode_query(text, prompt=prompt)
52
+ return [embedding.tolist() for embedding in embeddings]
53
  except Exception as e:
54
  logger.error(f"Embedding generation failed: {e}")
55
  raise
core/model_manager.py CHANGED
@@ -142,7 +142,6 @@ class ModelManager:
142
  "repository": config.repository,
143
  }
144
 
145
-
146
  def generate_api_description(self) -> str:
147
  """Generate a dynamic API description based on available models."""
148
 
 
142
  "repository": config.repository,
143
  }
144
 
 
145
  def generate_api_description(self) -> str:
146
  """Generate a dynamic API description based on available models."""
147