Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add key

Browse Source
This commit is contained in:
goulustis
2025-10-14 22:12:29 +08:00
parent 8a3130ee6f
commit bc898e526f
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lang_agent/pipeline.py
View File

@@ -5,6 +5,7 @@ import asyncio
import websockets import websockets
from websockets.asyncio.server import ServerConnection from websockets.asyncio.server import ServerConnection
from loguru import logger from loguru import logger
import os
from langchain.chat_models import init_chat_model from langchain.chat_models import init_chat_model
from langgraph.prebuilt import create_react_agent from langgraph.prebuilt import create_react_agent
@@ -42,6 +43,16 @@ class PipelineConfig(InstantiateConfig):
# NOTE: For reference # NOTE: For reference
tool_manager_config: ToolManagerConfig = field(default_factory=ToolManagerConfig) 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: class Pipeline:
def __init__(self, config:PipelineConfig): def __init__(self, config:PipelineConfig):