plane/web/components/instance/ai-form.tsx

import { FC } from "react";
import { Controller, useForm } from "react-hook-form";
// ui
import { Button, Input } from "@plane/ui";
// types
import { IFormattedInstanceConfiguration } from "types/instance";
// hooks
import useToast from "hooks/use-toast";
// mobx store
import { useMobxStore } from "lib/mobx/store-provider";

export interface IInstanceAIForm {
  config: IFormattedInstanceConfiguration;
}

export interface AIFormValues {
  OPENAI_API_KEY: string;
  GPT_ENGINE: string;
}

export const InstanceAIForm: FC<IInstanceAIForm> = (props) => {
  const { config } = props;
  // store
  const { instance: instanceStore } = useMobxStore();
  // toast
  const { setToastAlert } = useToast();
  // form data
  const {
    handleSubmit,
    control,
    formState: { errors, isSubmitting },
  } = useForm<AIFormValues>({
    defaultValues: {
      OPENAI_API_KEY: config["OPENAI_API_KEY"],
      GPT_ENGINE: config["GPT_ENGINE"],
    },
  });

  const onSubmit = async (formData: AIFormValues) => {
    const payload: Partial<AIFormValues> = { ...formData };

    await instanceStore
      .updateInstanceConfigurations(payload)
      .then(() =>
        setToastAlert({
          title: "Success",
          type: "success",
          message: "AI Settings updated successfully",
        })
      )
      .catch((err) => console.error(err));
  };

  return (
    <>
      <div className="grid grid-col grid-cols-1 lg:grid-cols-3 items-center justify-between gap-x-16 gap-y-8 w-full">
        <div className="flex flex-col gap-1">
          <h4 className="text-sm">GPT Engine</h4>
          <Controller
            control={control}
            name="GPT_ENGINE"
            render={({ field: { value, onChange, ref } }) => (
              <Input
                id="GPT_ENGINE"
                name="GPT_ENGINE"
                type="text"
                value={value}
                onChange={onChange}
                ref={ref}
                hasError={Boolean(errors.GPT_ENGINE)}
                placeholder="gpt-3.5-turbo"
                className="rounded-md font-medium w-full"
              />
            )}
          />
          <p className="text-xs text-custom-text-400">
            Choose an OpenAI engine.{" "}
            <a
              href="https://platform.openai.com/docs/models/overview"
              target="_blank"
              className="text-custom-primary-100 hover:underline"
              rel="noreferrer"
            >
              Learn more
            </a>
          </p>
        </div>

        <div className="flex flex-col gap-1">
          <h4 className="text-sm">API Key</h4>
          <Controller
            control={control}
            name="OPENAI_API_KEY"
            render={({ field: { value, onChange, ref } }) => (
              <Input
                id="OPENAI_API_KEY"
                name="OPENAI_API_KEY"
                type="text"
                value={value}
                onChange={onChange}
                ref={ref}
                hasError={Boolean(errors.OPENAI_API_KEY)}
                placeholder="sk-asddassdfasdefqsdfasd23das3dasdcasd"
                className="rounded-md font-medium w-full"
              />
            )}
          />
          <p className="text-xs text-custom-text-400">
            You will find your API key{" "}
            <a
              href="https://platform.openai.com/api-keys"
              target="_blank"
              className="text-custom-primary-100 hover:underline"
              rel="noreferrer"
            >
              here.
            </a>
          </p>
        </div>
      </div>

      <div className="flex items-center py-1">
        <Button variant="primary" onClick={handleSubmit(onSubmit)} loading={isSubmitting}>
          {isSubmitting ? "Saving..." : "Save Changes"}
        </Button>
      </div>
    </>
  );
};