62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
"""
|
|
Quick test to verify Ollama works with TradingAgents analysts.
|
|
This is a minimal test that should complete quickly.
|
|
"""
|
|
|
|
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
|
from tradingagents.default_config import DEFAULT_CONFIG
|
|
import os
|
|
|
|
print("="*60)
|
|
print("Quick Ollama Test with Market Analyst")
|
|
print("="*60)
|
|
print()
|
|
|
|
# Make sure we have Alpha Vantage key
|
|
if not os.getenv("ALPHA_VANTAGE_API_KEY"):
|
|
print("⚠️ Setting dummy Alpha Vantage key for testing...")
|
|
os.environ["ALPHA_VANTAGE_API_KEY"] = "demo"
|
|
|
|
# Configure for Ollama
|
|
config = DEFAULT_CONFIG.copy()
|
|
config["llm_provider"] = "ollama"
|
|
config["deep_think_llm"] = "llama3.2" # llama3.2 supports tool calling
|
|
config["quick_think_llm"] = "llama3.2" # llama3.2 supports tool calling
|
|
config["backend_url"] = "http://localhost:11434"
|
|
|
|
print("Configuration:")
|
|
print(f" Provider: {config['llm_provider']}")
|
|
print(f" Model: {config['quick_think_llm']}")
|
|
print()
|
|
|
|
# Create graph with just market analyst for quick test
|
|
print("Creating TradingAgentsGraph with market analyst only...")
|
|
ta = TradingAgentsGraph(
|
|
config=config,
|
|
debug=True,
|
|
selected_analysts=["market"] # Just market analyst for speed
|
|
)
|
|
print("✅ Graph created!")
|
|
print()
|
|
|
|
print("Testing with AAPL on 2024-05-10...")
|
|
print("This will test if Ollama can handle tool binding...")
|
|
print()
|
|
|
|
try:
|
|
state, decision = ta.propagate("AAPL", "2024-05-10")
|
|
print()
|
|
print("="*60)
|
|
print("✅ SUCCESS! Ollama works with TradingAgents!")
|
|
print("="*60)
|
|
print(f"\nDecision: {decision}")
|
|
|
|
except Exception as e:
|
|
print()
|
|
print("="*60)
|
|
print("❌ Error occurred")
|
|
print("="*60)
|
|
print(f"\nError: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|