add language support for typescript (#78)

* add language support for typescript

* add more stop words to typescript
add-more-languages
Meng Zhang 2023-04-10 23:17:42 +08:00 committed by GitHub
parent 0ddc1bc087
commit 882fdbb61f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 0 deletions

View File

@ -22,4 +22,16 @@ LanguagePresets = {
Language.JAVASCRIPT: LanguagePreset( Language.JAVASCRIPT: LanguagePreset(
max_length=128, stop_words=["\n\n", "\nfunction", "\n//", "\nimport", "\nclass"] 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",
],
),
} }

View File

@ -14,6 +14,7 @@ class Language(str, Enum):
UNKNOWN = "unknown" UNKNOWN = "unknown"
PYTHON = "python" PYTHON = "python"
JAVASCRIPT = "javascript" JAVASCRIPT = "javascript"
TYPESCRIPT = "typescript"
class CompletionRequest(BaseModel): class CompletionRequest(BaseModel):