charieswei
/
TradingAgents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: properly print errors to stderr

This commit is contained in:
mogita 2025-08-16 17:13:56 +08:00
parent d64a505f91
commit 605851f21d
No known key found for this signature in database
GPG Key ID: A0AA1B9C57A48ECF
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tradingagents/utils/provider_utils.py
View File

@ -3,6 +3,7 @@ Utility functions for LLM provider configuration and API key management.
""" """
import os import os
import sys
from openai import OpenAI from openai import OpenAI
def get_api_key_for_provider(config): def get_api_key_for_provider(config):
@ -20,7 +21,7 @@ def get_api_key_for_provider(config):
if provider.startswith("custom"): if provider.startswith("custom"):
api_key = os.getenv("CUSTOM_API_KEY") api_key = os.getenv("CUSTOM_API_KEY")
if not api_key: if not api_key:
print("Warning: CUSTOM_API_KEY not found in environment variables") print("Warning: CUSTOM_API_KEY not found in environment variables", file=sys.stderr)
return api_key return api_key
# Map providers to their environment variables # Map providers to their environment variables
@ -36,7 +37,7 @@ def get_api_key_for_provider(config):
api_key = os.getenv(env_var) api_key = os.getenv(env_var)
if not api_key and provider != "ollama": # Ollama typically doesn't need API keys if not api_key and provider != "ollama": # Ollama typically doesn't need API keys
print(f"Warning: {env_var} not found in environment variables") print(f"Warning: {env_var} not found in environment variables", file=sys.stderr)
return api_key return api_key