Fixed agent

app.py

@@ -256,4 +256,4 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for LangGraph Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=True)

example_usage.py

@@ -1,14 +0,0 @@
-# example_usage.py
-from graph.graph_builder import graph
-from langchain_core.messages import HumanMessage
-
-# Simple example
-def ask_agent(question: str):
-    initial_state = {"messages": [HumanMessage(content=question)]}
-    result = graph.invoke(initial_state)
-    return result["messages"][-1].content
-
-# Usage
-if __name__ == "__main__":
-    response = ask_agent("How many albums did Michael Jackson release in his lifetime?")
-    print(f"Agent: {response}")

local_test.py

@@ -1,17 +0,0 @@
-# local_test.py
-from app import BasicAgent
-
-agent = BasicAgent()
-
-# Test questions
-test_questions = [
-    "What is 15 + 27?",
-    "Calculate the square root of 144",
-    "What is 25% of 200?"
-]
-
-for q in test_questions:
-    answer = agent(q)
-    print(f"Q: {q}")
-    print(f"A: {answer}")
-    print("-" * 30)

nodes/core.py

@@ -31,7 +31,7 @@ if not openrouter_api_key:
 
 # Initialize OpenRouter ChatOpenAI with OpenRouter-specific configuration
 chat = ChatOpenAI(
-    model="google/gemini-2.5-flash-preview-05-20",  # Free multimodal model
+    model="google/gemini-2.5-pro-preview",  # Free multimodal model
     # Alternative models you can use:
     # model="mistralai/mistral-7b-instruct:free",  # Fast, free text model
     # model="google/gemma-2-9b-it:free",  # Google's free model

test_agent.py

@@ -1,50 +0,0 @@
-# test_agent.py
-import os
-from dotenv import load_dotenv
-from graph.graph_builder import graph
-from langchain_core.messages import HumanMessage
-
-load_dotenv()
-
-def test_agent():
-    """Test the LangGraph agent with various queries"""
-    
-    # Check if OpenRouter API key is available
-    if not os.getenv("OPENROUTER_API_KEY"):
-        print("❌ OPENROUTER_API_KEY not found in environment")
-        return
-    
-    print("🤖 Testing LangGraph Agent with OpenRouter")
-    print("=" * 50)
-    
-    # Test cases
-    test_queries = [
-        "What is 15 + 27?",
-        "Calculate the square root of 144",
-        "Search for information about artificial intelligence",
-        "What is 25% of 200?"
-    ]
-    
-    for i, query in enumerate(test_queries, 1):
-        print(f"\n🧪 Test {i}: {query}")
-        print("-" * 30)
-        
-        try:
-            # Create initial state
-            initial_state = {"messages": [HumanMessage(content=query)]}
-            
-            # Run the agent
-            result = graph.invoke(initial_state)
-            
-            # Get the final message
-            final_message = result["messages"][-1]
-            print(f"🤖 Response: {final_message.content}")
-            
-        except Exception as e:
-            print(f"❌ Error: {str(e)}")
-    
-    print("\n" + "=" * 50)
-    print("✅ Agent testing completed!")
-
-if __name__ == "__main__":
-    test_agent()

test_all_tools.py

@@ -1,63 +0,0 @@
-# test_all_tools.py
-import os
-import sys
-from dotenv import load_dotenv
-from pathlib import Path
-
-# Load environment variables
-load_dotenv()
-
-# Add tools directory to path
-sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
-
-def main():
-    """Run all tool tests"""
-    print("=" * 60)
-    print("🧪 TESTING ALL TOOLS")
-    print("=" * 60)
-    
-    # Check if OpenRouter API key is loaded
-    openrouter_key = os.getenv("OPENROUTER_API_KEY")
-    if openrouter_key:
-        print(f"✅ OpenRouter API Key loaded: {openrouter_key[:8]}...")
-    else:
-        print("❌ OpenRouter API Key not found in environment")
-        print("Please add OPENROUTER_API_KEY to your .env file")
-    
-    # Import and run individual test modules
-    try:
-        print("\n🔧 Testing Math Tools...")
-        from test_math_tools import test_math_tools
-        test_math_tools()
-        
-        print("\n🔍 Testing Search Tools...")
-        from test_search_tools import test_search_tools
-        test_search_tools()
-        
-        print("\n🎨 Testing Multimodal Tools...")
-        from test_multimodal_tools import test_multimodal_tools
-        test_multimodal_tools()
-        
-        print("\n📺 Testing YouTube Tools...")
-        from test_youtube_tools import test_youtube_tools
-        test_youtube_tools()
-        
-        print("\n" + "=" * 60)
-        print("✅ ALL TESTS COMPLETED SUCCESSFULLY!")
-        print("=" * 60)
-        
-        # Usage examples
-        print("\n📚 Quick Usage Examples:")
-        print("from tools import get_video_info, search_web, add, analyze_image")
-        print("video_info = get_video_info('https://youtube.com/watch?v=...')")
-        print("search_results = search_web('Python programming')")
-        print("result = add(10, 5)")
-        print("image_desc = analyze_image('photo.jpg', 'What is in this image?')")
-        
-    except Exception as e:
-        print(f"\n❌ Error running tests: {str(e)}")
-        import traceback
-        traceback.print_exc()
-
-if __name__ == "__main__":
-    main()

test_math_tools.py

@@ -1,78 +0,0 @@
-# test_math_tools.py
-import sys
-import os
-from dotenv import load_dotenv
-
-# Load environment variables
-load_dotenv()
-
-# Add tools to path
-sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
-
-def test_math_tools():
-    """Test all math tool functions"""
-    try:
-        from tools.math_tools import MathTools, add, subtract, multiply, divide, factorial, square_root, average, calculate_expression
-        
-        print("  📊 Testing basic arithmetic...")
-        
-        # Test basic operations
-        assert add(5, 3) == 8, "Addition failed"
-        assert subtract(10, 4) == 6, "Subtraction failed"
-        assert multiply(6, 7) == 42, "Multiplication failed"
-        assert divide(15, 3) == 5, "Division failed"
-        print("    ✅ Basic arithmetic: PASSED")
-        
-        # Test advanced operations
-        print("  🧮 Testing advanced operations...")
-        assert factorial(5) == 120, "Factorial failed"
-        assert square_root(16) == 4.0, "Square root failed"
-        assert average([1, 2, 3, 4, 5]) == 3.0, "Average failed"
-        print("    ✅ Advanced operations: PASSED")
-        
-        # Test error handling
-        print("  🚨 Testing error handling...")
-        assert "Error" in str(divide(5, 0)), "Division by zero should return error"
-        assert "Error" in str(factorial(-1)), "Negative factorial should return error"
-        assert "Error" in str(square_root(-1)), "Negative square root should return error"
-        print("    ✅ Error handling: PASSED")
-        
-        # Test expression calculator
-        print("  🧠 Testing expression calculator...")
-        assert calculate_expression("2 + 3 * 4") == 14, "Expression calculation failed"
-        assert calculate_expression("2^3") == 8, "Power expression failed"
-        print("    ✅ Expression calculator: PASSED")
-        
-        # Test quadratic solver
-        print("  📐 Testing quadratic solver...")
-        solutions = MathTools.solve_quadratic(1, -5, 6)
-        assert isinstance(solutions, tuple) and len(solutions) == 2, "Quadratic solver failed"
-        print("    ✅ Quadratic solver: PASSED")
-        
-        # Test practical examples
-        print("  💰 Testing practical calculations...")
-        compound_interest = MathTools.calculate_compound_interest(1000, 0.05, 2)
-        assert compound_interest > 1000, "Compound interest should be greater than principal"
-        
-        percentage_result = MathTools.percentage(25, 100)
-        assert percentage_result == 25.0, "Percentage calculation failed"
-        print("    ✅ Practical calculations: PASSED")
-        
-        print("  ✅ ALL MATH TESTS PASSED!")
-        
-        # Demo output
-        print("\n  📋 Math Tools Demo:")
-        print(f"    5 + 3 = {add(5, 3)}")
-        print(f"    10! = {factorial(10)}")
-        print(f"    √144 = {square_root(144)}")
-        print(f"    Average of [10,20,30] = {average([10,20,30])}")
-        print(f"    Expression '2^3 + 4*5' = {calculate_expression('2^3 + 4*5')}")
-        
-    except ImportError as e:
-        print(f"    ❌ Import error: {e}")
-    except Exception as e:
-        print(f"    ❌ Test failed: {e}")
-        raise
-
-if __name__ == "__main__":
-    test_math_tools()

test_multimodal_tools.py

@@ -1,145 +0,0 @@
-# test_multimodal_tools.py
-import sys
-import os
-from dotenv import load_dotenv
-import tempfile
-import base64
-
-# Load environment variables
-load_dotenv()
-
-# Add tools to path
-sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
-
-def create_test_image():
-    """Create a simple test image for testing"""
-    try:
-        from PIL import Image, ImageDraw
-        
-        # Create a simple test image
-        img = Image.new('RGB', (200, 100), color='white')
-        draw = ImageDraw.Draw(img)
-        draw.text((10, 40), "Test Image", fill='black')
-        
-        # Save to temporary file
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
-        img.save(temp_file.name)
-        return temp_file.name
-    except ImportError:
-        print("    ⚠️  PIL not available, creating mock image file")
-        # Create a mock image file for testing
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.txt')
-        temp_file.write(b"This is a mock image file for testing")
-        temp_file.close()
-        return temp_file.name
-
-def test_multimodal_tools():
-    """Test multimodal tool functions"""
-    try:
-        from tools.multimodal_tools import MultimodalTools, analyze_image, extract_text, analyze_transcript
-        
-        # Check OpenRouter API key
-        openrouter_key = os.getenv("OPENROUTER_API_KEY")
-        if not openrouter_key:
-            print("    ❌ OpenRouter API key not found!")
-            print("    Please add OPENROUTER_API_KEY to your .env file")
-            return
-        
-        print("  🎨 Initializing multimodal tools...")
-        multimodal_tools = MultimodalTools()
-        print(f"    ✅ API key loaded: {openrouter_key[:8]}...")
-        
-        # Test with transcript (doesn't require image)
-        print("  🎙️  Testing audio transcript analysis...")
-        test_transcript = "Hello, this is a test transcript. We're discussing artificial intelligence and machine learning technologies."
-        
-        transcript_result = multimodal_tools.analyze_audio_transcript(
-            test_transcript, 
-            "What is the main topic of this transcript?"
-        )
-        
-        if transcript_result and "Error" not in transcript_result:
-            print("    ✅ Transcript analysis: PASSED")
-            print(f"    Result: {transcript_result[:100]}...")
-        else:
-            print(f"    ❌ Transcript analysis failed: {transcript_result}")
-        
-        # Test convenience function
-        print("  🔧 Testing convenience functions...")
-        convenience_result = analyze_transcript(
-            "This is a test about Python programming and software development.",
-            "Summarize the main topics"
-        )
-        
-        if convenience_result and "Error" not in convenience_result:
-            print("    ✅ Convenience function: PASSED")
-        else:
-            print(f"    ⚠️  Convenience function result: {convenience_result}")
-        
-        # Test image analysis (if we can create a test image)
-        print("  🖼️  Testing image analysis...")
-        try:
-            test_image_path = create_test_image()
-            
-            # Test image description
-            image_result = multimodal_tools.analyze_image(
-                test_image_path,
-                "Describe what you see in this image"
-            )
-            
-            if image_result and "Error" not in image_result:
-                print("    ✅ Image analysis: PASSED")
-                print(f"    Result: {image_result[:100]}...")
-            else:
-                print(f"    ⚠️  Image analysis result: {image_result}")
-            
-            # Test OCR functionality
-            ocr_result = multimodal_tools.extract_text_from_image(test_image_path)
-            if ocr_result and "Error" not in ocr_result:
-                print("    ✅ OCR extraction: PASSED")
-                print(f"    Extracted: {ocr_result[:50]}...")
-            else:
-                print(f"    ⚠️  OCR result: {ocr_result}")
-            
-            # Clean up test image
-            os.unlink(test_image_path)
-            
-        except Exception as e:
-            print(f"    ⚠️  Image testing skipped: {str(e)}")
-        
-        # Test error handling
-        print("  🚨 Testing error handling...")
-        error_result = multimodal_tools.analyze_image(
-            "nonexistent_file.jpg",
-            "This should fail"
-        )
-        
-        if "Error" in error_result:
-            print("    ✅ Error handling: PASSED")
-        else:
-            print("    ⚠️  Error handling unexpected result")
-        
-        # Test empty transcript
-        empty_result = multimodal_tools.analyze_audio_transcript("", "Question")
-        if "Error" in empty_result:
-            print("    ✅ Empty input handling: PASSED")
-        
-        print("  ✅ MULTIMODAL TESTS COMPLETED!")
-        
-        # Demo output
-        print("\n  ���� Multimodal Tools Demo:")
-        demo_transcript = "We discussed the implementation of AI agents using CrewAI framework for automated workflows."
-        demo_result = analyze_transcript(demo_transcript, "What framework was mentioned?")
-        print(f"    Question: What framework was mentioned?")
-        print(f"    Answer: {demo_result[:150]}...")
-        
-    except ImportError as e:
-        print(f"    ❌ Import error: {e}")
-        print("    Make sure all dependencies are installed")
-    except Exception as e:
-        print(f"    ❌ Test failed: {e}")
-        import traceback
-        traceback.print_exc()
-
-if __name__ == "__main__":
-    test_multimodal_tools()

test_search_tools.py

@@ -1,96 +0,0 @@
-# test_search_tools.py
-import sys
-import os
-from dotenv import load_dotenv
-
-# Load environment variables
-load_dotenv()
-
-# Add tools to path
-sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
-
-def test_search_tools():
-    """Test search tool functions"""
-    try:
-        from tools.search_tools import SearchTools, search_web, search_news
-        
-        print("  🔍 Initializing search tools...")
-        search_tools = SearchTools()
-        
-        # Test DuckDuckGo search (free)
-        print("  🦆 Testing DuckDuckGo search...")
-        ddg_results = search_tools.search_duckduckgo("Python programming", max_results=3)
-        if ddg_results:
-            print(f"    ✅ DuckDuckGo returned {len(ddg_results)} results")
-            for i, result in enumerate(ddg_results[:2]):
-                print(f"      {i+1}. {result.get('title', 'No title')[:50]}...")
-        else:
-            print("    ⚠️  DuckDuckGo returned no results")
-        
-        # Test Tavily search (if API key available)
-        tavily_key = os.getenv("TAVILY_API_KEY")
-        if tavily_key:
-            print("  🔎 Testing Tavily search...")
-            tavily_results = search_tools.search_tavily("AI news 2025", max_results=3)
-            if tavily_results:
-                print(f"    ✅ Tavily returned {len(tavily_results)} results")
-            else:
-                print("    ⚠️  Tavily returned no results")
-        else:
-            print("  ⚠️  Tavily API key not found, skipping Tavily tests")
-        
-        # Test SerpAPI search (if API key available)
-        serpapi_key = os.getenv("SERPAPI_KEY")
-        if serpapi_key:
-            print("  🐍 Testing SerpAPI search...")
-            serp_results = search_tools.search_serpapi("machine learning", max_results=2)
-            if serp_results:
-                print(f"    ✅ SerpAPI returned {len(serp_results)} results")
-            else:
-                print("    ⚠️  SerpAPI returned no results")
-        else:
-            print("  ⚠️  SerpAPI key not found, skipping SerpAPI tests")
-        
-        # Test main search function (with fallback)
-        print("  🎯 Testing main search function...")
-        main_results = search_tools.search("CrewAI framework", max_results=3)
-        if main_results:
-            print(f"    ✅ Main search returned {len(main_results)} results")
-            print(f"    Provider used: {main_results[0].get('source', 'Unknown')}")
-        else:
-            print("    ❌ Main search failed")
-        
-        # Test convenience functions
-        print("  🚀 Testing convenience functions...")
-        web_results = search_web("OpenRouter API", max_results=2)
-        news_results = search_news("artificial intelligence", max_results=2)
-        
-        if web_results:
-            print(f"    ✅ Web search returned {len(web_results)} results")
-        if news_results:
-            print(f"    ✅ News search returned {len(news_results)} results")
-        
-        # Test specialized searches
-        print("  🎓 Testing specialized searches...")
-        academic_results = search_tools.search_academic("machine learning algorithms", max_results=2)
-        if academic_results:
-            print(f"    ✅ Academic search returned {len(academic_results)} results")
-        
-        print("  ✅ ALL SEARCH TESTS COMPLETED!")
-        
-        # Demo output
-        if main_results:
-            print("\n  📋 Search Results Demo:")
-            for i, result in enumerate(main_results[:2]):
-                title = result.get('title', 'No title')[:60]
-                source = result.get('source', 'Unknown')
-                print(f"    {i+1}. [{source}] {title}...")
-        
-    except ImportError as e:
-        print(f"    ❌ Import error: {e}")
-    except Exception as e:
-        print(f"    ❌ Test failed: {e}")
-        print(f"    Error details: {str(e)}")
-
-if __name__ == "__main__":
-    test_search_tools()

test_youtube_tools.py

@@ -1,111 +0,0 @@
-# test_youtube_tools.py (Updated with working test URLs)
-import sys
-import os
-import tempfile
-import shutil
-from dotenv import load_dotenv
-
-load_dotenv()
-sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
-
-def test_youtube_tools():
-    """Test YouTube tool functions with reliable test URLs"""
-    
-    # Updated test URLs - using more reliable, public content
-    test_video_url = "https://www.youtube.com/watch?v=dQw4w9WgXcQ"  # Rick Roll
-    # Use a known public playlist
-    test_playlist_url = "https://www.youtube.com/playlist?list=PLNVhQQnz1F5f11NXuOKF0oULKWWM2HfZS"  # Small public playlist
-    
-    try:
-        from tools.youtube_tools import YouTubeTools, get_video_info, download_video, get_captions
-        
-        print("  📺 Initializing YouTube tools...")
-        youtube_tools = YouTubeTools(max_retries=2, retry_delay=1.0)  # Configure retry settings
-        
-        # Test video info retrieval
-        print("  📋 Testing video info retrieval...")
-        video_info = youtube_tools.get_video_info(test_video_url)
-        
-        if video_info:
-            print("    ✅ Video info retrieval: PASSED")
-            print(f"    Title: {video_info.get('title', 'Unknown')[:50]}...")
-            print(f"    Author: {video_info.get('author', 'Unknown')}")
-            print(f"    Duration: {video_info.get('length', 0)} seconds")
-            print(f"    Views: {video_info.get('views', 0):,}")
-            print(f"    Available captions: {video_info.get('captions_available', [])}")
-        else:
-            print("    ❌ Video info retrieval failed")
-        
-        # Test available qualities with retry
-        print("  🎥 Testing available qualities...")
-        qualities = youtube_tools.get_available_qualities(test_video_url)
-        
-        if qualities:
-            print("    ✅ Quality detection: PASSED")
-            resolutions = [q['resolution'] for q in qualities if q['resolution'] != 'unknown'][:3]
-            print(f"    Available resolutions: {resolutions}")
-        else:
-            print("    ⚠️  Quality detection failed (network issue)")
-        
-        # Test captions (should work without deprecation warning)
-        print("  📝 Testing caption retrieval...")
-        captions = youtube_tools.get_captions(test_video_url, 'en')
-        
-        if captions:
-            print("    ✅ Caption retrieval: PASSED")
-            print(f"    Caption preview: {captions[:100]}...")
-        elif captions is None:
-            print("    ⚠️  No captions available for this video")
-        else:
-            print("    ❌ Caption retrieval failed")
-        
-        # Test playlist info with better URL
-        print("  📂 Testing playlist info...")
-        try:
-            playlist_info = youtube_tools.get_playlist_info(test_playlist_url)
-            
-            if playlist_info:
-                print("    ✅ Playlist info: PASSED")
-                print(f"    Playlist: {playlist_info.get('title', 'Unknown')[:50]}...")
-                print(f"    Videos: {playlist_info.get('video_count', 0)}")
-            else:
-                print("    ⚠️  Playlist info failed")
-        except Exception as e:
-            print(f"    ⚠️  Playlist test error: {str(e)}")
-        
-        # Test convenience functions
-        print("  🔧 Testing convenience functions...")
-        convenience_info = get_video_info(test_video_url)
-        if convenience_info:
-            print("    ✅ Convenience functions: PASSED")
-        
-        # Test error handling
-        print("  🚨 Testing error handling...")
-        invalid_info = youtube_tools.get_video_info("https://invalid-url")
-        if invalid_info is None:
-            print("    ✅ Error handling: PASSED")
-        
-        print("  ✅ YOUTUBE TOOLS TESTS COMPLETED!")
-        
-        # Demo output
-        if video_info:
-            print("\n  📋 YouTube Tools Demo:")
-            print(f"    Video: {video_info.get('title', 'Unknown')}")
-            print(f"    Channel: {video_info.get('author', 'Unknown')}")
-            print(f"    Published: {video_info.get('publish_date', 'Unknown')}")
-            print(f"    Length: {video_info.get('length', 0)} seconds")
-            
-            if qualities:
-                best_quality = qualities[0] if qualities else {}
-                print(f"    Best quality: {best_quality.get('resolution', 'unknown')}")
-        
-    except ImportError as e:
-        print(f"    ❌ Import error: {e}")
-        print("    Make sure pytubefix is installed: pip install pytubefix")
-    except Exception as e:
-        print(f"    ❌ Test failed: {e}")
-        import traceback
-        traceback.print_exc()
-
-if __name__ == "__main__":
-    test_youtube_tools()

tools/multimodal_tools.py

@@ -18,8 +18,8 @@ class MultimodalTools:
         }
         
         # Available free multimodal models
-        self.vision_model = "moonshotai/kimi-vl-a3b-thinking:free"
-        self.text_model = "meta-llama/llama-4-maverick:free"
+        self.vision_model = "google/gemini-2.5-flash-preview-05-20"
+        self.text_model = "google/gemini-2.5-flash-preview-05-20"
     
     def _make_openrouter_request(self, payload: Dict[str, Any]) -> str:
         """Make request to OpenRouter API with error handling"""