feat: Enhance user experience and functionality

- Bump extension version to 1.74.0+1
- Add support for custom commit templates using `commit_template.md` in `.cody/configs/`
- Implement visual confirmation with a checkmark icon for the copy code block button in chat, with a 5-second reset timer
- Enable local context for Enterprise Starter users and add Enterprise Starter development workspace URL
- Fix temperature application in chat and workflow functionalities

Author: PriNova

lib/shared/src/chat/chat.ts

@@ -87,7 +87,7 @@ export class ChatClient {
             messages = messages.concat([{ speaker: 'assistant' }])
         }
 
-        DEFAULT_CHAT_COMPLETION_PARAMETERS.temperature = this.temperature
+        DEFAULT_CHAT_COMPLETION_PARAMETERS.temperature = await this.getTemperature()
         const completionParams = {
             ...DEFAULT_CHAT_COMPLETION_PARAMETERS,
             ...params,

lib/shared/src/sourcegraph-api/environments.ts

@@ -43,6 +43,7 @@ export const DOTCOM_WORKSPACE_UPGRADE_URL = new URL('https://sourcegraph.com/cod
 
 const Workspaces_Host_Prod = '.sourcegraph.app'
 const Workspaces_Host_Dev = '.sourcegraphapp.test:3443'
+const Workspaces_Host_ES_DEV = '.sourcegraphdev.app'
 
 // 🚨 SECURITY: This is used to validate a set of URLs we will allow to be passed in
 //              to the editor in the URL handler.
@@ -56,7 +57,8 @@ export function isWorkspaceInstance(arg: Pick<AuthStatus, 'endpoint'> | undefine
     try {
         return (
             new URL(url).host.endsWith(Workspaces_Host_Prod) ||
-            new URL(url).host.endsWith(Workspaces_Host_Dev)
+            new URL(url).host.endsWith(Workspaces_Host_Dev) ||
+            new URL(url).host.endsWith(Workspaces_Host_ES_DEV)
         )
     } catch {
         return false

vscode/package.json

@@ -3,7 +3,7 @@
   "name": "cody-ai",
   "private": true,
   "displayName": "Cody: AI Code Assistant",
-  "version": "1.74.0+0",
+  "version": "1.74.0+1",
   "publisher": "sourcegraph",
   "license": "Apache-2.0",
   "icon": "resources/sourcegraph.png",

vscode/src/chat/initialContext.ts

@@ -4,7 +4,6 @@ import {
     ContextItemSource,
     type ContextItemTree,
     type DefaultContext,
-    FeatureFlag,
     REMOTE_REPOSITORY_PROVIDER_URI,
     abortableOperation,
     authStatus,
@@ -16,7 +15,6 @@ import {
     displayPathBasename,
     distinctUntilChanged,
     expandToLineRange,
-    featureFlagProvider,
     fromVSCodeEvent,
     isDotCom,
     isError,
@@ -51,7 +49,7 @@ export function observeDefaultContext({
         getCurrentFileOrSelection({ chatBuilder }).pipe(distinctUntilChanged()),
         getCorpusContextItemsForEditorState().pipe(distinctUntilChanged()),
         getOpenCtxContextItems().pipe(distinctUntilChanged()),
-        featureFlagProvider.evaluatedFeatureFlag(FeatureFlag.NoDefaultRepoChip)
+        Observable.of(false)
     ).pipe(
         debounceTime(50),
         map(
@@ -210,9 +208,8 @@ export function getCorpusContextItemsForEditorState(
     return combineLatest(relevantAuthStatus, remoteReposForAllWorkspaceFolders).pipe(
         abortableOperation(async ([authStatus, remoteReposForAllWorkspaceFolders], signal) => {
             const items: ContextItem[] = []
-
             // Local context is not available to enterprise users
-            if (!authStatus.isEnterpriseUser) {
+            if (!authStatus.isEnterpriseUser && authStatus.isEnterpriseStarterUser) {
                 // TODO(sqs): Support multi-root. Right now, this only supports the 1st workspace root.
                 const workspaceFolder = vscode.workspace.workspaceFolders?.at(0)
 

vscode/src/commands/scm/prompts.ts

@@ -1,4 +1,6 @@
-import { ps } from '@sourcegraph/cody-shared'
+import * as path from 'node:path'
+import { PromptString, ps } from '@sourcegraph/cody-shared'
+import * as vscode from 'vscode'
 
 const COMMIT_INTRO = ps`Review the following git command output to understand the changes you are about to generate a commit message for.`
 
@@ -29,3 +31,32 @@ export const COMMIT_COMMAND_PROMPTS = {
      */
     noTemplate: COMMMIT_TEMPLATE_NOT_FOUNT,
 }
+
+export async function getCustomCommitTemplate(): Promise<PromptString | undefined> {
+    try {
+        const workspaceFolders = vscode.workspace.workspaceFolders?.[0]
+        if (!workspaceFolders) {
+            return undefined
+        }
+
+        const commitTemplatePath = path.join(
+            workspaceFolders.uri.path,
+            '.cody',
+            'configs',
+            'commit_template.md'
+        )
+
+        const fileUri = vscode.Uri.file(commitTemplatePath)
+        const content = await vscode.workspace.fs.readFile(fileUri)
+        const commitTemplate = Buffer.from(content).toString('utf-8')
+
+        // Return undefined if content is empty or only whitespace
+        if (!commitTemplate.trim()) {
+            return undefined
+        }
+
+        return PromptString.unsafe_fromLLMResponse(commitTemplate)
+    } catch {
+        return undefined
+    }
+}

vscode/src/commands/scm/source-control.ts

@@ -20,7 +20,7 @@ import * as vscode from 'vscode'
 import { PromptBuilder } from '../../prompt-builder'
 import type { API, GitExtension, InputBox, Repository } from '../../repository/builtinGitExtension'
 import { getContextFilesFromGitApi as getContext } from '../context/git-api'
-import { COMMIT_COMMAND_PROMPTS } from './prompts'
+import { COMMIT_COMMAND_PROMPTS, getCustomCommitTemplate } from './prompts'
 
 export class CodySourceControl implements vscode.Disposable {
     private disposables: vscode.Disposable[] = []
@@ -228,9 +228,9 @@ export class CodySourceControl implements vscode.Disposable {
             throw new Error('Failed to get git output.')
         }
 
-        const templatePrompt = this.commitTemplate
-            ? COMMIT_COMMAND_PROMPTS.template
-            : COMMIT_COMMAND_PROMPTS.noTemplate
+        const templateMessage = await getCustomCommitTemplate()
+        const customTemplate = templateMessage ? templateMessage : COMMIT_COMMAND_PROMPTS.template
+        const templatePrompt = this.commitTemplate ? COMMIT_COMMAND_PROMPTS.noTemplate : customTemplate
         const text = COMMIT_COMMAND_PROMPTS.instruction.replace('{COMMIT_TEMPLATE}', templatePrompt)
         const transcript: ChatMessage[] = [{ speaker: 'human', text }]
 

vscode/src/workflow/workflow-executor.ts

@@ -536,8 +536,6 @@ async function executeLLMNode(
 ): Promise<string> {
     abortSignal?.throwIfAborted()
     const oldTemperature = await chatClient.getTemperature()
-    await chatClient.setTemperature((node as LLMNode).data.temperature)
-
     const inputs = combineParentOutputsByConnectionOrder(node.id, context).map(input =>
         sanitizeForPrompt(input)
     )
@@ -565,6 +563,7 @@ async function executeLLMNode(
         ]
 
         const streamPromise = new Promise<string>((resolve, reject) => {
+            chatClient.setTemperature((node as LLMNode).data.temperature)
             // Use the AsyncGenerator correctly
             chatClient
                 .chat(
@@ -580,6 +579,7 @@ async function executeLLMNode(
                     abortSignal
                 )
                 .then(async stream => {
+                    chatClient.setTemperature(oldTemperature)
                     const accumulated = new StringBuilder()
                     //let chunksProcessed = 0
                     try {
@@ -605,7 +605,6 @@ async function executeLLMNode(
                 })
                 .catch(reject)
         })
-        await chatClient.setTemperature(oldTemperature)
         return await Promise.race([streamPromise, timeout])
     } catch (error) {
         await chatClient.setTemperature(oldTemperature)

vscode/webviews/chat/cells/messageCell/assistant/AssistantMessageCell.tsx

@@ -19,6 +19,7 @@ import isEqual from 'lodash/isEqual'
 import { type FunctionComponent, type RefObject, memo, useMemo } from 'react'
 import type { ApiPostMessage, UserAccountInfo } from '../../../../Chat'
 import { chatModelIconComponent } from '../../../../components/ChatModelIcon'
+import { CheckCodeBlockIcon } from '../../../../icons/CodeBlockActionIcons'
 import { useOmniBox } from '../../../../utils/useOmniBox'
 import {
     ChatMessageContent,
@@ -175,11 +176,23 @@ export const AssistantMessageCell: FunctionComponent<{
                                         <button
                                             type="button"
                                             className="tw-flex tw-items-center tw-gap-2 tw-text-sm tw-text-muted-foreground hover:tw-text-foreground"
-                                            onClick={() => {
+                                            onClick={event => {
+                                                const button = event.currentTarget
+                                                const originalContent = button.innerHTML
+
+                                                // Change to check icon when clicked
+                                                button.innerHTML = CheckCodeBlockIcon
+
+                                                // Copy text to clipboard
                                                 navigator.clipboard.writeText(
                                                     message.text?.toString() || ''
                                                 )
                                                 copyButtonOnSubmit?.(message.text?.toString() || '')
+
+                                                // Reset after 5 seconds
+                                                setTimeout(() => {
+                                                    button.innerHTML = originalContent
+                                                }, 5000)
                                             }}
                                             title="Copy message to clipboard"
                                         >