Fix the bug of dataflow with ID OpenCSGs/csghub-dataflow#61

Author: zhanglongbin

label_studio/core/views.py

@@ -282,10 +282,20 @@ def collect_metrics(request):
 def system_config(request):
     """Proxy endpoint to fetch system config from external API"""
     try:
+        # 从session或请求参数中获取origin
+        origin = request.session.get('origin') or request.GET.get('origin')
+
+        # 构建API URL
+        if origin:
+            # 确保origin以/结尾，然后拼接接口路径
+            origin = origin.rstrip('/')
+            api_url = f"{origin}/internal_api/system_config"
+        else:
+            # 如果没有origin，使用默认URL
+            api_url = 'https://opencsg.com/internal_api/system_config'
+
         # 调用外部API
-        response = requests.get(
-            'https://opencsg-stg.com/internal_api/system_config'
-        )
+        response = requests.get(api_url)
         response.raise_for_status()
 
         # 返回JSON响应

label_studio/users/views.py

@@ -445,6 +445,9 @@ def login_reques(request):
     # 将语言保存到session中
     request.session['language'] = language
     origin = request.GET.get('origin')  # 获取origin参数
+    # 将origin保存到session中，供后续接口使用
+    if origin:
+        request.session['origin'] = origin
     user = request.user
     # 设置默认重定向页面
     if not next_page or not url_has_allowed_host_and_scheme(url=next_page, allowed_hosts=request.get_host()):