fix(lens): use { args: ... } in useQuery hooks for lens-react API

lens-react useQuery hooks also expect { args: {...} } pattern like
vanilla client calls. Fixed all useQuery calls in React hooks.

Changes:
- useCurrentSession.ts: getSession.useQuery uses { args: ... }
- useMessages.ts: listMessages.useQuery uses { args: ... }
- useModelDetails.ts: getModelDetails/getTokenizerInfo use { args: ... }
- useModels.ts: fetchModels.useQuery uses { args: ... }
- useAIConfig.ts: loadConfig.useQuery uses { args: ... }
- headless.tsx: getSession.useQuery uses { args: ... }
- useInputState.ts: getRecentUserMessages.useQuery uses { args: ... }
- session-query.ts: removed debug logging

Author: shtse8

packages/code-core/src/database/session/session-query.ts

@@ -25,7 +25,6 @@ export async function getSessionById(
 		return null;
 	}
 
-
 	// Get messages with all parts, attachments, usage
 	const sessionMessages = await getSessionMessages(db, sessionId);
 

packages/code/src/headless.tsx

@@ -48,7 +48,7 @@ function HeadlessApp({ prompt, options }: HeadlessProps) {
 
 	// Query hooks - use .useQuery() for React hook pattern
 	const sessionQuery = client.getSession.useQuery({
-		input: { id: sessionId || "" },
+		args: { id: sessionId || "" },
 		skip: !sessionId,
 	});
 

packages/code/src/hooks/client/useAIConfig.ts

@@ -18,7 +18,7 @@ export function useAIConfig() {
 	const client = useLensClient();
 
 	// Query hook - auto-loads config on mount
-	const configQuery = client.loadConfig.useQuery({ input: {} });
+	const configQuery = client.loadConfig.useQuery({ args: {} });
 
 	// Mutation hook - for saving config
 	const { mutate: saveConfigMutate } = client.saveConfig.useMutation();

packages/code/src/hooks/client/useCurrentSession.ts

@@ -80,7 +80,7 @@ export function useCurrentSession() {
 	// Use lens-react query hook for session data
 	// The status field uses .subscribe() in the resolver, so Lens auto-streams
 	const { data: session, loading, error, refetch } = client.getSession.useQuery({
-		input: { id: currentSessionId || "" },
+		args: { id: currentSessionId || "" },
 		skip,
 		debug: DEBUG ? {
 			onData: (data: SessionWithStatus | null) => {

packages/code/src/hooks/client/useMessages.ts

@@ -99,7 +99,7 @@ export function useMessages(sessionId: string | null | undefined) {
 	// Use lens-react query hook
 	// Steps/parts fields use .subscribe() in resolver, so Lens auto-streams
 	const messagesQuery = client.listMessages.useQuery({
-		input: { sessionId: sessionId || "" },
+		args: { sessionId: sessionId || "" },
 		skip: !sessionId,
 	});
 

packages/code/src/hooks/client/useModelDetails.ts

@@ -45,12 +45,12 @@ export function useModelDetails(providerId: string | null, modelId: string | nul
 
 	// Query hooks - reactive data fetching
 	const modelDetailsQuery = client.getModelDetails.useQuery({
-		input: { providerId: providerId || "", modelId: modelId || "" },
+		args: { providerId: providerId || "", modelId: modelId || "" },
 		skip: skipQueries,
 	});
 
 	const tokenizerQuery = client.getTokenizerInfo.useQuery({
-		input: { model: modelId || "" },
+		args: { model: modelId || "" },
 		skip: skipQueries,
 	});
 

packages/code/src/hooks/client/useModels.ts

@@ -19,7 +19,7 @@ export function useModels(providerId: string | null) {
 	const client = useLensClient();
 
 	const { data, loading, error, refetch } = client.fetchModels.useQuery({
-		input: { providerId: providerId ?? "" },
+		args: { providerId: providerId ?? "" },
 		skip: !providerId, // Skip when no provider selected
 	}) as {
 		data: { success: boolean; models: ModelInfo[]; error?: string } | null;

packages/code/src/screens/chat/hooks/useInputState.ts

@@ -54,7 +54,7 @@ export function useInputState(): InputState {
 	// Query hook for message history
 	// Use .useQuery() for React hook pattern (not vanilla Promise call)
 	const historyQuery = client.getRecentUserMessages.useQuery({
-		input: { limit: 100 },
+		args: { limit: 100 },
 	});
 
 	// Sync query data to global state