feat: if specify no models in generate still try

quarylabs · Sep 11, 2024 · c59d95e · c59d95e
1 parent 24a6227
commit c59d95e
Show file tree

Hide file tree

Showing 19 changed files with 2,293 additions and 1,306 deletions.
diff --git a/js/packages/proto/src/generated/quary/service/v1/wasm_rust_rpc_calls.ts b/js/packages/proto/src/generated/quary/service/v1/wasm_rust_rpc_calls.ts
diff --git a/js/packages/quary-extension/src/web/chatEdit.ts b/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 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)
 }
diff --git a/js/packages/quary-extension/src/web/chatExplain.ts b/js/packages/quary-extension/src/web/chatExplain.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)
 }