add test for ip_online check
This commit is contained in:
@@ -45,7 +45,6 @@ class Provider:
|
||||
self.api_key = self.get_api_key(self.provider_name)
|
||||
elif self.provider_name != "ollama":
|
||||
pretty_print(f"Provider: {provider_name} initialized at {self.server_ip}", color="success")
|
||||
print("IP online?", self.is_ip_online(self.server_ip.split(":")[0]))
|
||||
|
||||
def get_api_key(self, provider):
|
||||
load_dotenv()
|
||||
@@ -85,11 +84,11 @@ class Provider:
|
||||
"""
|
||||
if not address:
|
||||
return False
|
||||
if address.lower() in ["127.0.0.1", "localhost", "0.0.0.0"]:
|
||||
return True
|
||||
parsed = urlparse(address if address.startswith(('http://', 'https://')) else f'http://{address}')
|
||||
|
||||
hostname = parsed.hostname or address
|
||||
|
||||
if "127.0.0.1" in address or "localhost" in address:
|
||||
return True
|
||||
try:
|
||||
ip_address = socket.gethostbyname(hostname)
|
||||
except socket.gaierror:
|
||||
|
||||
Reference in New Issue
Block a user