add language support for typescript (#78)

* add language support for typescript * add more stop words to typescript
2023-04-10 23:17:42 +08:00 · 2023-04-10 23:17:42 +08:00 · 882fdbb61f
parent 0ddc1bc087
commit 882fdbb61f
2 changed files with 13 additions and 0 deletions
--- 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",
+        ],
+    ),
 }
--- 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):