diff --git a/litellm-proxy-extras/litellm_proxy_extras/migrations/20260305191532_add_agent_id_to_tool_table/migration.sql b/litellm-proxy-extras/litellm_proxy_extras/migrations/20260305191532_add_agent_id_to_tool_table/migration.sql
new file mode 100644
index 00000000000..3215bf9ca63
--- /dev/null
+++ b/litellm-proxy-extras/litellm_proxy_extras/migrations/20260305191532_add_agent_id_to_tool_table/migration.sql
@@ -0,0 +1,6 @@
+-- AlterTable
+ALTER TABLE "LiteLLM_MCPServerTable" ADD COLUMN     "spec_path" TEXT;
+
+-- AlterTable
+ALTER TABLE "LiteLLM_ToolTable" ADD COLUMN     "agent_id" TEXT;
+
diff --git a/litellm-proxy-extras/litellm_proxy_extras/schema.prisma b/litellm-proxy-extras/litellm_proxy_extras/schema.prisma
index 6d20a71f039..bf562894182 100644
--- a/litellm-proxy-extras/litellm_proxy_extras/schema.prisma
+++ b/litellm-proxy-extras/litellm_proxy_extras/schema.prisma
@@ -1090,6 +1090,7 @@ model LiteLLM_ToolTable {
   team_id       String?             // team that first called this tool
   key_alias     String?             // human-readable alias of the virtual key
   user_agent    String?             // user-agent of the first request that discovered this tool
+  agent_id      String?             // agent that first triggered this tool call
   last_used_at  DateTime?           // timestamp of the most recent call
   created_at    DateTime  @default(now())
   created_by    String?
diff --git a/schema.prisma b/schema.prisma
index f7c07112417..816808ef46e 100644
--- a/schema.prisma
+++ b/schema.prisma
@@ -1090,6 +1090,7 @@ model LiteLLM_ToolTable {
   team_id        String?            // team that first called this tool
   key_alias      String?            // human-readable alias of the virtual key
   user_agent     String?            // user-agent of the first request that discovered this tool
+  agent_id       String?            // agent that first triggered this tool call
   last_used_at   DateTime?          // timestamp of the most recent call
   created_at     DateTime @default(now())
   created_by     String?