Spaces:
Running
Running
:zap: [Enhance] Cleaner logger output
Browse files- networks/message_streamer.py +1 -1
- utils/enver.py +3 -4
networks/message_streamer.py
CHANGED
|
@@ -60,7 +60,7 @@ class MessageStreamer:
|
|
| 60 |
},
|
| 61 |
"stream": stream,
|
| 62 |
}
|
| 63 |
-
|
| 64 |
enver.set_envs(proxies=True)
|
| 65 |
stream = requests.post(
|
| 66 |
self.request_url,
|
|
|
|
| 60 |
},
|
| 61 |
"stream": stream,
|
| 62 |
}
|
| 63 |
+
logger.back(self.request_url)
|
| 64 |
enver.set_envs(proxies=True)
|
| 65 |
stream = requests.post(
|
| 66 |
self.request_url,
|
utils/enver.py
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
import json
|
| 2 |
-
import requests
|
| 3 |
import os
|
| 4 |
|
| 5 |
from pathlib import Path
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
class OSEnver:
|
|
@@ -15,8 +15,6 @@ class OSEnver:
|
|
| 15 |
|
| 16 |
def restore_envs(self):
|
| 17 |
self.envs = self.envs_stack.pop()
|
| 18 |
-
if self.global_scope:
|
| 19 |
-
os.environ = self.envs
|
| 20 |
|
| 21 |
def set_envs(self, secrets=True, proxies=None, store_envs=True):
|
| 22 |
# caller_info = inspect.stack()[1]
|
|
@@ -55,7 +53,8 @@ class OSEnver:
|
|
| 55 |
"https": self.proxy,
|
| 56 |
}
|
| 57 |
|
| 58 |
-
|
|
|
|
| 59 |
|
| 60 |
|
| 61 |
enver = OSEnver()
|
|
|
|
| 1 |
import json
|
|
|
|
| 2 |
import os
|
| 3 |
|
| 4 |
from pathlib import Path
|
| 5 |
+
from utils.logger import logger
|
| 6 |
|
| 7 |
|
| 8 |
class OSEnver:
|
|
|
|
| 15 |
|
| 16 |
def restore_envs(self):
|
| 17 |
self.envs = self.envs_stack.pop()
|
|
|
|
|
|
|
| 18 |
|
| 19 |
def set_envs(self, secrets=True, proxies=None, store_envs=True):
|
| 20 |
# caller_info = inspect.stack()[1]
|
|
|
|
| 53 |
"https": self.proxy,
|
| 54 |
}
|
| 55 |
|
| 56 |
+
if self.proxy:
|
| 57 |
+
logger.note(f"Using proxy: [{self.proxy}]")
|
| 58 |
|
| 59 |
|
| 60 |
enver = OSEnver()
|