bug fix

docker_as_a_service/docker_as_a_service.py

@@ -51,17 +51,14 @@ async def stream_generator(request_obj):
         # Get list of existing files before download
         existing_file_before_download = []
 
-        if request_obj.endpoint == "down_bilibili":
-            video_id = request_obj.client_command
-            cmd = [
-                '/root/.dotnet/tools/BBDown',
-                video_id,
-                '--use-app-api',
-                '--work-dir', 
-                f'{os.path.abspath(temp_dir)}'
-            ]
-        else:
-            assert False, "Invalid endpoint"
+        video_id = request_obj.client_command
+        cmd = [
+            '/root/.dotnet/tools/BBDown',
+            video_id,
+            '--use-app-api',
+            '--work-dir', 
+            f'{os.path.abspath(temp_dir)}'
+        ]
 
 
         cmd = ' '.join(cmd)
@@ -145,7 +142,6 @@ async def stream_response(file: UploadFile = File(...)):
     with io.BytesIO(content) as f:
         request_obj = pickle.load(f)
     # process the request_obj
-    request_obj.endpoint = "down_bilibili"
     return StreamingResponse(stream_generator(request_obj), media_type="application/octet-stream")
 
 @app.post("/search")