fix get fim provider, i broke it :)

twinnydotdev · Dec 17, 2024 · 2b88958 · 2b88958
1 parent 4f446ef
commit 2b88958
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/extension/base.ts b/src/extension/base.ts
@@ -3,6 +3,7 @@ import * as vscode from "vscode"
 import {
   ACTIVE_CHAT_PROVIDER_STORAGE_KEY,
   ACTIVE_EMBEDDINGS_PROVIDER_STORAGE_KEY,
+  ACTIVE_FIM_PROVIDER_STORAGE_KEY,
   EVENT_NAME,
   EXTENSION_CONTEXT_NAME
 } from "../common/constants"
@@ -29,6 +30,13 @@ export class Base {
     })
   }
 
+  public getFimProvider = () => {
+    const provider = this.context?.globalState.get<TwinnyProvider>(
+      ACTIVE_FIM_PROVIDER_STORAGE_KEY
+    )
+    return provider
+  }
+
   public getProvider = () => {
     const provider = this.context?.globalState.get<TwinnyProvider>(
       ACTIVE_CHAT_PROVIDER_STORAGE_KEY

diff --git a/src/extension/providers/completion.ts b/src/extension/providers/completion.ts
@@ -58,7 +58,7 @@ import { createStreamRequestBodyFim } from "../provider-options"
 import { TemplateProvider } from "../template-provider"
 import {
   getCurrentLineText,
-  getFimDataFromProvider as getProviderFimData,
+  getFimDataFromProvider as getFimProviderFimData,
   getIsMiddleOfString,
   getIsMultilineCompletion,
   getPrefixSuffix,
@@ -135,7 +135,7 @@ export class CompletionProvider
     context: InlineCompletionContext
   ): Promise<InlineCompletionItem[] | InlineCompletionList | null | undefined> {
     const editor = window.activeTextEditor
-    this._provider = this.getProvider()
+    this._provider = this.getFimProvider()
     const isLastCompletionAccepted =
       this._acceptedLastCompletion && !this.config.enableSubsequentCompletions
 
@@ -200,7 +200,7 @@ export class CompletionProvider
     return new Promise<ResolvedInlineCompletion>((resolve, reject) => {
       this._debouncer = setTimeout(() => {
         this._lock.acquire("twinny.completion", async () => {
-          const provider = this.getProvider()
+          const provider = this.getFimProvider()
           if (!provider) return
           const request = this.buildFimRequest(prompt, provider)
 
@@ -256,7 +256,7 @@ export class CompletionProvider
     )
 
     try {
-      const providerFimData = getProviderFimData(this._provider.provider, data)
+      const providerFimData = getFimProviderFimData(this._provider.provider, data)
       if (providerFimData === undefined) return ""
 
       this._completion = this._completion + providerFimData