add test for ip_online check

This commit is contained in:
martin legrand
2025-04-26 10:27:17 +02:00
parent dce9074969
commit e4ae8162a0
2 changed files with 73 additions and 4 deletions
+3 -4
View File
@@ -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: