diff --git a/lang_agent/pipeline.py b/lang_agent/pipeline.py index 6b025b4..0fabac4 100644 --- a/lang_agent/pipeline.py +++ b/lang_agent/pipeline.py @@ -5,6 +5,7 @@ import asyncio import websockets from websockets.asyncio.server import ServerConnection from loguru import logger +import os from langchain.chat_models import init_chat_model from langgraph.prebuilt import create_react_agent @@ -42,6 +43,16 @@ class PipelineConfig(InstantiateConfig): # NOTE: For reference tool_manager_config: ToolManagerConfig = field(default_factory=ToolManagerConfig) + def __post_init__(self): + if self.api_key == "wrong-key" or self.api_key is None: + # logger.info("wrong embedding key, using simple retrieval method") + self.api_key = os.environ.get("ALI_API_KEY") + if self.api_key is None: + logger.error(f"no ALI_API_KEY provided for embedding") + else: + logger.info("ALI_API_KEY loaded from environ") + + class Pipeline: def __init__(self, config:PipelineConfig):