Merge pull request #12 from VfBfoerst/master

added options for remote ollama api
twinnydotdev · Dec 27, 2023 · 5dbcec0 · 5dbcec0
2 parents a98d34a + 8697133
commit 5dbcec0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -65,6 +65,14 @@
         "twinny.ollamaModelName": {
           "type": "string",
           "default": "codellama:7b-code"
+        },
+        "twinny.ollamaBaseUrl": {
+          "type": "string",
+          "default": "localhost"
+        },
+        "twinny.ollamaApiPort": {
+          "type": "number",
+          "default": 11434
         }
       }
     }

diff --git a/src/completion.ts b/src/completion.ts
@@ -21,6 +21,8 @@ export class CompletionProvider implements InlineCompletionItemProvider {
   private _debounceWait = this._config.get('debounceWait') as number
   private _contextLength = this._config.get('contextLength') as number
   private _model = this._config.get('ollamaModelName') as string
+  private _baseurl = this._config.get('ollamaBaseUrl') as string
+  private _apiport = this._config.get('ollamaApiPort') as number
 
   constructor(statusBar: StatusBarItem) {
     this._statusBar = statusBar
@@ -68,8 +70,8 @@ export class CompletionProvider implements InlineCompletionItemProvider {
             this._statusBar.text = '$(loading~spin)'
             streamResponse(
               {
-                hostname: 'localhost',
-                port: 11434,
+                hostname: this._baseurl,
+                port: this._apiport,
                 method: 'POST',
                 path: '/api/generate'
               },