Add using kilo for work button

webview-ui/src/components/chat/ChatRow.tsx

@@ -1629,7 +1629,12 @@ export const ChatRowContent = ({
 					)
 
 				case "payment_required_prompt": {
-					return <LowCreditWarning message={message} />
+					return (
+						<LowCreditWarning
+							message={message}
+							isOrganization={!!apiConfiguration.kilocodeOrganizationId}
+						/>
+					)
 				}
 				case "invalid_model": {
 					return <InvalidModelWarning message={message} isLast={isLast} />

webview-ui/src/components/kilocode/chat/LowCreditWarning.tsx

@@ -1,13 +1,15 @@
-import { ClineMessage } from "@roo-code/types"
+import { ClineMessage, getAppUrl, TelemetryEventName } from "@roo-code/types"
 import { vscode } from "@src/utils/vscode"
 import { VSCodeButton } from "@vscode/webview-ui-toolkit/react"
 import { RetryIconButton } from "../common/RetryIconButton"
 import styled from "styled-components"
 import { useTranslation } from "react-i18next"
-import { FreeModelsLink } from "../FreeModelsLink"
+import { VSCodeButtonLink } from "@/components/common/VSCodeButtonLink"
+import { telemetryClient } from "@/utils/TelemetryClient"
 
 type LowCreditWarningProps = {
 	message: ClineMessage
+	isOrganization: boolean
 }
 
 const HeaderContainer = styled.div`
@@ -24,7 +26,7 @@ const Description = styled.div`
 	overflow-wrap: anywhere;
 `
 
-export const LowCreditWarning = ({ message }: LowCreditWarningProps) => {
+export const LowCreditWarning = ({ message, isOrganization }: LowCreditWarningProps) => {
 	const { t } = useTranslation()
 	let data = { title: "Error", message: "Payment required.", balance: "-?.??", buyCreditsUrl: "" }
 
@@ -77,7 +79,19 @@ export const LowCreditWarning = ({ message }: LowCreditWarningProps) => {
 					}}>
 					{t("kilocode:lowCreditWarning.addCredit")}
 				</VSCodeButton>
-				<FreeModelsLink className="p-1 w-full rounded mt-1" origin="chat" />
+				{!isOrganization && (
+					<VSCodeButtonLink
+						onClick={() => {
+							telemetryClient.capture(TelemetryEventName.CREATE_ORGANIZATION_LINK_CLICKED, {
+								origin: "low-credit-warning",
+							})
+						}}
+						href={getAppUrl("/organizations/new")}
+						appearance="primary"
+						className="p-1 w-full rounded">
+						{t("kilocode:lowCreditWarning.newOrganization")}
+					</VSCodeButtonLink>
+				)}
 			</div>
 		</>
 	)

webview-ui/src/i18n/locales/en/kilocode.json

@@ -11,7 +11,8 @@
 	},
 	"lowCreditWarning": {
 		"addCredit": "Add Credit",
-		"lowBalance": "Your Kilo Code balance is low"
+		"lowBalance": "Your Kilo Code balance is low",
+		"newOrganization": "Using Kilo for work?"
 	},
 	"memoryWarning": {
 		"message": "Memory usage is at {{percentage}}%. Please restart your IDE to prevent crashes."