scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

chore - vector store permissions enterprise

Browse source
This commit is contained in:
Ishaan Jaff 2025-05-31 12:01:51 -07:00
parent 372de1476b
commit 75f87724bd
5 changed files with 76 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
ui/litellm-dashboard/src/components/common_components/PremiumVectorStoreSelector.tsx Normal file
View file

@ -0,0 +1,50 @@
import React from 'react';
import { Text } from "@tremor/react";
import VectorStoreSelector from "../vector_store_management/VectorStoreSelector";
interface PremiumVectorStoreSelectorProps {
onChange: (values: string[]) => void;
value: string[];
accessToken: string;
placeholder?: string;
premiumUser?: boolean;
}
export function PremiumVectorStoreSelector({
onChange,
value,
accessToken,
placeholder = "Select vector stores",
premiumUser = false
}: PremiumVectorStoreSelectorProps) {
if (!premiumUser) {
return (
<div>
<div className="flex flex-wrap gap-2 mb-3">
<div className="inline-flex items-center px-3 py-1.5 rounded-lg bg-blue-50 border border-blue-200 text-blue-800 text-sm font-medium opacity-50">
✨ premium-vector-store-1
</div>
<div className="inline-flex items-center px-3 py-1.5 rounded-lg bg-blue-50 border border-blue-200 text-blue-800 text-sm font-medium opacity-50">
✨ premium-vector-store-2
</div>
</div>
<div className="p-3 bg-yellow-50 border border-yellow-200 rounded-lg">
<Text className="text-sm text-yellow-800">
Vector store access control is a LiteLLM Enterprise feature. Get a trial key <a href="https://litellm.ai/pricing" target="_blank" rel="noopener noreferrer" className="underline">here</a>.
</Text>
</div>
</div>
);
}
return (
<VectorStoreSelector
onChange={onChange}
value={value}
accessToken={accessToken}
placeholder={placeholder}
/>
);
}
export default PremiumVectorStoreSelector;

6
ui/litellm-dashboard/src/components/create_key_button.tsx
View file

