Bill/AI-Trader
1
0
Fork 0
mirror of https://github.com/Xe138/AI-Trader.git synced 2026-04-08 11:47:24 -04:00
Code Issues Packages Projects Releases Wiki Activity

support multi url and key

Browse Source
This commit is contained in:
tianyufan
2025-10-27 21:37:46 +08:00
parent 5bf8ddbc77
commit eaf9379c21
3 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
agent/base_agent/base_agent.py
View File

@@ -66,6 +66,7 @@ class BaseAgent:
max_retries: int = 3,
base_delay: float = 0.5,
openai_base_url: Optional[str] = None,
openai_api_key: Optional[str] = None,
initial_cash: float = 10000.0,
init_date: str = "2025-10-13"
):
@@ -82,6 +83,7 @@ class BaseAgent:
max_retries: Maximum retry attempts
base_delay: Base delay time for retries
openai_base_url: OpenAI API base URL
openai_api_key: OpenAI API key
initial_cash: Initial cash amount
init_date: Initialization date
"""
@@ -101,7 +103,14 @@ class BaseAgent:
self.base_log_path = log_path or "./data/agent_data"
# Set OpenAI configuration
self.openai_base_url = openai_base_url or os.getenv("OPENAI_API_BASE")
if openai_base_url==None:
self.openai_base_url = os.getenv("OPENAI_API_BASE")
else:
self.openai_base_url = openai_base_url
if openai_api_key==None:
self.openai_api_key = os.getenv("OPENAI_API_KEY")
else:
self.openai_api_key = openai_api_key
# Initialize components
self.client: Optional[MultiServerMCPClient] = None
@@ -149,6 +158,7 @@ class BaseAgent:
self.model = ChatOpenAI(
model=self.basemodel,
base_url=self.openai_base_url,
api_key=self.openai_api_key,
max_retries=3,
timeout=30
)