exa-py 1.14.11__py3-none-any.whl → 1.14.12__py3-none-any.whl

exa_py/websets/__init__.py

@@ -1,7 +1,9 @@
 from .client import WebsetsClient
 from .imports import ImportsClient
+from .events import EventsClient
 
 __all__ = [
     "WebsetsClient",
     "ImportsClient",
+    "EventsClient",
 ] 

exa_py/websets/client.py

@@ -18,6 +18,7 @@ from .enrichments import WebsetEnrichmentsClient
 from .webhooks import WebsetWebhooksClient
 from .monitors import MonitorsClient
 from .imports import ImportsClient
+from .events import EventsClient
 
 class WebsetsClient(WebsetsBaseClient):
     """Client for managing Websets."""
@@ -30,6 +31,7 @@ class WebsetsClient(WebsetsBaseClient):
         self.webhooks = WebsetWebhooksClient(client)
         self.monitors = MonitorsClient(client)
         self.imports = ImportsClient(client)
+        self.events = EventsClient(client)
 
     def create(self, params: Union[Dict[str, Any], CreateWebsetParameters]) -> Webset:
         """Create a new Webset.

exa_py/websets/events/__init__.py

@@ -0,0 +1,3 @@
+from .client import EventsClient
+
+__all__ = ["EventsClient"]

exa_py/websets/events/client.py

@@ -0,0 +1,106 @@
+from __future__ import annotations
+
+from typing import List, Optional, Union
+
+from ..types import (
+    EventType,
+    ListEventsResponse,
+    WebsetCreatedEvent,
+    WebsetDeletedEvent,
+    WebsetIdleEvent,
+    WebsetPausedEvent,
+    WebsetItemCreatedEvent,
+    WebsetItemEnrichedEvent,
+    WebsetSearchCreatedEvent,
+    WebsetSearchUpdatedEvent,
+    WebsetSearchCanceledEvent,
+    WebsetSearchCompletedEvent,
+)
+from ..core.base import WebsetsBaseClient
+
+# Type alias for all event types
+Event = Union[
+    WebsetCreatedEvent,
+    WebsetDeletedEvent,
+    WebsetIdleEvent,
+    WebsetPausedEvent,
+    WebsetItemCreatedEvent,
+    WebsetItemEnrichedEvent,
+    WebsetSearchCreatedEvent,
+    WebsetSearchUpdatedEvent,
+    WebsetSearchCanceledEvent,
+    WebsetSearchCompletedEvent,
+]
+
+class EventsClient(WebsetsBaseClient):
+    """Client for managing Events."""
+    
+    def __init__(self, client):
+        super().__init__(client)
+
+    def list(self, *, cursor: Optional[str] = None, limit: Optional[int] = None, 
+             types: Optional[List[EventType]] = None) -> ListEventsResponse:
+        """List all Events.
+        
+        Args:
+            cursor (str, optional): The cursor to paginate through the results.
+            limit (int, optional): The number of results to return.
+            types (List[EventType], optional): The types of events to filter by.
+        
+        Returns:
+            ListEventsResponse: List of events.
+        """
+        params = {}
+        if cursor is not None:
+            params["cursor"] = cursor
+        if limit is not None:
+            params["limit"] = limit
+        if types is not None:
+            # Convert EventType enums to their string values
+            params["types"] = [t.value if hasattr(t, 'value') else t for t in types]
+            
+        response = self.request("/v0/events", params=params, method="GET")
+        return ListEventsResponse.model_validate(response)
+
+    def get(self, id: str) -> Event:
+        """Get an Event by ID.
+        
+        Args:
+            id (str): The ID of the Event.
+        
+        Returns:
+            Event: The retrieved event.
+        """
+        response = self.request(f"/v0/events/{id}", method="GET")
+        
+        # The response should contain a 'type' field that helps us determine
+        # which specific event class to use for validation
+        event_type = response.get('type')
+        
+        # Map event types to their corresponding classes
+        event_type_map = {
+            'webset.created': WebsetCreatedEvent,
+            'webset.deleted': WebsetDeletedEvent,
+            'webset.idle': WebsetIdleEvent,
+            'webset.paused': WebsetPausedEvent,
+            'webset.item.created': WebsetItemCreatedEvent,
+            'webset.item.enriched': WebsetItemEnrichedEvent,
+            'webset.search.created': WebsetSearchCreatedEvent,
+            'webset.search.updated': WebsetSearchUpdatedEvent,
+            'webset.search.canceled': WebsetSearchCanceledEvent,
+            'webset.search.completed': WebsetSearchCompletedEvent,
+        }
+        
+        event_class = event_type_map.get(event_type)
+        if event_class:
+            return event_class.model_validate(response)
+        else:
+            # Fallback - try each type until one validates
+            # This shouldn't happen in normal operation
+            for event_class in event_type_map.values():
+                try:
+                    return event_class.model_validate(response)
+                except Exception:
+                    continue
+            
+            raise ValueError(f"Unknown event type: {event_type}")

exa_py/websets/types.py

@@ -6,7 +6,7 @@ from __future__ import annotations
 
 from datetime import datetime
 from enum import Enum
-from typing import Any, Dict, List, Literal, Optional, Union
+from typing import Any, Dict, List, Literal, Optional, Union, Annotated
 
 from pydantic import AnyUrl, Field, PositiveInt, confloat, constr
 from .core.base import ExaBaseModel
@@ -287,20 +287,21 @@ class ImportSource(Enum):
 
 
 class ListEventsResponse(ExaBaseModel):
-    data: List[
-        Union[
-            WebsetCreatedEvent,
-            WebsetDeletedEvent,
-            WebsetIdleEvent,
-            WebsetPausedEvent,
-            WebsetItemCreatedEvent,
-            WebsetItemEnrichedEvent,
-            WebsetSearchCreatedEvent,
-            WebsetSearchUpdatedEvent,
-            WebsetSearchCanceledEvent,
-            WebsetSearchCompletedEvent,
-        ]
-    ] = Field(..., discriminator='type')
+    data: List[Annotated[
+      Union[
+          WebsetCreatedEvent,
+          WebsetDeletedEvent,
+          WebsetIdleEvent,
+          WebsetPausedEvent,
+          WebsetItemCreatedEvent,
+          WebsetItemEnrichedEvent,
+          WebsetSearchCreatedEvent,
+          WebsetSearchUpdatedEvent,
+          WebsetSearchCanceledEvent,
+          WebsetSearchCompletedEvent,
+      ],
+      Field(discriminator='type')
+  ]]
     """
     The list of events
     """
@@ -1440,6 +1441,17 @@ class WebsetItemPersonProperties(ExaBaseModel):
     """
 
 
