gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

proxy with authorize

Browse Source
This commit is contained in:
streetycat
2023-09-20 07:46:42 +00:00
committed by zhangzhen
parent c9af0cfc0d
commit c5184154b5
2 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/aios_kernel/local_llama_compute_node.py
+1 -1
View File
@@ -47,7 +47,7 @@ class LocalLlama_ComputeNode(Queue_ComputeNode):
"state": ComputeTaskState.ERROR,
"error": {
"code": response.status_code,
"message": "http request failed: " + response.status_code
"message": "http request failed: " + str(response.status_code)
}
}
else:
src/service/aios_shell/proxy.py
+14 -6
View File
@@ -16,17 +16,25 @@ def apply_storage():
return
host_url = proxy_cfg.value
if host_url is not None:
(proxy_type, host) = host_url.split("@")
if host_url is not None and len(host_url) > 3:
url_fields = host_url.split("@")
proxy_type, host, username, password = url_fields[0], None, None, None
if len(url_fields) > 1:
host = url_fields[1]
if len(url_fields) > 2:
username = url_fields[2]
if len(url_fields) > 3:
password = url_fields[3]
match proxy_type:
case "socks5":
(host, port) = host.split(":")
socks.set_default_proxy(socks.SOCKS5, host, int(port))
socks.set_default_proxy(socks.SOCKS5, host, int(port), username = username, password = password)
socket.socket = socks.socksocket
print("proxy {host_url} will be used.")
print(f"proxy {host_url} will be used.")
case _:
print("the proxy type ({proxy_type}) has not support.")
print(f"the proxy type ({proxy_type}) has not support.")
def declare_user_config():
user_config = AIStorage.get_instance().get_user_config()
user_config.add_user_config("proxy", "set your proxy service as 'proxy_type@host:port', 'proxy_type' = 'socks5'", True, None)
user_config.add_user_config("proxy", "set your proxy service as 'proxy_type@host:port@username@password', 'proxy_type' = 'socks5'", True, None)