From 882fdbb61f305c5efb2473284eac4d63c3d57dad Mon Sep 17 00:00:00 2001
From: Meng Zhang <wsxiaoys.lh@gmail.com>
Date: Mon, 10 Apr 2023 23:17:42 +0800
Subject: [PATCH] add language support for typescript (#78)

* add language support for typescript

* add more stop words to typescript
---
 tabby/server/backend/language_presets.py | 12 ++++++++++++
 tabby/server/models.py                   |  1 +
 2 files changed, 13 insertions(+)

diff --git a/tabby/server/backend/language_presets.py b/tabby/server/backend/language_presets.py
index 0e8951e..04fa844 100644
--- a/tabby/server/backend/language_presets.py
+++ b/tabby/server/backend/language_presets.py
@@ -22,4 +22,16 @@ LanguagePresets = {
     Language.JAVASCRIPT: LanguagePreset(
         max_length=128, stop_words=["\n\n", "\nfunction", "\n//", "\nimport", "\nclass"]
     ),
+    Language.TYPESCRIPT: LanguagePreset(
+        max_length=128,
+        stop_words=[
+            "\n\n",
+            "\nfunction",
+            "\n//",
+            "\nimport",
+            "\nclass",
+            "\ninterface",
+            "\ntype",
+        ],
+    ),
 }
diff --git a/tabby/server/models.py b/tabby/server/models.py
index 44bcf7f..8c68332 100644
--- a/tabby/server/models.py
+++ b/tabby/server/models.py
@@ -14,6 +14,7 @@ class Language(str, Enum):
     UNKNOWN = "unknown"
     PYTHON = "python"
     JAVASCRIPT = "javascript"
+    TYPESCRIPT = "typescript"
 
 
 class CompletionRequest(BaseModel):