tcsenpai/agenticSeek
1
0
Fork 0
mirror of https://github.com/tcsenpai/agenticSeek.git synced 2025-06-05 02:25:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix : is_ip_online function

Browse Source
This commit is contained in:
martin legrand 2025-04-26 10:06:20 +02:00
parent b7da34ff93
commit dce9074969
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
sources/llm_provider.py
View File

@ -45,6 +45,7 @@ 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()
@ -86,7 +87,9 @@ class Provider:
return False
if address.lower() in ["127.0.0.1", "localhost", "0.0.0.0"]:
return True
hostname = urlparse(f'http://{address}' if not address.startswith(('http://', 'https://')) else address).hostname or address
parsed = urlparse(address if address.startswith(('http://', 'https://')) else f'http://{address}')
hostname = parsed.hostname or address
try:
ip_address = socket.gethostbyname(hostname)
except socket.gaierror:
@ -100,6 +103,7 @@ class Provider:
except (subprocess.TimeoutExpired, subprocess.SubprocessError) as e:
return False
def server_fn(self, history, verbose = False):
"""
Use a remote server with LLM to generate text.