regenerated models

impresso · Oct 23, 2024 · 1661253 · 1661253
1 parent 67b1658
commit 1661253
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/impresso/api_client/models/impresso_named_entity_recognition_request.py b/impresso/api_client/models/impresso_named_entity_recognition_request.py
@@ -14,9 +14,9 @@ class ImpressoNamedEntityRecognitionRequest:
 
     Attributes:
         text (str): Text to be processed for named entity recognition
-        method (Union[Unset, ImpressoNamedEntityRecognitionRequestMethod]): NER method to be used: `ner` (default) or
-            `ner-nel` (named entity recognition with named entity linking). Default:
-            ImpressoNamedEntityRecognitionRequestMethod.NER.
+        method (Union[Unset, ImpressoNamedEntityRecognitionRequestMethod]): NER method to be used: `ner` (default),
+            `ner-nel` (named entity recognition with named entity linking) and `nel` (linking only - enclose entities in
+            [START] [END] tags). Default: ImpressoNamedEntityRecognitionRequestMethod.NER.
     """
 
     text: str

diff --git a/impresso/api_client/models/impresso_named_entity_recognition_request_method.py b/impresso/api_client/models/impresso_named_entity_recognition_request_method.py
@@ -3,6 +3,7 @@
 
 
 class ImpressoNamedEntityRecognitionRequestMethod(str, Enum):
+    NEL = "nel"
     NER = "ner"
     NER_NEL = "ner-nel"
 
@@ -11,6 +12,7 @@ def __str__(self) -> str:
 
 
 ImpressoNamedEntityRecognitionRequestMethodLiteral = Literal[
+    "nel",
     "ner",
     "ner-nel",
 ]
diff --git a/impresso/api_models.py b/impresso/api_models.py
@@ -330,10 +330,10 @@ class ImpressoNerRequest(BaseModel):
         ),
     ]
     method: Annotated[
-        Optional[Literal['ner', 'ner-nel']],
+        Optional[Literal['ner', 'ner-nel', 'nel']],
         Field(
             'ner',
-            description='NER method to be used: `ner` (default) or `ner-nel` (named entity recognition with named entity linking).',
+            description='NER method to be used: `ner` (default), `ner-nel` (named entity recognition with named entity linking) and `nel` (linking only - enclose entities in [START] [END] tags).',
         ),
     ]