Spaces:
Sleeping
Sleeping
| #!/usr/bin/env python | |
| """ | |
| Test script to demonstrate dynamic responses from Gemma-2B | |
| Shows how the same query gets different responses based on context | |
| """ | |
| import sys | |
| import os | |
| from datetime import datetime, timedelta | |
| sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) | |
| from software_bug_assistant.hybrid_agent import HybridAgent, LocalGemmaModel | |
| def test_dynamic_responses(): | |
| print("=" * 70) | |
| print("Testing Dynamic Responses from Gemma-2B") | |
| print("=" * 70) | |
| print() | |
| # Initialize model | |
| print("Loading Gemma-2B...") | |
| model = LocalGemmaModel() | |
| if not model.is_available(): | |
| print("❌ Model not available. Run: python download_gemma.py") | |
| return | |
| print("✓ Model loaded!\n") | |
| # Test 1: Same query with different contexts | |
| print("Test 1: Same Query, Different Contexts") | |
| print("-" * 70) | |
| query = "What time is it?" | |
| # Morning context | |
| morning_time = datetime.now().replace(hour=9, minute=30) | |
| context1 = {"current_time": morning_time.strftime("%Y-%m-%d %H:%M:%S")} | |
| # Evening context | |
| evening_time = datetime.now().replace(hour=21, minute=15) | |
| context2 = {"current_time": evening_time.strftime("%Y-%m-%d %H:%M:%S")} | |
| print(f"Query: '{query}'") | |
| print(f"\nContext 1 (Morning): {context1['current_time']}") | |
| response1 = model.generate(query, context=context1) | |
| print(f"Response: {response1}") | |
| print(f"\nContext 2 (Evening): {context2['current_time']}") | |
| response2 = model.generate(query, context=context2) | |
| print(f"Response: {response2}") | |
| print("\n" + "=" * 70) | |
| print("Test 2: Multiple Responses to Same Query (Variety)") | |
| print("-" * 70) | |
| query = "Hello!" | |
| print(f"Query: '{query}'") | |
| print("\nGenerating 3 different responses:") | |
| for i in range(3): | |
| response = model.generate(query, max_tokens=100) | |
| print(f"\nResponse {i+1}: {response}") | |
| print("\n" + "=" * 70) | |
| print("Test 3: Context-Aware Responses") | |
| print("-" * 70) | |
| # With user name | |
| context_with_name = { | |
| "current_time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), | |
| "user_name": "Alice" | |
| } | |
| query = "How are you?" | |
| print(f"Query: '{query}'") | |
| print(f"Context: User name = {context_with_name['user_name']}") | |
| response = model.generate(query, context=context_with_name) | |
| print(f"Response: {response}") | |
| print("\n" + "=" * 70) | |
| print("Test 4: Math Queries (Dynamic Calculations)") | |
| print("-" * 70) | |
| math_queries = [ | |
| "What is 15 + 27?", | |
| "Calculate 100 * 5", | |
| "What's 144 divided by 12?", | |
| ] | |
| for query in math_queries: | |
| response = model.generate(query) | |
| print(f"Query: {query}") | |
| print(f"Response: {response}\n") | |
| print("=" * 70) | |
| print("✓ Dynamic response testing complete!") | |
| print("=" * 70) | |
| if __name__ == "__main__": | |
| test_dynamic_responses() | |