diff --git a/tradingagents/graph/portfolio_analysis.py b/tradingagents/graph/portfolio_analysis.py
index a0b96a46..52d1012c 100644
--- a/tradingagents/graph/portfolio_analysis.py
+++ b/tradingagents/graph/portfolio_analysis.py
@@ -5,7 +5,7 @@ import traceback
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Tuple
 
-from langchain_openai import ChatOpenAI
+from langchain_core.language_models.chat_models import BaseChatModel
 
 
 class PortfolioAnalyzer: