PR #24855: feat: add environment and user tracking to prompt management

• Repository: BerriAI/litellm
• Author: talhaanwarch
• State: open | merged: False
• Link: https://github.com/BerriAI/litellm/pull/24855

Description (problem / solution / changelog)

Relevant issues

Fixes #24854

Pre-Submission checklist

• I have Added testing in the
  tests/test_litellm/ directory, Adding at least 1 test is a hard requirement - see
  details
• My PR passes all unit tests on make test-unit
• My PR's scope is as isolated as possible, it only solves 1 specific problem
• I have requested a Greptile review by commenting @greptileai and received a Confidence
  Score of at least 4/5 before requesting a maintainer review

Delays in PR merge?

If you're seeing a delay in your PR being merged, ping the LiteLLM Team on [Slack (#pr-review)](htt ps://join.slack.com/t/litellmossslack/shared_invite/zt-3o7nkuyfr-p_kbNJj8taRfXGgQI1~YyA).

CI (LiteLLM team)

CI status guideline:

• 50-55 passing tests: main is stable with minor issues.
• 45-49 passing tests: acceptable but needs attention
• <= 40 passing tests: unstable; be careful with your merges and assess the risk.

• Branch creation CI run
  Link:

• CI run for the last commit
  Link:

• Merge / cherry-pick CI run
  Links:

Type

🆕 New Feature

Changes

Database

• Add environment (String, default "development") and created_by (String, nullable) columns to LiteLLM_PromptTable
• Change unique constraint from [prompt_id, version] to [prompt_id, version, environment]
• Add index on [prompt_id, environment]
• Prisma migration included — backfills existing rows with environment = 'development'

Backend (litellm/proxy/prompts/prompt_endpoints.py)

• All CRUD endpoints support optional environment query/body parameter
• created_by set automatically from authenticated user on create/update
• get_prompt_info and get_prompt_versions query DB for environment-specific data
• PromptInfoResponse includes environments field listing all environments a prompt is deployed to
• Supports fetching specific version via versioned prompt_id (e.g.,
  testprompt.v2?environment=staging)

UI

• Prompt list: Environment filter dropdown, Environment and Created By columns
• Prompt detail: Environment tabs (development/staging/production) that switch the entire view, version history table with Created By and Edit action per row, old-version banner
• Prompt editor: Environment selector dropdown, switching environments re-fetches correct
  version

Tests (tests/test_litellm/proxy/prompts/test_prompt_environment.py)

• 8 new tests covering: model defaults, custom environments, create/update with environment+created_by, delete scoped to environment, versioned prompt spec creation

Changed files

• litellm-proxy-extras/litellm_proxy_extras/migrations/20260331000000_add_prompt_environment_and_created_by/migration.sql (added, +12/-0)
• litellm-proxy-extras/litellm_proxy_extras/schema.prisma (modified, +4/-1)
• litellm/proxy/prompts/prompt_endpoints.py (modified, +289/-115)
• litellm/proxy/schema.prisma (modified, +4/-1)
• litellm/types/prompts/init_prompts.py (modified, +6/-0)
• schema.prisma (modified, +4/-1)
• tests/test_litellm/proxy/prompts/test_prompt_environment.py (added, +253/-0)
• ui/litellm-dashboard/src/components/networking.tsx (modified, +22/-6)
• ui/litellm-dashboard/src/components/prompts.tsx (modified, +19/-6)
• ui/litellm-dashboard/src/components/prompts/prompt_editor_view/PromptEditorHeader.tsx (modified, +16/-1)
• ui/litellm-dashboard/src/components/prompts/prompt_editor_view/index.tsx (modified, +20/-1)
• ui/litellm-dashboard/src/components/prompts/prompt_editor_view/types.ts (modified, +1/-0)
• ui/litellm-dashboard/src/components/prompts/prompt_editor_view/utils.ts (modified, +1/-0)
• ui/litellm-dashboard/src/components/prompts/prompt_info.tsx (modified, +225/-90)
• ui/litellm-dashboard/src/components/prompts/prompt_table.tsx (modified, +30/-0)