Merge pull request #1216 from vespa-engine/thomasht86/add-missing-tags

Thomasht86/add missing tags

Author: thomasht86

vespa/configuration/deployment.pyi

@@ -1,21 +1,21 @@
 from vespa.configuration.vt import VT
 
-def region(*c, **kwargs) -> VT: ...
-def steps(*c, **kwargs) -> VT: ...
-def instance(*c, **kwargs) -> VT: ...
+def allow(*c, **kwargs) -> VT: ...
 def bcp(*c, **kwargs) -> VT: ...
-def endpoint(*c, **kwargs) -> VT: ...
-def staging(*c, **kwargs) -> VT: ...
-def nodes(*c, **kwargs) -> VT: ...
 def block_change(*c, **kwargs) -> VT: ...
+def delay(*c, **kwargs) -> VT: ...
+def deployment(*c, **kwargs) -> VT: ...
 def dev(*c, **kwargs) -> VT: ...
+def endpoint(*c, **kwargs) -> VT: ...
+def endpoints(*c, **kwargs) -> VT: ...
+def group(*c, **kwargs) -> VT: ...
+def instance(*c, **kwargs) -> VT: ...
+def nodes(*c, **kwargs) -> VT: ...
 def parallel(*c, **kwargs) -> VT: ...
-def allow(*c, **kwargs) -> VT: ...
 def prod(*c, **kwargs) -> VT: ...
-def deployment(*c, **kwargs) -> VT: ...
+def region(*c, **kwargs) -> VT: ...
+def staging(*c, **kwargs) -> VT: ...
+def steps(*c, **kwargs) -> VT: ...
+def test_(*c, **kwargs) -> VT: ...
 def tester(*c, **kwargs) -> VT: ...
 def upgrade(*c, **kwargs) -> VT: ...
-def group(*c, **kwargs) -> VT: ...
-def test_(*c, **kwargs) -> VT: ...
-def delay(*c, **kwargs) -> VT: ...
-def endpoints(*c, **kwargs) -> VT: ...

vespa/configuration/generate_pyi.py

@@ -14,5 +14,5 @@
     with open(pyi_file, "w") as f:
         # write import statement for VT
         f.write("from vespa.configuration.vt import VT\n\n")
-        for tag in tag_set:
+        for tag in sorted(tag_set):
             f.write(f"def {tag}(*c, **kwargs) -> VT: ...\n")

vespa/configuration/query_profiles.pyi

@@ -1,10 +1,10 @@
 from vespa.configuration.vt import VT
 
-def dimensions(*c, **kwargs) -> VT: ...
-def strict(*c, **kwargs) -> VT: ...
-def query_profile_type(*c, **kwargs) -> VT: ...
 def description(*c, **kwargs) -> VT: ...
+def dimensions(*c, **kwargs) -> VT: ...
+def field(*c, **kwargs) -> VT: ...
+def match_(*c, **kwargs) -> VT: ...
 def query_profile(*c, **kwargs) -> VT: ...
+def query_profile_type(*c, **kwargs) -> VT: ...
 def ref(*c, **kwargs) -> VT: ...
-def match_(*c, **kwargs) -> VT: ...
-def field(*c, **kwargs) -> VT: ...
+def strict(*c, **kwargs) -> VT: ...

vespa/configuration/services.py

@@ -15,6 +15,8 @@
     "admin",
     "adminserver",
     "age",
+    "api-key-secret-ref",
+    "batching",
     "binding",
     "bucket-splitting",
     "cache",
@@ -37,6 +39,7 @@
     "container",
     "content",
     "coverage",
+    "dimensions",
     "disk",
     "disk-limit-factor",
     "diskbloatfactor",
@@ -49,6 +52,7 @@
     "document-token-id",
     "documentprocessor",
     "documents",
+    "endpoint",
     "engine",
     "environment-variables",
     "execution-mode",
@@ -104,12 +108,14 @@
     "min-wait-after-coverage-factor",
     "minimum",
     "model",
+    "model-config-override",
     "model-evaluation",
     "models",
     "native",
     "niceness",
     "node",
     "nodes",
+    "normalize",
     "onnx",
     "onnx-execution-mode",
     "onnx-gpu-device",
@@ -124,6 +130,7 @@
     "proton",
     "provider",
     "prune",
+    "quantization",
     "query",
     "query-timeout",
     "query-token-id",
@@ -157,6 +164,7 @@
     "timeout",
     "token",
     "tokenizer-model",
+    "tokenizer-vocab",
     "top-k-probability",
     "total",
     "tracelevel",
@@ -169,7 +177,9 @@
     "transformer-output",
     "transformer-pad-token",
     "transformer-start-sequence-token",
+    "transformer-token-type-ids",
     "transition-time",
+    "truncate",
     "tuning",
     "type",
     "unpack",