Spaces:
Runtime error
Runtime error
File size: 1,789 Bytes
1ca83c3 |
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 |
import sys
import os
import requests
# Ensure we can import from parent directory
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
try:
from logos.connectors import get_connector
except ImportError:
# If running from hf_space root, try direct import
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
from logos.connectors import get_connector
def test_local_connection(port=1234):
print(f"--- Telemetery Probe: Localhost:{port} ---")
# 1. Initialize Connector
try:
url = f"http://localhost:{port}/v1"
print(f"Connecting to {url}...")
connector = get_connector('local', base_url=url, model="local-model")
except Exception as e:
print(f"[FAIL] Factory Error: {e}")
return
# 2. Send Ping
prompt = "Reply with exactly one word: 'CONNECTED'."
print(f"Sending Probe: \"{prompt}\"")
try:
response = connector.chat(prompt)
print(f"\n[RESPONSE] >> {response}")
if "CONNECTED" in response.upper():
print("\n[SUCCESS] Link Established. The Local Cluster is Online.")
else:
print("\n[WARN] Link Active, but response was unexpected. (Model might be chatty)")
except Exception as e:
print(f"\n[FAIL] Connection Refused. {e}")
print("troubleshooting:")
print("1. Is LM Studio Server ON? (Green light?)")
print("2. Is Port correct? (Default 1234)")
print("3. Is CORS enabled in LM Studio?")
if __name__ == "__main__":
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--port", type=int, default=1234, help="LM Studio/Ollama Port")
args = parser.parse_args()
test_local_connection(args.port)
|