fix indentation issue

sourcegraph · Nov 9, 2024 · f3f897c · f3f897c
1 parent 5711fc8
commit f3f897c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 17 deletions.
diff --git a/vscode/src/autoedits/adapters/cody-gateway.ts b/vscode/src/autoedits/adapters/cody-gateway.ts
@@ -8,7 +8,6 @@ import { autoeditsLogger } from '../logger'
 import type { AutoeditsModelAdapter, ChatPrompt, PromptResponseData } from '../prompt-provider'
 import { getModelResponse } from '../prompt-provider'
 import { type CodeToReplaceData, SYSTEM_PROMPT, getBaseUserPrompt } from '../prompt-utils'
-import * as utils from '../utils'
 
 export class CodyGatewayAdapter implements AutoeditsModelAdapter {
     getPrompt(
@@ -39,13 +38,7 @@ export class CodyGatewayAdapter implements AutoeditsModelAdapter {
     }
 
     postProcessResponse(codeToReplace: CodeToReplaceData, response: string): string {
-        // TODO(hitesh): The finetuned model is messing up the indentation of the first line. [Issue](https://linear.app/sourcegraph/issue/CODY-4266/fix-the-indentation-issue-with-the-fine-tuned-models)
-        // Correct it manually for now, by checking the first line of the code to rewrite and adding the same indentation to the first line of the completion
-        const fixedIndentationResponse = utils.fixFirstLineIndentation(
-            codeToReplace.codeToRewrite,
-            response
-        )
-        return fixedIndentationResponse
+        return response
     }
 
     async getModelResponse(

diff --git a/vscode/src/autoedits/adapters/fireworks.ts b/vscode/src/autoedits/adapters/fireworks.ts
@@ -8,7 +8,6 @@ import { autoeditsLogger } from '../logger'
 import type { AutoeditsModelAdapter, ChatPrompt, PromptResponseData } from '../prompt-provider'
 import { getModelResponse } from '../prompt-provider'
 import { type CodeToReplaceData, SYSTEM_PROMPT, getBaseUserPrompt } from '../prompt-utils'
-import * as utils from '../utils'
 
 export class FireworksAdapter implements AutoeditsModelAdapter {
     getPrompt(
@@ -42,13 +41,7 @@ export class FireworksAdapter implements AutoeditsModelAdapter {
     }
 
     postProcessResponse(codeToReplace: CodeToReplaceData, response: string): string {
-        // TODO(hitesh): The finetuned model is messing up the indentation of the first line. [Issue](https://linear.app/sourcegraph/issue/CODY-4266/fix-the-indentation-issue-with-the-fine-tuned-models)
-        // Correct it manually for now, by checking the first line of the code to rewrite and adding the same indentation to the first line of the completion
-        const fixedIndentationResponse = utils.fixFirstLineIndentation(
-            codeToReplace.codeToRewrite,
-            response
-        )
-        return fixedIndentationResponse
+        return response
     }
 
     async getModelResponse(

diff --git a/vscode/src/autoedits/autoedits-provider.ts b/vscode/src/autoedits/autoedits-provider.ts
@@ -251,7 +251,7 @@ export class AutoeditsProvider implements vscode.InlineCompletionItemProvider, v
         autoeditsLogger.logDebug(
             'Autoedits',
             '========================== Response:\n',
-            JSON.stringify(prediction, null, 2),
+            prediction.prediction,
             '\n',
             '========================== Time Taken For LLM (Msec): ',
             (Date.now() - start).toString(),