ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix LiteLLM_ManagementEndpoint_MetadataFields

Browse Source
This commit is contained in:
Ishaan Jaffer 2025-11-12 15:20:09 -08:00
parent 0078d8a45f
commit 99f2cd1841
2 changed files with 20 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
litellm-proxy-extras/litellm_proxy_extras/migrations/20251113000001_add_project_fields/migration.sql Normal file
View File

@ -0,0 +1,5 @@
-- AlterTable: Add new fields to LiteLLM_ProjectTable
ALTER TABLE "LiteLLM_ProjectTable" ADD COLUMN "description" TEXT;
ALTER TABLE "LiteLLM_ProjectTable" ADD COLUMN "model_rpm_limit" JSONB NOT NULL DEFAULT '{}';
ALTER TABLE "LiteLLM_ProjectTable" ADD COLUMN "model_tpm_limit" JSONB NOT NULL DEFAULT '{}';

34
litellm/proxy/management_endpoints/project_endpoints.py
View File

@ -277,31 +277,27 @@ async def new_project(
prisma_client=prisma_client,
)
project_row = LiteLLM_ProjectTable(
**data.json(exclude_none=True),
object_permission_id=object_permission_id,
created_by=user_api_key_dict.user_id or litellm_proxy_admin_name,
updated_by=user_api_key_dict.user_id or litellm_proxy_admin_name,
)
# Prepare project data for database insertion
project_data = data.json(exclude_none=True)
project_data = prisma_client.jsonify_object(project_data)
# Add non-data fields
project_data["object_permission_id"] = object_permission_id
project_data["created_by"] = user_api_key_dict.user_id or litellm_proxy_admin_name
project_data["updated_by"] = user_api_key_dict.user_id or litellm_proxy_admin_name
# Handle metadata fields
for field in LiteLLM_ManagementEndpoint_MetadataFields:
if getattr(data, field, None) is not None:
_set_object_metadata_field(
object_data=project_row,
field_name=field,
value=getattr(data, field),
)
if field in project_data and project_data[field] is not None:
if project_data.get("metadata") is None:
project_data["metadata"] = {}
project_data["metadata"][field] = project_data.pop(field)
new_project_row = prisma_client.jsonify_object(
project_row.json(exclude_none=True)
)
verbose_proxy_logger.info(
f"new_project_row: {json.dumps(new_project_row, indent=2)}"
f"new_project_row: {json.dumps(project_data, indent=2)}"
)
response = await prisma_client.db.litellm_projecttable.create(
data={
**new_project_row, # type: ignore
},
data=project_data, # type: ignore
include={"litellm_budget_table": True},
)