Skip to main content
PATCH
/
agents
/
{agent_id}
Python
import os
from datagrid_ai import Datagrid

client = Datagrid(
    api_key=os.environ.get("DATAGRID_API_KEY"),  # This is the default and can be omitted
)
agent = client.agents.update(
    agent_id="agent_id",
)
print(agent.id)
{
  "object": "agent",
  "id": "<string>",
  "name": "<string>",
  "description": "<string>",
  "created_at": "2023-11-07T05:31:56Z",
  "agent_model": "magpie-1.1",
  "llm_model": "gemini-2.5-flash",
  "knowledge_ids": [
    "<string>"
  ],
  "tools": [
    {
      "name": "data_analysis",
      "connection_id": "<string>"
    }
  ],
  "system_prompt": "<string>",
  "custom_prompt": "<string>",
  "planning_prompt": "<string>"
}

Authorizations

​
Authorization
string
header
required

Bearer authentication header of the form Bearer <token>, where <token> is your auth token.

Path Parameters

​
agent_id
string
required

The ID of the agent to update

Body

application/json
​
agent_model
default:magpie-1.1

The version of Datagrid's agent brain.

  • magpie-1.1 is the default and most powerful model.
  • magpie-1.1-flash is a faster model useful for RAG usecases, it currently only supports semantic_search tool. Structured outputs are not supported with this model.
  • Can also accept any custom string value for future model versions.
  • Magpie-2.0 our latest agentic model with more proactive planning and reasoning capabilities.
Available options:
magpie-1.1,
magpie-1.1-flash,
magpie-2.0,
magpie-1
​
llm_model
default:gemini-2.5-flash

The LLM used to generate responses.

Available options:
gemini-3-pro-preview,
gemini-2.5-pro,
gemini-2.5-pro-preview-05-06,
gemini-2.5-flash,
gemini-2.5-flash-preview-04-17,
gemini-2.5-flash-lite,
gpt-5,
gpt-5.1,
gemini-2.0-flash-001,
gemini-2.0-flash,
gemini-1.5-pro-001,
gemini-1.5-pro-002,
gemini-1.5-flash-002,
gemini-1.5-flash-001,
chatgpt-4o-latest,
gpt-4o,
gpt-4,
gpt-4-turbo,
gpt-4o-mini
​
knowledge_ids
string[] | null

Array of Knowledge IDs the agent should use during the converse. When ommited, all knowledge is used.

​
tools
object[]

Array of the agent tools to enable. If not provided - default tools of the agent are used. If empty list provided - none of the tools are used. If null provided - all tools are used. When connection_id is set for a tool, it will use that specific connection instead of the default one.

Knowledge management tools:

  • data_analysis: Answer statistical or analytical questions like "Show my quarterly revenue growth"
  • semantic_search: Search knowledge through natural language queries.
  • agent_memory: Agents can remember experiences, conversations and user preferences.
  • schema_info: Helps the Agent understand column names and dataset purpose. Avoid disabling
  • table_info: Allow the AI Agent to get information about datasets and schemas
  • create_dataset: Agents respond with data tables

Actions:

  • calendar: Allow the Agent to access and make changes to your Google Calendar
  • schedule_recurring_message_tool: Eliminate busywork such as: "Send a summary of today's meetings at 5pm on workdays"

Data processing tools:

  • data_classification: Agents handle queries like "Label these emails as high, medium, or low priority"
  • data_extraction: Helps the agent understand data from other tools. Avoid disabling
  • image_detection: Extract information from images using AI
  • pdf_extraction: Extraction of information from PDFs using AI

Enhanced response tools:

  • connect_data: Agents provide buttons to import data in response to queries like "Connect Hubspot"
  • download_data: Agents handle queries like "download the table as CSV"

Web tools:

  • web_search: Agents search the internet, and provide links to their sources
  • fetch_url: Fetch URL content
  • company_prospect_researcher: Agents provide information about companies
  • people_prospect_researcher: Agents provide information about people
​
system_prompt
string | null

Directs your AI Agent's operational behavior.

​
custom_prompt
string | null

Use custom prompt to instruct the style and formatting of the agent's response

​
planning_prompt
string | null

Define the planning strategy your AI Agent should use when breaking down tasks and solving problems

​
disabled_tools
(enum<string> | string | object)[] | null

Array of the agent tools to disable. Disabling is performed after the 'agent_tools' rules are applied. For example, agent_tools: null and disabled_tools: [data_analysis] will enable everything but the data_analysis tool. If nothing or [] is provided, nothing is disabled and therefore only the agent_tools setting is relevant.

Available options:
data_analysis,
semantic_search,
agent_memory,
schema_info,
table_info,
create_dataset,
find_files,
read_file_contents,
file_analysis,
calendar,
email,
schedule_recurring_message_tool,
procore,
egnyte,
notion,
slack,
microsoft_teams,
sharepoint,
drive,
fieldwire,
planner,
webbrowser,
pdf_manipulation,
pdf_generator,
acc,
docusign,
webflow,
hubspot,
nec,
github,
trimble_project_site,
trimble,
linkedin,
google_docs,
google_slides,
google_sheets,
avoma,
content_writer,
code_tool,
data_classification,
data_extraction,
image_detection,
attachment_extraction,
pdf_extraction,
pdf_page_info,
youtube_video_analysis,
calculate,
pdf_form_filling,
image_generator,
video_generator,
connect_data,
download_data,
web_search,
fetch_url,
company_prospect_researcher,
people_prospect_researcher
​
name
string | null

The name of the agent

Maximum string length: 255
​
description
string | null

The description of the agent

Maximum string length: 1000

Response

Updated agent

​
object
enum<string>
required

The object type, always 'agent'

Available options:
agent
​
id
string
required

Unique identifier for the agent

​
name
string
required

The name of the agent

​
description
string | null
required

The description of the agent

​
created_at
string<date-time>
required

The ISO string for when the agent was created

​
agent_model
default:magpie-1.1
required

The version of Datagrid's agent brain.

  • magpie-1.1 is the default and most powerful model.
  • magpie-1.1-flash is a faster model useful for RAG usecases, it currently only supports semantic_search tool. Structured outputs are not supported with this model.
  • Can also accept any custom string value for future model versions.
  • Magpie-2.0 our latest agentic model with more proactive planning and reasoning capabilities.
Available options:
magpie-1.1,
magpie-1.1-flash,
magpie-2.0,
magpie-1
​
llm_model
default:gemini-2.5-flash
required

The LLM used to generate responses.

Available options:
gemini-3-pro-preview,
gemini-2.5-pro,
gemini-2.5-pro-preview-05-06,
gemini-2.5-flash,
gemini-2.5-flash-preview-04-17,
gemini-2.5-flash-lite,
gpt-5,
gpt-5.1,
gemini-2.0-flash-001,
gemini-2.0-flash,
gemini-1.5-pro-001,
gemini-1.5-pro-002,
gemini-1.5-flash-002,
gemini-1.5-flash-001,
chatgpt-4o-latest,
gpt-4o,
gpt-4,
gpt-4-turbo,
gpt-4o-mini
​
knowledge_ids
string[] | null
required

Array of Knowledge IDs the agent should use during the converse. When ommited, all knowledge is used.

​
tools
object[]
required

Tools that this agent can use.

​
system_prompt
string | null
required

Directs your AI Agent's operational behavior.

​
custom_prompt
string | null
required

Use custom prompt to instruct the style and formatting of the agent's response

​
planning_prompt
string | null
required

Define the planning strategy your AI Agent should use when breaking down tasks and solving problems