File size: 2,168 Bytes
1add76f | 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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 | """
Quick test script for SOCAR LLM API
"""
import requests
import json
from docs.sample_questions import questions
# API base URL
BASE_URL = "http://localhost:8000"
def test_health():
"""Test health endpoint"""
print("🔍 Testing health endpoint...")
response = requests.get(f"{BASE_URL}/health")
print(f"Status: {response.status_code}")
print(json.dumps(response.json(), indent=2))
print()
def test_root():
"""Test root endpoint"""
print("🔍 Testing root endpoint...")
response = requests.get(BASE_URL)
print(f"Status: {response.status_code}")
print(json.dumps(response.json(), indent=2))
print()
def test_llm(question: str):
"""Test LLM endpoint"""
print(f"🔍 Testing LLM endpoint...")
print(f"Question: {question}\n")
payload = {
"messages": [
{"role": "user", "content": question}
],
"temperature": 0.2,
"max_tokens": 1000
}
response = requests.post(f"{BASE_URL}/llm", json=payload)
print(f"Status: {response.status_code}")
if response.status_code == 200:
result = response.json()
print(f"Response time: {result['response_time']}s")
print(f"Model: {result['model']}")
print(f"\nAnswer:\n{result['response']}")
print(f"\nSources:")
for source in result['sources']:
print(f" - {source['pdf_name']}, Page {source['page_number']} (score: {source['relevance_score']})")
else:
print(f"Error: {response.text}")
print()
if __name__ == "__main__":
print("="*80)
print("SOCAR LLM API Test Suite")
print("="*80)
print()
# Test health
try:
test_health()
except Exception as e:
print(f"❌ Health check failed: {e}\n")
# Test root
try:
test_root()
except Exception as e:
print(f"❌ Root endpoint failed: {e}\n")
# Test LLM with sample question
try:
test_llm("Palçıq vulkanlarının təsir radiusu nə qədərdir?")
except Exception as e:
print(f"❌ LLM endpoint failed: {e}\n")
print("="*80)
print("✅ Test suite completed!")
print("="*80)
|