github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix get_ada_embedding return type (#3263)

Browse Source
This commit is contained in:
Robin Richtsfeld
2023-04-26 01:52:38 +02:00
committed by GitHub
parent 8670b3039e
commit d75379358f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
autogpt/llm_utils.py
View File

@@ -209,14 +209,14 @@ def create_chat_completion(
return resp
def get_ada_embedding(text: str) -> List[int]:
def get_ada_embedding(text: str) -> List[float]:
"""Get an embedding from the ada model.
Args:
text (str): The text to embed.
Returns:
List[int]: The embedding.
List[float]: The embedding.
"""
cfg = Config()
model = "text-embedding-ada-002"