Merge pull request #329 from algorithmicsuperintelligence/fix-key-embedding-model

Fix key embedding model

Author: codelion

openevolve/_version.py

@@ -1,3 +1,3 @@
 """Version information for openevolve package."""
 
-__version__ = "0.2.21"
+__version__ = "0.2.22"

openevolve/embedding.py

@@ -40,7 +40,10 @@ def __init__(
 
     def _get_client_model(self, model_name: str) -> tuple[openai.OpenAI, str]:
         if model_name in OPENAI_EMBEDDING_MODELS:
-            client = openai.OpenAI()
+            # Use OPENAI_EMBEDDING_API_KEY if set, otherwise fall back to OPENAI_API_KEY
+            # This allows users to use OpenRouter for LLMs while using OpenAI for embeddings
+            embedding_api_key = os.getenv("OPENAI_EMBEDDING_API_KEY") or os.getenv("OPENAI_API_KEY")
+            client = openai.OpenAI(api_key=embedding_api_key)
             model_to_use = model_name
         elif model_name in AZURE_EMBEDDING_MODELS:
             # get rid of the azure- prefix