test_basic_agent.py

"""
Simple SmoLagent Test
====================

Test the basic SmoLagent setup with just CodeAgent and DuckDuckGoSearchTool
"""

from smolagents import CodeAgent, DuckDuckGoSearchTool

def test_basic_agent():
    """Test basic SmoLagent setup without language model"""
    print("🤖 TESTING BASIC SMOLAGENT SETUP")
    print("=" * 35)
    
    try:
        # Create agent with just tools (no language model)
        agent = CodeAgent(tools=[DuckDuckGoSearchTool()])
        print("✅ Basic SmoLagent created successfully!")
        print(f"📋 Agent has {len(agent.tools)} tool(s) available")
        
        # List available tools
        for i, tool in enumerate(agent.tools, 1):
            print(f"   {i}. {tool.__class__.__name__}")
        
        # Test basic functionality
        print("\n🧪 Testing agent functionality...")
        
        # Note: Without a language model, we can't run complex queries
        # But we can verify the agent structure is correct
        print("✅ Agent structure is valid")
        print("⚠️  Note: Full AI analysis requires a language model")
        
        return agent
        
    except Exception as e:
        print(f"❌ Basic agent setup failed: {e}")
        return None

def test_with_search():
    """Test the search functionality"""
    print("\n🔍 TESTING SEARCH FUNCTIONALITY")
    print("=" * 30)
    
    try:
        # Create search tool directly
        search_tool = DuckDuckGoSearchTool()
        print("✅ DuckDuckGo search tool created")
        
        # Test search (if possible)
        print("🔍 Search tool ready for use")
        return search_tool
        
    except Exception as e:
        print(f"❌ Search tool failed: {e}")
        return None

def main():
    """Main test function"""
    print("🚀 SMOLAGENT BASIC SETUP TEST")
    print("=" * 30)
    
    # Test 1: Basic agent
    agent = test_basic_agent()
    
    # Test 2: Search tool
    search_tool = test_with_search()
    
    # Summary
    print("\n📊 TEST SUMMARY")
    print("=" * 15)
    print(f"Basic Agent: {'✅ Working' if agent else '❌ Failed'}")
    print(f"Search Tool: {'✅ Working' if search_tool else '❌ Failed'}")
    
    if agent:
        print("\n💡 Your basic SmoLagent is ready!")
        print("   To add AI capabilities, configure a language model")
        print("   Options: Ollama (free), OpenAI (paid), Hugging Face (free)")
    else:
        print("\n❌ Basic setup failed. Check SmoLagents installation.")
    
    print("\n🚀 Next: Run 'python upload.py' to use enhanced data analysis!")

if __name__ == "__main__":
    main()