✨ feat: add 01.AI APIKey Error Form

lobehub · arvinxx · Mar 25, 2024 · Mar 18, 2024 · Mar 18, 2024 · Mar 18, 2024
commit 774f4c6528888b1064f0c7b71ebae60d39b26097
diff --git a/src/features/Conversation/Error/APIKeyForm/ZeroOne.tsx b/src/features/Conversation/Error/APIKeyForm/ZeroOne.tsx
@@ -0,0 +1,60 @@
+import { ZeroOne } from '@lobehub/icons';
+import { Input } from 'antd';
+import { memo } from 'react';
+import { useTranslation } from 'react-i18next';
+
+import { ModelProvider } from '@/libs/agent-runtime';
+import { useGlobalStore } from '@/store/global';
+import { modelProviderSelectors } from '@/store/global/selectors';
+
+import { FormAction } from '../style';
+
+const ZeroOneForm = memo(() => {
+  const { t } = useTranslation('error');
+  // const [showProxy, setShow] = useState(false);
+
+  const [apiKey, setConfig] = useGlobalStore((s) => [
+    modelProviderSelectors.zerooneAPIKey(s),
+    s.setModelProviderConfig,
+  ]);
+
+  return (
+    <FormAction
+      avatar={<ZeroOne size={56} />}
+      description={t('unlock.apikey.ZeroOne.description')}
+      title={t('unlock.apikey.ZeroOne.title')}
+    >
+      <Input.Password
+        autoComplete={'new-password'}
+        onChange={(e) => {
+          setConfig(ModelProvider.ZeroOne, { apiKey: e.target.value });
+        }}
+        placeholder={'*********************************'}
+        type={'block'}
+        value={apiKey}
+      />
+      {/*{showProxy ? (*/}
+      {/*  <Input*/}
+      {/*    onChange={(e) => {*/}
+      {/*      setConfig({ endpoint: e.target.value });*/}
+      {/*    }}*/}
+      {/*    placeholder={'https://api.openai.com/v1'}*/}
+      {/*    type={'block'}*/}
+      {/*    value={proxyUrl}*/}
+      {/*  />*/}
+      {/*) : (*/}
+      {/*  <Button*/}
+      {/*    icon={<Icon icon={Network} />}*/}
+      {/*    onClick={() => {*/}
+      {/*      setShow(true);*/}
+      {/*    }}*/}
+      {/*    type={'text'}*/}
+      {/*  >*/}
+      {/*    {t('unlock.apikey.addProxyUrl')}*/}
+      {/*  </Button>*/}
+      {/*)}*/}
+    </FormAction>
+  );
+});
+
+export default ZeroOneForm;
diff --git a/src/features/Conversation/Error/APIKeyForm/index.tsx b/src/features/Conversation/Error/APIKeyForm/index.tsx
@@ -15,6 +15,7 @@ import MoonshotForm from './Moonshot';
 import OpenAIForm from './OpenAI';
 import OpenRouterForm from './OpenRouter';
 import PerplexityForm from './Perplexity';
+import ZeroOneForm from './ZeroOne';
 import ZhipuForm from './Zhipu';
 
 interface APIKeyFormProps {
@@ -60,11 +61,15 @@ const APIKeyForm = memo<APIKeyFormProps>(({ id, provider }) => {
       case ModelProvider.Groq: {
         return <GroqForm />;
       }
-      
+
       case ModelProvider.OpenRouter: {
         return <OpenRouterForm />;
       }
 
+      case ModelProvider.ZeroOne: {
+        return <ZeroOneForm />;
+      }
+
       default:
       case ModelProvider.OpenAI: {
         return <OpenAIForm />;