allow multiple values in company enrichment (#160)

* allow multiple values in ce

* updates

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "peopledatalabs"
-version = "4.1.1"
+version = "4.1.2"
 description = "Official Python client for the People Data Labs API"
 homepage = "https://www.peopledatalabs.com"
 repository = "https://github.com/peopledatalabs/peopledatalabs-python"

src/peopledatalabs/__init__.py

@@ -5,6 +5,6 @@
 from .main import PDLPY
 
 
-__version__ = "4.1.1"
+__version__ = "4.1.2"
 
 __all__ = ["PDLPY"]

src/peopledatalabs/models/company.py

@@ -22,17 +22,17 @@ class CompanyBaseModel(BaseModel):
     Base parameters model for the enrichment API.
     """
 
-    country: Optional[str]
-    locality: Optional[str]
+    country: Optional[Union[str, List[str]]]
+    locality: Optional[Union[str, List[str]]]
     location: Optional[Union[List[str], str]]
-    name: Optional[str]
+    name: Optional[Union[str, List[str]]]
     pdl_id: Optional[str]
-    postal_code: Optional[str]
-    profile: Optional[str]
-    region: Optional[str]
-    street_address: Optional[str]
-    ticker: Optional[str]
-    website: Optional[str]
+    postal_code: Optional[Union[str, List[str]]]
+    profile: Optional[Union[str, List[str]]]
+    region: Optional[Union[str, List[str]]]
+    street_address: Optional[Union[str, List[str]]]
+    ticker: Optional[Union[str, List[str]]]
+    website: Optional[Union[str, List[str]]]
 
     @root_validator(pre=True)
     def non_ambiguous(cls, v):

tests/client/test_client.py

@@ -19,7 +19,7 @@ def test_version():
     """
     Version check.
     """
-    assert __version__ == "4.1.1"
+    assert __version__ == "4.1.2"
 
 
 @pytest.mark.usefixtures("fake_api_key")