diff --git a/agenthub/monologue_agent/utils/memory.py b/agenthub/monologue_agent/utils/memory.py index 50e76946d4..62380dcb5c 100644 --- a/agenthub/monologue_agent/utils/memory.py +++ b/agenthub/monologue_agent/utils/memory.py @@ -21,7 +21,8 @@ if embedding_strategy == "llama2": elif embedding_strategy == "openai": from llama_index.embeddings.openai import OpenAIEmbedding embed_model = OpenAIEmbedding( - model="text-embedding-ada-002" + model="text-embedding-ada-002", + api_key=config.get_or_error("LLM_API_KEY") ) elif embedding_strategy == "azureopenai": from llama_index.embeddings.azure_openai import AzureOpenAIEmbedding # Need to instruct to set these env variables in documentation