Research-Stack/5-Applications/scripts/test_knowledge_ingestion.py

45 lines
1.4 KiB
Python

#!/usr/bin/env python3
"""
Test script for knowledge ingestion with Wolfram Alpha API
"""
import sys
import os
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '.')))
from infra.knowledge_ingestion import KnowledgeIngestion
def test_wolfram_alpha():
"""Test Wolfram Alpha API integration"""
api_key = "HYJE3R3R63"
print("Testing Wolfram Alpha API integration...")
ingestion = KnowledgeIngestion(api_key)
# Test a simple query
print("\n1. Testing simple query: 'What is 2+2?'")
result = ingestion.wolfram.query("What is 2+2?")
if result:
print(f"Result: {result}")
else:
print("Query failed")
# Test domain knowledge retrieval
print("\n2. Testing domain knowledge: 'mathematics'")
domain_knowledge = ingestion.wolfram.get_domain_knowledge("mathematics")
print(f"Domain knowledge: {domain_knowledge}")
# Test OpenMath ingestion
print("\n3. Testing OpenMath Content Dictionary ingestion...")
openmath_result = ingestion.openmath.fetch_content_dictionary("arith1")
print(f"OpenMath result: {openmath_result}")
# Test nLab scraping
print("\n4. Testing nLab wiki scraping...")
nlab_result = ingestion.nlab.scrape_page("topological_space")
print(f"nLab result: {nlab_result}")
print("\nAll tests complete!")
if __name__ == "__main__":
test_wolfram_alpha()