feat: if specify no models in generate still try

Author: benfdking

js/packages/proto/src/generated/quary/service/v1/wasm_rust_rpc_calls.ts

@@ -1,6 +1,7 @@
 import * as vscode from 'vscode'
-import { isErr, Ok, Result } from '@shared/result'
+import { isErr, Result } from '@shared/result'
 import { getServices, preInitSetup } from './services'
+import { chatReturnFinalChat } from './chatHelpers'
 
 export const chatEdit = async (
   extensionContext: vscode.ExtensionContext,
@@ -43,19 +44,5 @@ export const chatEdit = async (
     modelName,
     prompt: request.prompt,
   })
-  if (isErr(prompt)) {
-    return prompt
-  }
-
-  const craftedPrompt = [
-    vscode.LanguageModelChatMessage.Assistant(prompt.value.agentPrompt),
-    vscode.LanguageModelChatMessage.User(prompt.value.userPrompt),
-  ]
-  const modelRequest = await model.sendRequest(craftedPrompt, {}, token)
-
-  for await (const fragment of modelRequest.text) {
-    stream.push(new vscode.ChatResponseMarkdownPart(fragment))
-  }
-
-  return Ok(undefined)
+  return chatReturnFinalChat(prompt, model, token, stream)
 }

js/packages/quary-extension/src/web/chatEdit.ts

@@ -1,6 +1,7 @@
 import * as vscode from 'vscode'
-import { isErr, Ok, Result } from '@shared/result'
+import { isErr, Result } from '@shared/result'
 import { getServices, preInitSetup } from './services'
+import { chatReturnFinalChat } from './chatHelpers'
 
 export const chatExplain = async (
   extensionContext: vscode.ExtensionContext,
@@ -41,16 +42,5 @@ export const chatExplain = async (
     modelName,
     userPrompt: request.prompt,
   })
-  if (isErr(prompt)) {
-    return prompt
-  }
-  const craftedPrompt = [
-    vscode.LanguageModelChatMessage.Assistant(prompt.value.agentPrompt),
-    vscode.LanguageModelChatMessage.User(prompt.value.userPrompt),
-  ]
-  const modelRequest = await model.sendRequest(craftedPrompt, {}, token)
-  for await (const fragment of modelRequest.text) {
-    stream.push(new vscode.ChatResponseMarkdownPart(fragment))
-  }
-  return Ok(undefined)
+  return chatReturnFinalChat(prompt, model, token, stream)
 }