Add missing auth() calls to file upload/download methods

[Copilot]

The post_file() and get_file() methods (and async variants) in DataAPI were not calling auth(), causing 401 Unauthorized errors on file_system.upload() and file_system.download() operations.

Changes

• agentrun/utils/__data_api_async_template.py: Added auth() calls to post_file_async() and get_file_async()
• agentrun/utils/data_api.py: Regenerated via codegen, propagating auth() to all four methods (sync + async variants)

Implementation

Applied the same authentication pattern used in get_video():

url = self.with_path(path, query=query)
req_headers = self.config.get_headers()
req_headers.update(headers or {})

# Added these 3 lines
cfg = Config.with_configs(self.config, config)
url, req_headers, query = self.auth(url, req_headers, query, config=cfg)

This ensures authentication tokens (X-API-Key or Agentrun-Access-Token) are properly included in file operation requests.

Original prompt

---

This section details on the original issue you should resolve

<issue_title>post_file() and get_file() methods missing auth() call in DataAPI</issue_title>
<issue_description>## Description

The post_file() and get_file() methods in agentrun/utils/data_api.py do not call the auth() method to add authentication headers, resulting in 401 Unauthorized errors when using file_system.upload() or file_system.download().

Environment

• agentrun-sdk version: 0.0.14
• Python version: 3.12
• OS: Windows 10

Steps to Reproduce

from agentrun.sandbox import Sandbox, TemplateType

# Create sandbox
sandbox = Sandbox.create(
    template_type=TemplateType.AIO,
    template_name="your-template-name",
)
sandbox.__enter__()

# This will fail with 401 error
result = sandbox.file_system.upload(
    local_file_path="local_file.txt",
    target_file_path="/tmp/file.txt"
)

Expected Behavior

File upload/download should work with proper authentication, similar to other API methods.

Actual Behavior

Returns 401 Unauthorized error:

{
  "code": "ERR_UNAUTHORIZED",
  "requestId": "1f34cca5-438c-4535-a885-259c041cd5f9",
  "message": "missing API key in header X-API-Key or query parameter"
}

The error indicates that neither X-API-Key header nor Agentrun-Access-Token header is being sent with the request.

Root Cause Analysis

In agentrun/utils/data_api.py:

Correct implementation (e.g., get_video() at line 1076-1107):

def get_video(self, path, save_path, query=None, headers=None, config=None):
    url = self.with_path(path, query=query)
    req_headers = self.config.get_headers()
    req_headers.update(headers or {})
    # ✅ Correctly calls auth() to add authentication token
    cfg = Config.with_configs(self.config, config)
    url, req_headers, query = self.auth(url, req_headers, query, config=cfg)
    # ... rest of the method

Buggy implementation (post_file() at line 880-935 and get_file() at line 983-1025):

def post_file(self, path, local_file_path, target_file_path, ...):
    url = self.with_path(path, query=query)
    req_headers = self.config.get_headers()
    req_headers.update(headers or {})
    # ❌ Missing auth() call - no authentication token added!
    # Directly proceeds to make request without authentication

Suggested Fix

Add the auth() call to both post_file() / post_file_async() and get_file() / get_file_async() methods:

def post_file(self, path, local_file_path, target_file_path, form_data=None, 
              query=None, headers=None, config=None):
    url = self.with_path(path, query=query)
    req_headers = self.config.get_headers()
    req_headers.update(headers or {})
    
    # Add this line (same as get_video):
    cfg = Config.with_configs(self.config, config)
    url, req_headers, query = self.auth(url, req_headers, query, config=cfg)
    
    # ... rest of the method

Workaround

Until this is fixed, users can work around by ensuring data_api.access_token is initialized before calling file operations:

# After sandbox creation, call any data_api method to trigger token initialization
sandbox.data_api.check_health()

# Now file operations should work (if the token is properly cached)
sandbox.file_system.upload(...)

Affected Methods

Method	File	Line	Status
post_file()	data_api.py	880-935	❌ Missing auth()
post_file_async()	data_api.py	821-878	❌ Missing auth()
get_file()	data_api.py	983-1025	❌ Missing auth()
get_file_async()	data_api.py	937-981	❌ Missing auth()
get_video()	data_api.py	1076-1121	✅ Has auth()
get_video_async()	data_api.py	1027-1074	✅ Has auth()

Impact

• Users cannot use sandbox.file_system.upload() for uploading files
• Users cannot use sandbox.file_system.download() for downloading files
• This affects both CodeInterpreterSandbox and AioSandbox

---

Discovered while developing LangChain integration with AgentRun sandbox.
</issue_description>

<agent_instructions>将 post_file get_file 移动到 data api 中，并且使用 auth 函数获取鉴权 header</agent_instructions>

Comments on the Issue (you are @copilot in this section)

• Fixes post_file() and get_file() methods missing auth() call in DataAPI #41

---

💬 We'd love your input! Share your thoughts on Copilot coding agent in our 2 minute survey.