Spaces:
Runtime error
Runtime error
| """ | |
| Basic test script for ConversAI modules | |
| Run this to verify core functionality | |
| """ | |
| import json | |
| from llm_backend import LLMBackend, LLMProvider | |
| from survey_generator import SurveyGenerator | |
| from survey_translator import SurveyTranslator | |
| from data_analyzer import DataAnalyzer | |
| def test_llm_backend(): | |
| """Test LLM backend initialization""" | |
| print("\n=== Testing LLM Backend ===") | |
| try: | |
| backend = LLMBackend(provider=LLMProvider.LM_STUDIO) | |
| print(f"✓ Backend initialized with provider: {backend.provider}") | |
| print(f"✓ Model: {backend.model}") | |
| print(f"✓ API URL: {backend.api_url}") | |
| return backend | |
| except Exception as e: | |
| print(f"✗ Backend initialization failed: {e}") | |
| return None | |
| def test_survey_generator(backend): | |
| """Test survey generation""" | |
| print("\n=== Testing Survey Generator ===") | |
| if not backend: | |
| print("✗ Skipping (no backend)") | |
| return None | |
| try: | |
| gen = SurveyGenerator(backend) | |
| print("✓ Survey generator initialized") | |
| # Note: Actual generation requires LLM connection | |
| print(" (Actual survey generation requires LLM connection)") | |
| return gen | |
| except Exception as e: | |
| print(f"✗ Survey generator failed: {e}") | |
| return None | |
| def test_survey_translator(backend): | |
| """Test survey translator""" | |
| print("\n=== Testing Survey Translator ===") | |
| if not backend: | |
| print("✗ Skipping (no backend)") | |
| return None | |
| try: | |
| translator = SurveyTranslator(backend) | |
| print("✓ Translator initialized") | |
| # Test language list | |
| langs = translator.get_supported_languages() | |
| print(f"✓ Supports {len(langs)} languages") | |
| print(f" Sample languages: {', '.join(list(langs.values())[:5])}") | |
| return translator | |
| except Exception as e: | |
| print(f"✗ Translator failed: {e}") | |
| return None | |
| def test_data_analyzer(backend): | |
| """Test data analyzer""" | |
| print("\n=== Testing Data Analyzer ===") | |
| if not backend: | |
| print("✗ Skipping (no backend)") | |
| return None | |
| try: | |
| analyzer = DataAnalyzer(backend) | |
| print("✓ Analyzer initialized") | |
| # Test with sample data | |
| sample_responses = [ | |
| {"q1": "I had a great experience", "q2": "Very satisfied"}, | |
| {"q1": "It was okay", "q2": "Neutral feelings"}, | |
| {"q1": "Not very good", "q2": "Disappointed"} | |
| ] | |
| # Note: Actual analysis requires LLM connection | |
| print(" (Actual analysis requires LLM connection)") | |
| return analyzer | |
| except Exception as e: | |
| print(f"✗ Analyzer failed: {e}") | |
| return None | |
| def test_modules(): | |
| """Test all modules""" | |
| print("="*50) | |
| print("ConversAI Module Tests") | |
| print("="*50) | |
| # Test backend | |
| backend = test_llm_backend() | |
| # Test generators | |
| gen = test_survey_generator(backend) | |
| translator = test_survey_translator(backend) | |
| analyzer = test_data_analyzer(backend) | |
| # Summary | |
| print("\n=== Test Summary ===") | |
| modules = { | |
| "LLM Backend": backend is not None, | |
| "Survey Generator": gen is not None, | |
| "Survey Translator": translator is not None, | |
| "Data Analyzer": analyzer is not None | |
| } | |
| for module, status in modules.items(): | |
| symbol = "✓" if status else "✗" | |
| print(f"{symbol} {module}") | |
| all_passed = all(modules.values()) | |
| print(f"\n{'✓ All tests passed!' if all_passed else '✗ Some tests failed'}") | |
| if not all_passed: | |
| print("\nNote: Make sure your LLM backend is configured correctly.") | |
| print("Check environment variables or .env file.") | |
| return all_passed | |
| if __name__ == "__main__": | |
| test_modules() | |