m97j commited on
Commit
67a03d1
·
1 Parent(s): 9f3b43b

Initial codes commit

Browse files
Files changed (1) hide show
  1. modules/embedder.py +2 -2
modules/embedder.py CHANGED
@@ -11,7 +11,7 @@ def _l2_normalize(vec: List[float]) -> List[float]:
11
  return [x / norm for x in vec]
12
 
13
  def get_embedding(text: str) -> List[float]:
14
- # feature_extraction은 항상 2차원 배열 반환: [batch_size, embedding_dim]
15
- embedding_2d = _client.feature_extraction(text)
16
  vec = embedding_2d[0] # 첫 번째 행이 입력 문장의 벡터
17
  return _l2_normalize(vec)
 
11
  return [x / norm for x in vec]
12
 
13
  def get_embedding(text: str) -> List[float]:
14
+ # embeddings()는 [batch_size, embedding_dim] 형태 반환
15
+ embedding_2d = _client.embeddings(text)
16
  vec = embedding_2d[0] # 첫 번째 행이 입력 문장의 벡터
17
  return _l2_normalize(vec)