@ -34,6 +34,7 @@ import {
userFilterUICall,
} from "./networking";
import VectorStoreSelector from "./vector_store_management/VectorStoreSelector";
import PremiumVectorStoreSelector from "./common_components/PremiumVectorStoreSelector";
import { Team } from "./key_team_helpers/key_list";
import TeamDropdown from "./common_components/team_dropdown";
import { InfoCircleOutlined } from '@ant-design/icons';
@ -55,6 +56,7 @@ interface CreateKeyProps {
data: any[] | null;
teams: Team[] | null;
addKey: (data: any) => void;
premiumUser?: boolean;
}
interface User {
@ -152,6 +154,7 @@ const CreateKey: React.FC<CreateKeyProps> = ({
accessToken,
data,
addKey,
premiumUser = false,
}) => {
const [form] = Form.useForm();
const [isModalVisible, setIsModalVisible] = useState(false);
@ -699,11 +702,12 @@ const CreateKey: React.FC<CreateKeyProps> = ({
className="mt-4"
help="Select vector stores this key can access. Leave empty for access to all vector stores"
>
<VectorStoreSelector
<PremiumVectorStoreSelector
onChange={(values) => form.setFieldValue('allowed_vector_store_ids', values)}
value={form.getFieldValue('allowed_vector_store_ids')}
accessToken={accessToken}
placeholder="Select vector stores (optional)"
premiumUser={premiumUser}
/>
</Form.Item>

9
ui/litellm-dashboard/src/components/key_edit_view.tsx
View file

@ -6,6 +6,7 @@ import { fetchTeamModels } from "../components/create_key_button";
import { modelAvailableCall } from "./networking";
import NumericalInput from "./shared/numerical_input";
import VectorStoreSelector from "./vector_store_management/VectorStoreSelector";
import PremiumVectorStoreSelector from "./common_components/PremiumVectorStoreSelector";
interface KeyEditViewProps {
keyData: KeyResponse;
@ -15,6 +16,7 @@ interface KeyEditViewProps {
accessToken: string | null;
userID: string | null;
userRole: string | null;
premiumUser?: boolean;
}
// Add this helper function
@ -43,7 +45,9 @@ export function KeyEditView({
teams,
accessToken,
userID,
userRole }: KeyEditViewProps) {
userRole,
premiumUser = false
}: KeyEditViewProps) {
const [form] = Form.useForm();
const [userModels, setUserModels] = useState<string[]>([]);
const team = teams?.find(team => team.team_id === keyData.team_id);
@ -169,11 +173,12 @@ export function KeyEditView({
</Form.Item>
<Form.Item label="Vector Stores" name="vector_stores">
<VectorStoreSelector
<PremiumVectorStoreSelector
onChange={(values) => form.setFieldValue('vector_stores', values)}
value={form.getFieldValue('vector_stores')}
accessToken={accessToken || ""}
placeholder="Select vector stores"
premiumUser={premiumUser}
/>
</Form.Item>

10
ui/litellm-dashboard/src/components/team/team_info.tsx
View file

@ -35,6 +35,7 @@ import { getModelDisplayName } from "../key_team_helpers/fetch_available_models_
import { isAdminRole } from "@/utils/roles";
import ObjectPermissionsView from "../object_permissions_view";
import VectorStoreSelector from "../vector_store_management/VectorStoreSelector";
import PremiumVectorStoreSelector from "../common_components/PremiumVectorStoreSelector";
export interface TeamData {
team_id: string;
@ -79,6 +80,7 @@ export interface TeamInfoProps {
is_proxy_admin: boolean;
userModels: string[];
editTeam: boolean;
premiumUser?: boolean;
}
const TeamInfoView: React.FC<TeamInfoProps> = ({
@ -88,7 +90,8 @@ const TeamInfoView: React.FC<TeamInfoProps> = ({
is_team_admin,
is_proxy_admin,
userModels,
editTeam
editTeam,
premiumUser = false
}) => {
const [teamData, setTeamData] = useState<TeamData | null>(null);
const [loading, setLoading] = useState(true);
@ -326,6 +329,7 @@ const TeamInfoView: React.FC<TeamInfoProps> = ({
objectPermission={info.object_permission}
variant="card"
accessToken={accessToken}
premiumUser={premiumUser}
/>
</Grid>
</TabPanel>
@ -456,11 +460,12 @@ const TeamInfoView: React.FC<TeamInfoProps> = ({
</Form.Item>
<Form.Item label="Vector Stores" name="vector_stores">
<VectorStoreSelector
<PremiumVectorStoreSelector
onChange={(values) => form.setFieldValue('vector_stores', values)}
value={form.getFieldValue('vector_stores')}
accessToken={accessToken || ""}
placeholder="Select vector stores"
premiumUser={premiumUser}
/>
</Form.Item>
@ -532,6 +537,7 @@ const TeamInfoView: React.FC<TeamInfoProps> = ({
variant="inline"
className="pt-4 border-t border-gray-200"
accessToken={accessToken}
premiumUser={premiumUser}
/>
</div>
)}

8
ui/litellm-dashboard/src/components/teams.tsx
View file

@ -58,6 +58,7 @@ import {
import { CogIcon } from "@heroicons/react/outline";
import AvailableTeamsPanel from "@/components/team/available_teams";
import VectorStoreSelector from "./vector_store_management/VectorStoreSelector";
import PremiumVectorStoreSelector from "./common_components/PremiumVectorStoreSelector";
import type { KeyResponse, Team } from "./key_team_helpers/key_list";
const isLocal = process.env.NODE_ENV === "development";
const proxyBaseUrl = isLocal ? "http://localhost:4000" : null;
@ -72,6 +73,7 @@ interface TeamProps {
userID: string | null;
userRole: string | null;
organizations: Organization[] | null;
premiumUser?: boolean;
}
interface FilterState {
@ -134,7 +136,8 @@ const Teams: React.FC<TeamProps> = ({
setTeams,
userID,
userRole,
organizations
organizations,
premiumUser = false
}) => {
const [lastRefreshed, setLastRefreshed] = useState("");
const [currentOrg, setCurrentOrg] = useState<Organization | null>(null);
@ -1086,11 +1089,12 @@ const Teams: React.FC<TeamProps> = ({
className="mt-8"
help="Select vector stores this team can access. Leave empty for access to all vector stores"
>
<VectorStoreSelector
<PremiumVectorStoreSelector
onChange={(values) => form.setFieldValue('allowed_vector_store_ids', values)}
value={form.getFieldValue('allowed_vector_store_ids')}
accessToken={accessToken || ''}
placeholder="Select vector stores (optional)"
premiumUser={premiumUser}
/>
</Form.Item>
</AccordionBody>