github/OpenHands
1
0
Fork 0
mirror of https://github.com/OpenHands/OpenHands.git synced 2025-12-26 05:48:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix to pass api key to openai embedding (#754)

Browse Source
This commit is contained in:
Vincent 2024-04-05 13:28:10 -04:00 committed by GitHub
parent a202b2550f
commit da12d70bc2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
agenthub/monologue_agent/utils/memory.py
View File

@ -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