+class WebsetItemPersonCompanyPropertiesFields(ExaBaseModel):
+    name: str
+    """
+    The name of the company
+    """
+    location: Optional[str] = None
+    """
+    The location the person is working at the company
+    """
+
+
 class WebsetItemPersonPropertiesFields(ExaBaseModel):
     name: str
     """
@@ -1453,6 +1465,10 @@ class WebsetItemPersonPropertiesFields(ExaBaseModel):
     """
     The current work position of the person
     """
+    company: Optional[WebsetItemPersonCompanyPropertiesFields] = None
+    """
+    The company the person is working at
+    """
     picture_url: Optional[AnyUrl] = Field(None, alias='pictureUrl')
     """
     The URL of the person's picture

{exa_py-1.14.11.dist-info → exa_py-1.14.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: exa-py
-Version: 1.14.11
+Version: 1.14.12
 Summary: Python SDK for Exa API.
 Home-page: https://github.com/exa-labs/exa-py
 Author: Exa

{exa_py-1.14.11.dist-info → exa_py-1.14.12.dist-info}/RECORD

@@ -5,13 +5,15 @@ exa_py/utils.py,sha256=sq1KOajuMWI1lygYeb9MqjXn8yqVyK0BtFZZmTwSaMY,2580
 exa_py/research/__init__.py,sha256=QeY-j6bP4QP5tF9ytX0IeQhJvd0Wn4cJCD69U8pP7kA,271
 exa_py/research/client.py,sha256=x9b299ROfbhdMrdto6jZCHGRJE5U5V2fWHsSlnnNkWo,12471
 exa_py/research/models.py,sha256=j7YgRoMRp2MLgnaij7775x_hJEeV5gksKpfLwmawqxY,3704
-exa_py/websets/__init__.py,sha256=6HQWR8_ABPeZx2sHfwGV4QQf1Eh8MRaO-kzoIO_1ua8,126
-exa_py/websets/client.py,sha256=nP0XZAEMIFvinb1DJ6gNF-8jCnKMUmBun5wZ9EgeZ0M,4891
-exa_py/websets/types.py,sha256=UC41VqvBcUueSqUjxmb0nLb-c7qmScoNkcmdyp9npqY,41736
+exa_py/websets/__init__.py,sha256=x7Dc0MS8raRXA7Ud6alKgnsUmLi6X9GTqfB8kOwC9iQ,179
+exa_py/websets/client.py,sha256=v8Y0p5PosjLkb7EYQ83g3nmoIIHmKaXF7JQVT8K5h2E,4967
+exa_py/websets/types.py,sha256=ykJHPXNiqr2fbRYYprshgBcG9X5IRRwi2mu56g7XCZA,42085
 exa_py/websets/core/__init__.py,sha256=xOyrFaqtBocMUu321Jpbk7IzIQRNZufSIGJXrKoG-Bg,323
 exa_py/websets/core/base.py,sha256=thVIeRtlabbvueP0dAni5Nwtl9AWYv1I1Mmyc_jlYO0,4086
 exa_py/websets/enrichments/__init__.py,sha256=5dJIEKKceUost3RnI6PpCSB3VjUCBzxseEsIXu-ZY-Y,83
 exa_py/websets/enrichments/client.py,sha256=obUjn4vH6tKBMtHEBVdMzlN8in0Fx3sCP-bXx-Le1zM,2338
+exa_py/websets/events/__init__.py,sha256=aFJ9O5UudtQQzndVmdB96IaM2l07qyM1B_8xKY7rp58,60
+exa_py/websets/events/client.py,sha256=Hzatqp3X-K0ZGe36cjFMgbhnsErcDLdGWQVirhmHjvY,3622
 exa_py/websets/imports/__init__.py,sha256=iEl-fZZSdcvKaqLgjMES_0RwYn7hZDCMf6BZriCrjgw,64
 exa_py/websets/imports/client.py,sha256=nJs46hxlSkZm7qjboYHNBuJ62gLmA_Yzr9fc-NDky0Y,6795
 exa_py/websets/items/__init__.py,sha256=DCWZJVtRmUjnMEkKdb5gW1LT9cHcb-J8lENMnyyBeKU,71
@@ -24,7 +26,7 @@ exa_py/websets/searches/__init__.py,sha256=_0Zx8ES5fFTEL3T8mhLxq_xK2t0JONx6ad6At
 exa_py/websets/searches/client.py,sha256=X3f7axWGfecmxf-2tBTX0Yf_--xToz1X8ZHbbudEzy0,1790
 exa_py/websets/webhooks/__init__.py,sha256=iTPBCxFd73z4RifLQMX6iRECx_6pwlI5qscLNjMOUHE,77
 exa_py/websets/webhooks/client.py,sha256=zsIRMTeJU65yj-zo7Zz-gG02Prtzgcx6utGFSoY4HQQ,4222
-exa_py-1.14.11.dist-info/METADATA,sha256=ndNf4lDM1Vz44xlJO29qVVHGl6vvb9ozXiq3L6XOmpw,3719
-exa_py-1.14.11.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-exa_py-1.14.11.dist-info/top_level.txt,sha256=Mfkmscdw9HWR1PtVhU1gAiVo6DHu_tyiVdb89gfZBVI,7
-exa_py-1.14.11.dist-info/RECORD,,
+exa_py-1.14.12.dist-info/METADATA,sha256=sbyg90b0KSXgCm2UwwQBm6M5So9-0eh1eFUm4i8Fa_0,3719
+exa_py-1.14.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+exa_py-1.14.12.dist-info/top_level.txt,sha256=Mfkmscdw9HWR1PtVhU1gAiVo6DHu_tyiVdb89gfZBVI,7
+exa_py-1.14.12.dist-info/RECORD,,

{exa_py-1.14.11.dist-info → exa_py-1.14.12.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.3.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any