Add auth() calls to post_file and get_file methods

Co-authored-by: OhYee <13498329+OhYee@users.noreply.github.com>

Author: Copilot

agentrun/utils/__data_api_async_template.py

@@ -604,6 +604,9 @@ async def post_file_async(
         url = self.with_path(path, query=query)
         req_headers = self.config.get_headers()
         req_headers.update(headers or {})
+        # Apply authentication (may modify URL, headers, and query)
+        cfg = Config.with_configs(self.config, config)
+        url, req_headers, query = self.auth(url, req_headers, query, config=cfg)
 
         try:
             with open(local_file_path, "rb") as f:
@@ -654,6 +657,9 @@ async def get_file_async(
         url = self.with_path(path, query=query)
         req_headers = self.config.get_headers()
         req_headers.update(headers or {})
+        # Apply authentication (may modify URL, headers, and query)
+        cfg = Config.with_configs(self.config, config)
+        url, req_headers, query = self.auth(url, req_headers, query, config=cfg)
 
         try:
             async with httpx.AsyncClient(

agentrun/utils/data_api.py

@@ -447,7 +447,9 @@ def _make_request(
         )
 
         try:
-            with httpx.Client(timeout=self.config.get_timeout()) as client:
+            with httpx.Client(
+                timeout=self.config.get_timeout()
+            ) as client:
                 response = client.request(
                     method,
                     url,
@@ -856,6 +858,9 @@ async def post_file_async(
         url = self.with_path(path, query=query)
         req_headers = self.config.get_headers()
         req_headers.update(headers or {})
+        # Apply authentication (may modify URL, headers, and query)
+        cfg = Config.with_configs(self.config, config)
+        url, req_headers, query = self.auth(url, req_headers, query, config=cfg)
 
         try:
             with open(local_file_path, "rb") as f:
@@ -915,6 +920,9 @@ def post_file(
         url = self.with_path(path, query=query)
         req_headers = self.config.get_headers()
         req_headers.update(headers or {})
+        # Apply authentication (may modify URL, headers, and query)
+        cfg = Config.with_configs(self.config, config)
+        url, req_headers, query = self.auth(url, req_headers, query, config=cfg)
 
         try:
             with open(local_file_path, "rb") as f:
@@ -923,7 +931,9 @@ def post_file(
                 data = form_data or {}
                 data["path"] = target_file_path
 
-                with httpx.Client(timeout=self.config.get_timeout()) as client:
+                with httpx.Client(
+                    timeout=self.config.get_timeout()
+                ) as client:
                     response = client.post(
                         url, files=files, data=data, headers=req_headers
                     )
@@ -963,6 +973,9 @@ async def get_file_async(
         url = self.with_path(path, query=query)
         req_headers = self.config.get_headers()
         req_headers.update(headers or {})
+        # Apply authentication (may modify URL, headers, and query)
+        cfg = Config.with_configs(self.config, config)
+        url, req_headers, query = self.auth(url, req_headers, query, config=cfg)
 
         try:
             async with httpx.AsyncClient(
@@ -1009,9 +1022,14 @@ def get_file(
         url = self.with_path(path, query=query)
         req_headers = self.config.get_headers()
         req_headers.update(headers or {})
+        # Apply authentication (may modify URL, headers, and query)
+        cfg = Config.with_configs(self.config, config)
+        url, req_headers, query = self.auth(url, req_headers, query, config=cfg)
 
         try:
-            with httpx.Client(timeout=self.config.get_timeout()) as client:
+            with httpx.Client(
+                timeout=self.config.get_timeout()
+            ) as client:
                 response = client.get(url, headers=req_headers)
                 response.raise_for_status()
 
@@ -1107,7 +1125,9 @@ def get_video(
         url, req_headers, query = self.auth(url, req_headers, query, config=cfg)
 
         try:
-            with httpx.Client(timeout=self.config.get_timeout()) as client:
+            with httpx.Client(
+                timeout=self.config.get_timeout()
+            ) as client:
                 response = client.get(url, headers=req_headers)
                 response.raise_for_status()