tests.system.databricks.example_databricks_agent

Invoke the deterministic agent in resources/agent_server on Databricks Apps.

Deploy the bundled app and configure DATABRICKS_AGENT_APP_URL and the databricks_oauth connection before running this system test.

Attributes

DAG_ID

APP_URL

CONN_ID

INPUT

SESSION_ID

invoke

test_run

Functions

invoke_managed_agent(app_url, conn_id, session_id)

verify_result(result, session_id)

Verify that the real agent server completed and preserved the input.

Module Contents

tests.system.databricks.example_databricks_agent.DAG_ID = 'example_databricks_agent'[source]
tests.system.databricks.example_databricks_agent.APP_URL[source]
tests.system.databricks.example_databricks_agent.CONN_ID[source]
tests.system.databricks.example_databricks_agent.INPUT[source]
tests.system.databricks.example_databricks_agent.SESSION_ID = '{{ dag.dag_id }}-{{ run_id }}'[source]
tests.system.databricks.example_databricks_agent.invoke_managed_agent(app_url, conn_id, session_id)[source]
tests.system.databricks.example_databricks_agent.verify_result(result, session_id)[source]

Verify that the real agent server completed and preserved the input.

tests.system.databricks.example_databricks_agent.invoke[source]
tests.system.databricks.example_databricks_agent.test_run[source]

Was this entry helpful?