tcsenpai/agenticSeek
1
0
Fork 0
mirror of https://github.com/tcsenpai/agenticSeek.git synced 2025-06-06 19:15:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

allow domains for provider_server_address

Browse Source
This commit is contained in:
aguspiza 2025-03-16 17:43:20 +01:00
parent f4e4c08bb9
commit c3bdcab79e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
sources/llm_provider.py
View File

@ -30,7 +30,7 @@ class Provider:
print("Warning: you are using an API provider. You data will be sent to the cloud.") print("Warning: you are using an API provider. You data will be sent to the cloud.")
self.get_api_key(self.provider_name) self.get_api_key(self.provider_name)
elif self.server != "": elif self.server != "":
print("Provider initialized at ", self.server) print("Provider", provider_name, "initialized at", self.server)
def get_api_key(self, provider): def get_api_key(self, provider):
load_dotenv() load_dotenv()
@ -48,7 +48,8 @@ class Provider:
""" """
try: try:
ip, port = address.rsplit(":", 1) ip, port = address.rsplit(":", 1)
ipaddress.ip_address(ip) if all(c.lower() in ".:abcdef0123456789" for c in ip):
ipaddress.ip_address(ip)
if not port.isdigit() or not (0 <= int(port) <= 65535): if not port.isdigit() or not (0 <= int(port) <= 65535):
raise ValueError("Port must be a number between 0 and 65535.") raise ValueError("Port must be a number between 0 and 65535.")
except ValueError as e: except ValueError as e: