diff --git a/web/app/components/try/app/index.tsx b/web/app/components/try/app/index.tsx
index d5c1063705..5509acc86d 100644
--- a/web/app/components/try/app/index.tsx
+++ b/web/app/components/try/app/index.tsx
@@ -16,7 +16,7 @@ const TryApp: FC<Props> = ({
 }) => {
   const { isFetching: isFetchingAppInfo, data: appInfo } = useGetTryAppInfo(appId)
   const mode = appInfo?.mode
-  const isChat = mode === 'chat' || mode === 'advanced-chat'
+  const isChat = ['chat', 'advanced-chat', 'agent-chat'].includes(mode!)
   const isCompletion = !isChat
   if (isFetchingAppInfo) {
     return (
diff --git a/web/app/components/try/app/text-generation.tsx b/web/app/components/try/app/text-generation.tsx
index c795dc14c8..fe246c75f5 100644
--- a/web/app/components/try/app/text-generation.tsx
+++ b/web/app/components/try/app/text-generation.tsx
@@ -45,7 +45,6 @@ const TextGeneration: FC<Props> = ({
     doSetInputs(newInputs)
     inputsRef.current = newInputs
   }, [])
-  // console.log(isPC, setInputs)
 
   const updateAppInfo = useWebAppStore(s => s.updateAppInfo)
   const { data: tryAppParams } = useGetTryAppParams(appId)
diff --git a/web/service/try-app.ts b/web/service/try-app.ts
index 5384cba9f7..78df7ba8ed 100644
--- a/web/service/try-app.ts
+++ b/web/service/try-app.ts
@@ -1,3 +1,4 @@
+import type { AppMode } from '@/types/app'
 import {
   get,
 } from './base'
@@ -7,7 +8,7 @@ import type {
 
 type TryAppInfo = {
   name: string
-  mode: 'chat' | 'advanced-chat' | 'text-generation' | 'workflow'
+  mode: AppMode
   site: SiteInfo
 }