Skip to content

Commit d753793

Browse files
authored
Fix get_ada_embedding return type (#3263)
1 parent 8670b30 commit d753793

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

autogpt/llm_utils.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -209,14 +209,14 @@ def create_chat_completion(
209209
return resp
210210

211211

212-
def get_ada_embedding(text: str) -> List[int]:
212+
def get_ada_embedding(text: str) -> List[float]:
213213
"""Get an embedding from the ada model.
214214
215215
Args:
216216
text (str): The text to embed.
217217
218218
Returns:
219-
List[int]: The embedding.
219+
List[float]: The embedding.
220220
"""
221221
cfg = Config()
222222
model = "text-embedding-ada-002"

0 commit comments

Comments
 (0)