databricks-model-serving
Warn
Audited by Gen Agent Trust Hub on Aug 11, 2026
Risk Level: MEDIUMREMOTE_CODE_EXECUTIONCOMMAND_EXECUTIONEXTERNAL_DOWNLOADSPROMPT_INJECTION
Full Analysis
- [REMOTE_CODE_EXECUTION]: The custom model template in
2-custom-pyfunc.mdusespickle.load()to deserialize preprocessors and models from file artifacts. This is a common pattern in MLflow but can lead to arbitrary code execution if the artifact files are compromised. Evidence:pickle.load(f)in2-custom-pyfunc.md. - [COMMAND_EXECUTION]: An example tool in
4-tools-integration.mduses theeval()function to calculate mathematical expressions. While it attempts to restrict the environment by limiting__builtins__, dynamic execution of user-supplied strings carries inherent risk. Evidence:eval(expression, {"__builtins__": {}}, allowed)in4-tools-integration.md. - [EXTERNAL_DOWNLOADS]: The skill instructs the agent to install several Python packages including
mlflow,databricks-langchain,langgraph, anddatabricks-agents. These are well-known libraries relevant to the skill's purpose. - [PROMPT_INJECTION]: The GenAI agents described in
3-genai-agents.mdingest untrusted user messages via theResponsesAgentRequestobject. As these agents are designed to call tools, they are susceptible to indirect prompt injection where malicious data influences tool execution. 1. Ingestion points:request.inputin3-genai-agents.md. 2. Boundary markers: None identified. 3. Capability inventory: Tool calling (4-tools-integration.md), file uploads (SKILL.md), and command execution on clusters (SKILL.md). 4. Sanitization: No explicit sanitization is demonstrated in the examples.
Audit Metadata