File size: 1,310 Bytes
bc197a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
#!/usr/bin/env python3
"""Test API mode with documents."""
import os
import sys
# Set environment
os.environ['LLM_PROVIDER'] = 'api'
os.environ['HF_API_BASE_URL'] = 'https://davidtran999-hue-portal-backend.hf.space/api'
# Add path
sys.path.insert(0, 'hue_portal')
# Setup Django
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'hue_portal.settings')
import django
django.setup()
from hue_portal.chatbot.llm_integration import get_llm_generator
from hue_portal.core.models import Fine
# Get LLM
llm = get_llm_generator()
print(f"✅ LLM Provider: {llm.provider}")
print(f"✅ API URL: {llm.api_base_url}")
print(f"✅ Available: {llm.is_available()}\n")
# Get some documents
fines = Fine.objects.all()[:3]
print(f"📄 Found {len(fines)} documents\n")
# Test with documents
query = "Mức phạt vượt đèn đỏ là bao nhiêu?"
print(f"❓ Query: {query}\n")
# Build prompt
prompt = llm._build_prompt(query, None, list(fines))
print(f"📝 Prompt length: {len(prompt)} chars")
print(f"📝 Prompt preview:\n{prompt[:500]}...\n")
# Test API call
print("🔗 Calling HF Spaces API...\n")
result = llm._generate_api(prompt, None)
if result:
print(f"✅ Success! Response length: {len(result)}")
print(f"📥 Response:\n{result[:500]}...\n")
else:
print("❌ No response from API\n")
|