mirror of
https://github.com/ParisNeo/ollama_proxy_server.git
synced 2025-09-06 13:42:26 +00:00
Update main.py
This commit is contained in:
@@ -77,6 +77,7 @@ def main():
|
||||
self.proxy()
|
||||
|
||||
def _validate_user_and_key(self):
|
||||
try:
|
||||
# Extract the bearer token from the headers
|
||||
auth_header = self.headers.get('Authorization')
|
||||
if not auth_header or not auth_header.startswith('Bearer '):
|
||||
@@ -91,7 +92,8 @@ def main():
|
||||
else:
|
||||
self.user = "unknown"
|
||||
return False
|
||||
|
||||
except:
|
||||
return False
|
||||
def proxy(self):
|
||||
if not self._validate_user_and_key():
|
||||
ASCIIColors.red(f'User is not authorized')
|
||||
|
Reference in New Issue
Block a user