Initial codes commit
Browse files- 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 |
-
#
|
| 15 |
-
embedding_2d = _client.
|
| 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)
|