github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-02-01 18:35:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

Changes for Azure embedding handling

Browse Source
This commit is contained in:
Peter Edwards
2023-04-11 13:45:37 +02:00
parent 5a6e565c52
commit 9d33a75083
4 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/memory/base.py
View File

@@ -1,12 +1,17 @@
"""Base class for memory providers."""
import abc
from config import AbstractSingleton
from config import Config
import openai
cfg = Config()
def get_ada_embedding(text):
text = text.replace("\n", " ")
return openai.Embedding.create(input=[text], model="text-embedding-ada-002")["data"][0]["embedding"]
if cfg.use_azure:
return openai.Embedding.create(input=[text], engine=cfg.openai_embedding_deployment_id)["data"][0]["embedding"]
else:
return openai.Embedding.create(input=[text], model="text-embedding-ada-002")["data"][0]["embedding"]
class MemoryProviderSingleton(AbstractSingleton):