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

Merge pull request #700 from Rokniel/azure-deployment-additions

Browse Source 
Add required env parameters for azure usage
This commit is contained in:
Richard Beales
2023-04-12 18:08:54 +01:00
committed by GitHub
parent c98471e333 63d3238c6c
commit 9c9758c720
5 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/memory/base.py
View File

@@ -1,12 +1,16 @@
"""Base class for memory providers."""
import abc
from config import AbstractSingleton
from config import AbstractSingleton, 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.azure_embeddigs_deployment_id, model="text-embedding-ada-002")["data"][0]["embedding"]
else:
return openai.Embedding.create(input=[text], model="text-embedding-ada-002")["data"][0]["embedding"]
class MemoryProviderSingleton(AbstractSingleton):