江不江 commited on
Commit
d3dad46
·
1 Parent(s): eec0415

fix embedding_model (#1698)

Browse files

### What problem does this PR solve?
fix embedding_model #1692
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Signed-off-by: seaver <[email protected]>

Files changed (1) hide show
  1. rag/llm/embedding_model.py +2 -2
rag/llm/embedding_model.py CHANGED
@@ -128,8 +128,8 @@ class LocalAIEmbed(Base):
128
  ) # local embedding for LmStudio donot count tokens
129
 
130
  def encode_queries(self, text):
131
- res = self.client.embeddings.create(text, model=self.model_name)
132
- return np.array(res.data[0].embedding), 1024
133
 
134
 
135
  class AzureEmbed(OpenAIEmbed):
 
128
  ) # local embedding for LmStudio donot count tokens
129
 
130
  def encode_queries(self, text):
131
+ embds, cnt = self.encode([text])
132
+ return np.array(embds[0]), cnt
133
 
134
 
135
  class AzureEmbed(OpenAIEmbed):