edit create key button

This commit is contained in:
Ishaan Jaff 2025-01-27 15:58:51 -08:00
parent 4dd9bd1bf7
commit 817f34b649

View file

@ -23,6 +23,7 @@ import {
message,
Radio,
} from "antd";
import { unfurlWildcardModelsInList, getModelDisplayName } from "./key_team_helpers/fetch_available_models_team_key";
import {
keyCreateCall,
slackBudgetAlertsHealthCheck,
@ -80,8 +81,8 @@ const CreateKey: React.FC<CreateKeyProps> = ({
const [isModalVisible, setIsModalVisible] = useState(false);
const [apiKey, setApiKey] = useState(null);
const [softBudget, setSoftBudget] = useState(null);
const [userModels, setUserModels] = useState([]);
const [modelsToPick, setModelsToPick] = useState([]);
const [userModels, setUserModels] = useState<string[]>([]);
const [modelsToPick, setModelsToPick] = useState<string[]>([]);
const [keyOwner, setKeyOwner] = useState("you");
const [predefinedTags, setPredefinedTags] = useState(getPredefinedTags(data));
const [guardrailsList, setGuardrailsList] = useState<string[]>([]);
@ -213,6 +214,8 @@ const CreateKey: React.FC<CreateKeyProps> = ({
tempModelsToPick = userModels;
}
tempModelsToPick = unfurlWildcardModelsInList(tempModelsToPick, userModels);
setModelsToPick(tempModelsToPick);
}, [team, userModels]);
@ -310,7 +313,7 @@ const CreateKey: React.FC<CreateKeyProps> = ({
</Option>
{modelsToPick.map((model: string) => (
<Option key={model} value={model}>
{model}
{getModelDisplayName(model)}
</Option>
))}
</Select>