Merge pull request #150 from abstracta/remove-account-usage-link-azure

Remove account usage link when using azure
dotneet · Sep 20, 2023 · 7401afe · 7401afe
2 parents 95261fc + a427a28
commit 7401afe
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/components/Chat/ModelSelect.tsx b/components/Chat/ModelSelect.tsx
@@ -14,7 +14,7 @@ export const ModelSelect = () => {
  const [_, conversationsAction] = useConversations();
 
  const {
- state: { selectedConversation, models, defaultModelId },
+ state: { selectedConversation, models, defaultModelId, isAzureOpenAI },
  } = useContext(HomeContext);
 
  const handleChange = (e: React.ChangeEvent<HTMLSelectElement>) => {
@@ -52,7 +52,7 @@ export const ModelSelect = () => {
  ))}
  </select>
  </div>
- <div className="w-full mt-3 text-left text-neutral-700 dark:text-neutral-400 flex items-center">
+ {!isAzureOpenAI && <div className="w-full mt-3 text-left text-neutral-700 dark:text-neutral-400 flex items-center">
  <a
  href="https://platform.openai.com/account/usage"
  target="_blank"
@@ -62,6 +62,7 @@ export const ModelSelect = () => {
  {t('View Account Usage')}
  </a>
  </div>
+ }
  </div>
  );
 };
diff --git a/pages/api/home/home.state.tsx b/pages/api/home/home.state.tsx
@@ -31,6 +31,7 @@ export interface HomeInitialState {
  serverSideApiKeyIsSet: boolean;
  serverSidePluginKeysSet: boolean;
  stopConversationRef: MutableRefObject<boolean>;
+ isAzureOpenAI: boolean;
  supportEmail: string;
  promptSharingEnabled: boolean;
 }
@@ -62,6 +63,7 @@ export const initialState: Partial<HomeInitialState> = {
  defaultModelId: undefined,
  serverSideApiKeyIsSet: false,
  serverSidePluginKeysSet: false,
+ isAzureOpenAI: false,
  supportEmail: '',
  promptSharingEnabled: false,
 };
diff --git a/pages/api/home/home.tsx b/pages/api/home/home.tsx
@@ -8,7 +8,7 @@ import Head from 'next/head';
 import { useCreateReducer } from '@/hooks/useCreateReducer';
 
 import { cleanConversationHistory } from '@/utils/app/clean';
-import { DEFAULT_SYSTEM_PROMPT, SUPPORT_EMAIL, PROMPT_SHARING_ENABLED } from '@/utils/app/const';
+import { DEFAULT_SYSTEM_PROMPT, OPENAI_API_TYPE, PROMPT_SHARING_ENABLED, SUPPORT_EMAIL } from '@/utils/app/const';
 import { trpc } from '@/utils/trpc';
 
 import { Conversation } from '@/types/chat';
@@ -25,14 +25,16 @@ interface Props {
  serverSideApiKeyIsSet: boolean;
  serverSidePluginKeysSet: boolean;
  defaultModelId: OpenAIModelID;
- supportEmail: string,
- promptSharingEnabled: boolean,
+ isAzureOpenAI: boolean;
+ promptSharingEnabled: boolean;
+ supportEmail: string;
 }
 
 const Home = ({
  serverSideApiKeyIsSet,
  serverSidePluginKeysSet,
  defaultModelId,
+ isAzureOpenAI,
  supportEmail,
  promptSharingEnabled,
 }: Props) => {
@@ -51,6 +53,7 @@ const Home = ({
  initialState: {
  ...initialState,
  stopConversationRef: stopConversationRef,
+ isAzureOpenAI,
  supportEmail,
  promptSharingEnabled: promptSharingEnabled,
  } as HomeInitialState,
@@ -278,6 +281,7 @@ export const getServerSideProps: GetServerSideProps = async ({ locale }) => {
  props: {
  serverSideApiKeyIsSet: !!process.env.OPENAI_API_KEY,
  defaultModelId,
+ isAzureOpenAI: OPENAI_API_TYPE === "azure",
  serverSidePluginKeysSet,
  supportEmail: SUPPORT_EMAIL,
  ...(await serverSideTranslations(locale ?? 'en', [