100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
import requests
|
|
import json
|
|
import time
|
|
|
|
def test_deepseek_search_direct():
|
|
"""Test DeepSeek search functionality directly with authentication"""
|
|
|
|
base_url = "http://localhost:3015"
|
|
|
|
print("=== DeepSeek API Search Test ===\n")
|
|
|
|
# First, let's try to login with common credentials
|
|
test_credentials = [
|
|
{"username": "jleu3482", "password": "jleu1212"},
|
|
{"username": "admin", "password": "admin"},
|
|
{"username": "user", "password": "password"},
|
|
{"username": "test", "password": "test"},
|
|
]
|
|
|
|
headers = None
|
|
|
|
for creds in test_credentials:
|
|
print(f"Trying login with: {creds['username']}")
|
|
login_data = {
|
|
"username": creds["username"],
|
|
"password": creds["password"]
|
|
}
|
|
|
|
login_response = requests.post(
|
|
f"{base_url}/login",
|
|
data=login_data,
|
|
headers={"Content-Type": "application/x-www-form-urlencoded"}
|
|
)
|
|
|
|
if login_response.status_code == 200:
|
|
token = login_response.json().get("access_token")
|
|
headers = {"Authorization": f"Bearer {token}"}
|
|
print(f"✅ Login successful with: {creds['username']}")
|
|
break
|
|
else:
|
|
print(f"❌ Login failed: {login_response.status_code}")
|
|
|
|
if not headers:
|
|
print("\n❌ Could not authenticate. Please check server authentication configuration.")
|
|
return
|
|
|
|
# Test search with authentication
|
|
print("\n=== Testing Search with Authentication ===")
|
|
search_data = {
|
|
"query": "optical character recognition",
|
|
"top_k": 3
|
|
}
|
|
|
|
try:
|
|
search_response = requests.post(f"{base_url}/search", json=search_data, headers=headers)
|
|
print(f"Search response status: {search_response.status_code}")
|
|
|
|
if search_response.status_code == 200:
|
|
search_results = search_response.json()
|
|
print("✅ Search successful!")
|
|
print(f"Found {len(search_results.get('results', []))} results")
|
|
|
|
for i, result in enumerate(search_results.get('results', [])):
|
|
print(f"\nResult {i+1}:")
|
|
print(f" Score: {result.get('score', 0):.4f}")
|
|
print(f" Content: {result.get('content', '')[:200]}...")
|
|
print(f" Source: {result.get('source', '')}")
|
|
|
|
else:
|
|
print(f"Search error: {search_response.text}")
|
|
|
|
# Check if it's the DeepSeek regional error
|
|
if "unsupported_country_region_territory" in search_response.text:
|
|
print("\n❌ DEEPSEEK REGIONAL RESTRICTION ERROR STILL EXISTS")
|
|
print("This indicates the root cause is NOT resolved")
|
|
|
|
except Exception as e:
|
|
print(f"Search request failed: {e}")
|
|
|
|
# Test a simple query to see if DeepSeek API is working
|
|
print("\n=== Testing Simple Query ===")
|
|
simple_query = {
|
|
"query": "test",
|
|
"top_k": 1
|
|
}
|
|
|
|
try:
|
|
simple_response = requests.post(f"{base_url}/search", json=simple_query, headers=headers)
|
|
print(f"Simple query status: {simple_response.status_code}")
|
|
|
|
if simple_response.status_code == 200:
|
|
print("✅ Simple query successful - DeepSeek API is working!")
|
|
else:
|
|
print(f"Simple query error: {simple_response.text}")
|
|
|
|
except Exception as e:
|
|
print(f"Simple query failed: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
test_deepseek_search_direct() |