[ISSUE] WorkspaceClient.files.dowload can't infer requester network zone in v0.69 when used in model serving endpoint

[crankswagon]

Description

WorkspaceClient.files.download('/Volumes/xyz') returns Can't infer requester network zone. error.

Reproduction

w = WorkspaceClient()
w.files.download(/Volumes/somefile)

Register to MLFlow model and serve via model serving endpoint.

Backend Azure Databricks, volume defined on external location.

Expected behavior
File downloads correctly from Volume with no error.

Is it a regression?
Yes.

Last known working in v0.67; errors in v0.69 with no code change.

[RobHark]

Same issue seen with WorkspaceClient.files.upload()

Description

WorkspaceClient.files.upload() fails with "Can't infer requester network zone"

Reproduction

import streamlit as st
from io import BytesIO
from databricks.sdk import WorkspaceClient
from dotenv import load_dotenv

load_dotenv()

w = WorkspaceClient()
uploaded = st.file_uploader("Upload a file")

if uploaded:
    path = f"/Volumes/<catologue>/<schema>/<volume>/{uploaded.name}"
    w.files.upload(path, BytesIO(uploaded.read()), overwrite=True)
    st.success(f"Upload succeeded: {uploaded.name}")

Full error message and stack trace:

databricks.sdk.errors.platform.InternalError: Can't infer requester network zone.
Traceback:
File "/app/python/source_code/.venv/lib/python3.11/site-packages/streamlit/runtime/scriptrunner/exec_code.py", line 128, in exec_func_with_error_handling
    result = func()
             ^^^^^^
File "/app/python/source_code/.venv/lib/python3.11/site-packages/streamlit/runtime/scriptrunner/script_runner.py", line 667, in code_to_exec
    _mpa_v1(self._main_script_path)
File "/app/python/source_code/.venv/lib/python3.11/site-packages/streamlit/runtime/scriptrunner/script_runner.py", line 165, in _mpa_v1
    page.run()
File "/app/python/source_code/.venv/lib/python3.11/site-packages/streamlit/navigation/page.py", line 300, in run
    exec(code, module.__dict__)  # noqa: S102
    ^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/app/python/source_code/pages/BugDemo.py", line 29, in <module>
    w.files.upload(target_path, binary_data, overwrite=True)
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/mixins/files.py", line 1138, in upload
    self._parallel_upload_from_stream(ctx, content)
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/mixins/files.py", line 1336, in _parallel_upload_from_stream
    raise e from None
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/mixins/files.py", line 1318, in _parallel_upload_from_stream
    self._parallel_multipart_upload_from_stream(ctx, session_token, contents, cloud_provider_session)
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/mixins/files.py", line 1465, in _parallel_multipart_upload_from_stream
    etag = self._do_upload_one_part(
           ^^^^^^^^^^^^^^^^^^^^^^^^^
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/mixins/files.py", line 1645, in _do_upload_one_part
    upload_part_urls_response = self._api.do(
                                ^^^^^^^^^^^^^
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/core.py", line 85, in do
    return self._api_client.do(
           ^^^^^^^^^^^^^^^^^^^^
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/_base_client.py", line 199, in do
    response = call(
               ^^^^^
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/retries.py", line 59, in wrapper
    raise err
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/retries.py", line 38, in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
File "/app/python/source_code/.venv/lib/python3.11/site-packages/databricks/sdk/_base_client.py", line 301, in _perform
    raise error from None

Expected behavior
File is uploaded to the volume with no error.

Is it a regression?
Yes.
The code worked running locally and on a deployed databricks app in databricks-sdk==0.67.0
Error appearing on databricks-sdk==0.69.0

[walthertimmer]

I'm experiencing the same error while having a fastapi app running in Databricks apps trying to upload a file to a Databricks volume.

[markpascani]

I'm getting the same error too in Databricks while i try to read volumes.

"InternalError: Can't infer requester network zone."

[bkinder-vrify]

I’m seeing the same issue on databricks-sdk==0.70.0 when using WorkspaceClient().files.download(...) inside a Databricks App.

The call fails with:

databricks.sdk.errors.platform.InternalError: Can't infer requester network zone

Downgrading to databricks-sdk==0.67.0 resolves the problem in the exact same app environment.

Seems consistent with the regression reported for v0.69 — still reproducible in v0.70 within serverless contexts (Apps).

[mfulz]

I too can verfiy this issue:

<=0.68 -> working
0.69 - 0.71 -> Can't infer requester network zone. If it helps that's happening for me under a compute / app (flask). Using a notebook with similar code is working fine also using >0.68