feat(completion): update userinfo and improve performance (#622)

* feat(webview): add copy functionality for user ID and improve metrics logging

* feat(logging): conditionally adjust log levels based on environment

* feat(completion): improve performance

* fix(webview-ui): handle encoding issues in display name for user info

* fix(webview-ui): add error handling to text decoding in useZgsmUserInfo

Author: mini2s

packages/telemetry/src/prometheus/MetricsSerializer.ts

@@ -45,15 +45,26 @@ export class MetricsSerializer {
 					}
 				} else if (metric instanceof Histogram) {
 					const sumMetrics = metricData.values.filter((v) => v.metricName?.endsWith("_sum"))
-
+					let HistogramLogs: { [key: string]: number[] } | null = null
 					sumMetrics.forEach((sumValue) => {
 						if (sumValue && sumValue.value > 0) {
 							metric.observe(sumValue.labels, sumValue.value)
-							this.logger.debug(
-								`[MetricsSerializer] Histogram sum for ${metricData.name} loaded with value: ${sumValue.value}`,
-							)
+							if (HistogramLogs === null) {
+								HistogramLogs = {}
+							}
+							if (HistogramLogs[metricData.name]) {
+								HistogramLogs[metricData.name].push(sumValue.value)
+							} else {
+								HistogramLogs[metricData.name] = [sumValue.value]
+							}
 						}
 					})
+					// eslint-disable-next-line @typescript-eslint/no-unused-expressions
+					HistogramLogs && Object.keys(HistogramLogs).forEach(name => {						
+						this.logger.debug(
+							`[MetricsSerializer] Histogram sum for ${name} loaded with value: ${HistogramLogs![name].join()}`,
+						)
+					})
 				} else if (metric instanceof Gauge) {
 					for (const item of metricData.values) {
 						metric.set(item.labels, item.value)

src/api/providers/zgsm.ts

@@ -195,8 +195,9 @@ export class ZgsmAiHandler extends BaseProvider implements SingleCompletionHandl
 				if (this.options.zgsmModelId === autoModeModelId) {
 					selectedLlm = response.headers.get("x-select-llm") || ""
 					selectReason = response.headers.get("x-select-reason") || ""
+					const isDev = process.env.NODE_ENV === "development"
 
-					const userInputHeader = response.headers.get("x-user-input")
+					const userInputHeader = isDev ? response.headers.get("x-user-input") : null
 					if (userInputHeader) {
 						const decodedUserInput = Buffer.from(userInputHeader, "base64").toString("utf-8")
 						this.logger.info(`[x-user-input]: ${decodedUserInput}`)
@@ -431,6 +432,7 @@ export class ZgsmAiHandler extends BaseProvider implements SingleCompletionHandl
 		const contentBuffer: string[] = []
 		let time = Date.now()
 		let isPrinted = false
+		const isDev = process.env.NODE_ENV === "development"
 
 		// Yield selected LLM info if available (for Auto model mode)
 		if (selectedLlm && this.options.zgsmModelId === autoModeModelId) {
@@ -447,7 +449,7 @@ export class ZgsmAiHandler extends BaseProvider implements SingleCompletionHandl
 			// Cache content for batch processing
 			if (delta.content) {
 				contentBuffer.push(delta.content)
-				if (!isPrinted && chunk.model && this.options.zgsmModelId === autoModeModelId) {
+				if (isDev && !isPrinted && chunk.model && this.options.zgsmModelId === autoModeModelId) {
 					this.logger.info(`[Current Model]: ${chunk.model}`)
 					isPrinted = true
 				}

src/core/costrict/codebase-index/workspace-event-monitor.ts

@@ -147,7 +147,7 @@ export class WorkspaceEventMonitor {
 	 * Handle VSCode close event
 	 */
 	public handleVSCodeClose() {
-		this.log.info("[WorkspaceEventMonitor] VSCode close event detected")
+		this.log.debug("[WorkspaceEventMonitor] VSCode close event detected")
 
 		// Send workspace close events
 		ZgsmCodebaseIndexManager.getInstance().client?.publishSyncWorkspaceEvents({
@@ -170,7 +170,7 @@ export class WorkspaceEventMonitor {
 	 * Destroy event monitor
 	 */
 	public async dispose() {
-		this.log.info("[WorkspaceEventMonitor] Starting to destroy event monitor")
+		this.log.debug("[WorkspaceEventMonitor] Starting to destroy event monitor")
 		// Cancel timers
 		if (this.flushTimer) {
 			clearTimeout(this.flushTimer)
@@ -183,15 +183,15 @@ export class WorkspaceEventMonitor {
 
 		// Clean up document content cache
 		this.documentContentCache.clear()
-		this.log.info("[WorkspaceEventMonitor] Document content cache cleared")
+		this.log.debug("[WorkspaceEventMonitor] Document content cache cleared")
 
 		// Send remaining events
 		if (this.eventBuffer.size > 0) {
 			await this.flushEventsSync()
 		}
 
 		this.isInitialized = false
-		this.log.info("[WorkspaceEventMonitor] Event monitor disposed")
+		this.log.debug("[WorkspaceEventMonitor] Event monitor disposed")
 	}
 
 	/**
@@ -420,7 +420,7 @@ export class WorkspaceEventMonitor {
 		} else if (oldHash !== nowHash) {
 			this.documentContentCache.set(filePath, { contentHash: nowHash })
 		} else {
-			this.log.info(`[WorkspaceEventMonitor] Document content unchanged, skipping event trigger`)
+			this.log.debug(`[WorkspaceEventMonitor] Document content unchanged, skipping event trigger`)
 			return
 		}
 

src/core/costrict/completion/CompletionClient.ts

@@ -73,6 +73,7 @@ export class CompletionClient {
 
 			throw new Error(OPENAI_CLIENT_NOT_INITIALIZED)
 		}
+		// console.time("Completion response timeout");
 
 		try {
 			const response = await client.doCallApi(cp, scores, latestCompletion)
@@ -106,6 +107,7 @@ export class CompletionClient {
 			if (client) {
 				client.reqs.delete(cp.id)
 			}
+			// console.timeEnd("Completion response timeout");
 		}
 	}
 
@@ -155,7 +157,8 @@ export class CompletionClient {
 				...COSTRICT_DEFAULT_HEADERS,
 				"X-Request-ID": uuidv7(),
 			},
-			timeout: 4500
+			timeout: 5000,
+			maxRetries: 0,
 		})
 
 		if (!this.openai) {

src/utils/logger.ts

@@ -82,7 +82,7 @@ class ChannelLogger implements ILogger {
 	) {
 		// Reuse externally injected OutputChannel; create one if not provided
 		this.channel = opts.channel ?? vscode.window.createOutputChannel(name)
-		this.level = opts.level ?? LogLevel.Debug
+		this.level = opts.level ?? (process.env.NODE_ENV === "development" ? LogLevel.Debug : LogLevel.Info)
 		this.enabled = opts.enabled ?? true
 		this.timeFn = opts.timeFn ?? (() => new Date().toLocaleString())
 	}

webview-ui/src/components/cloud/ZgsmAccountView.tsx

@@ -11,6 +11,9 @@ import { telemetryClient } from "@src/utils/TelemetryClient"
 import { useEvent } from "react-use"
 import { ExtensionMessage } from "@roo/ExtensionMessage"
 import { useZgsmUserInfo } from "@src/hooks/useZgsmUserInfo"
+import { useCopyToClipboard } from "@src/utils/clipboard"
+import { ClipboardCopy } from "lucide-react"
+import { StandardTooltip } from "../ui"
 
 type AccountViewProps = {
 	apiConfiguration?: ProviderSettings
@@ -239,12 +242,12 @@ const QuotaInfoDisplay = memo(
 
 const ZgsmAccountViewComponent = ({ apiConfiguration, onDone }: AccountViewProps) => {
 	const { t } = useAppTranslation()
+	const { copyWithFeedback } = useCopyToClipboard()
 	const [quotaInfo, setQuotaInfo] = useState<QuotaInfo>()
 	const [inviteCodeInfo, setInviteCodeInfo] = useState<InviteCodeInfo>()
 	const [showQuotaInfo, setShowQuotaInfo] = useState(false)
 	const [isLoadingQuota, setIsLoadingQuota] = useState(false)
 	const { userInfo, logoPic, hash } = useZgsmUserInfo(apiConfiguration?.zgsmAccessToken)
-	console.log("New Credit hash: ", hash)
 
 	// Cache static resource URI
 	const coLogoUri = useMemo(() => (window as any).COSTRICT_BASE_URI + "/logo.svg", [])
@@ -392,7 +395,17 @@ const ZgsmAccountViewComponent = ({ apiConfiguration, onDone }: AccountViewProps
 								<p className="text-xs text-vscode-descriptionForeground mb-1">{userInfo?.email}</p>
 							)}
 							{userInfo.id && (
-								<h2 className="text-xs text-vscode-descriptionForeground mb-1">ID: {userInfo.id}</h2>
+								<h2 className="text-xs text-vscode-descriptionForeground mb-1 flex items-center gap-1 whitespace-nowrap">
+									ID: {userInfo.id}
+									<StandardTooltip content={t("common:mermaid.buttons.copy")}>
+												<ClipboardCopy
+													onClick={(e) => {
+														e.stopPropagation()
+														copyWithFeedback(userInfo.id || "")
+													}}
+													aria-label="Copy message icon" className="cursor-pointer w-[14px] -translate-y-0.5" />
+									</StandardTooltip>
+								</h2>
 							)}
 							{/* Star status card */}
 							{quotaInfo?.is_star != null && (

webview-ui/src/hooks/__tests__/useZgsmUserInfo.spec.ts

@@ -86,7 +86,7 @@ describe("useZgsmUserInfo", () => {
 
 		await waitFor(() => {
 			expect(result.current.userInfo).toEqual({
-				id: "user123",
+				id: undefined,
 				name: "testuser",
 				picture: "https://example.com/avatar.png",
 				email: "[email protected]",

webview-ui/src/hooks/useZgsmUserInfo.ts

@@ -106,12 +106,17 @@ export function useZgsmUserInfo(tokenOrConfig?: string | ProviderSettings): Zgsm
 			try {
 				const parsedJwt = parseJwt(token)
 
+				const id = parsedJwt.universal_id
 				const userInfo: ZgsmUserInfo = {
-					id: parsedJwt.universal_id || parsedJwt.id,
-					name: parsedJwt?.properties?.oauth_GitHub_username || parsedJwt.id || parsedJwt.phone,
+					id,
+					name:
+						parsedJwt?.properties?.oauth_GitHub_username ||
+						fixEncoding(parsedJwt.displayName) ||
+						parsedJwt.phone ||
+						(id ? `user_${id}`.slice(0, 10) : id),
 					picture: parsedJwt.avatar || parsedJwt?.properties?.oauth_GitHub_avatarUrl,
 					email: parsedJwt.email || parsedJwt?.properties?.oauth_GitHub_email,
-					phone: parsedJwt.phone,
+					phone: parsedJwt.phone || parsedJwt.phone_number,
 					organizationName: parsedJwt.organizationName,
 					organizationImageUrl: parsedJwt.organizationImageUrl,
 				}
@@ -152,3 +157,18 @@ export function useZgsmUserInfo(tokenOrConfig?: string | ProviderSettings): Zgsm
 
 	return data
 }
+
+function fixEncoding(name?: string) {
+	try {
+		const misencodedPattern = /[\x80-\xFF]+/
+		if (!name || !misencodedPattern.test(name)) {
+			return name
+		}
+
+		const bytes = Array.from(name, (c) => c.charCodeAt(0))
+		return new TextDecoder("utf-8").decode(new Uint8Array(bytes))
+	} catch (error) {
+		console.error(error)
+		return name
+	}
+}