vysion 2.2.0__tar.gz → 2.2.2__tar.gz

{vysion-2.2.0 → vysion-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vysion
-Version: 2.2.0
+Version: 2.2.2
 Summary: The official Python client library for Vysion
 Home-page: https://vysion.ai
 License: Apache-2.0
@@ -29,7 +29,7 @@ Welcome to the PyPi webpage for Vysion, our implementation as a Python library t
 
 You can request a demo for the web app or an API-key to use in this library at [vysion.ai](https://vysion.ai).
 
-Latest version: [2.2.0](https://pypi.org/project/vysion/)
+Latest version: [2.2.2](https://pypi.org/project/vysion/)
 
 You can visit [the documentation](https://developers.vysion.ai/?python) for more information on the searches and requests performed with the library or directly on the API.
 

{vysion-2.2.0 → vysion-2.2.2}/README.md

@@ -4,7 +4,7 @@ Welcome to the PyPi webpage for Vysion, our implementation as a Python library t
 
 You can request a demo for the web app or an API-key to use in this library at [vysion.ai](https://vysion.ai).
 
-Latest version: [2.2.0](https://pypi.org/project/vysion/)
+Latest version: [2.2.2](https://pypi.org/project/vysion/)
 
 You can visit [the documentation](https://developers.vysion.ai/?python) for more information on the searches and requests performed with the library or directly on the API.
 

{vysion-2.2.0 → vysion-2.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vysion"
-version = "2.2.0"
+version = "2.2.2"
 description = "The official Python client library for Vysion"
 homepage = "https://vysion.ai"
 repository = "https://github.com/ByronLabs/vysion-py"

{vysion-2.2.0 → vysion-2.2.2}/vysion/client/client.py

@@ -30,6 +30,7 @@ from vysion.dto import (
     ErrorCode,
     DocumentHit,
     VysionResponse,
+    VysionPaginationResponse,
     RansomwareHit,
     Stat,
     Network,
@@ -510,14 +511,22 @@ class Client(BaseClient):
     
 
     @vysion_error_manager
-    def get_im_chat(
-        self, platform: str, channelId: str, gte: datetime = None, lte: datetime = None
-    ) -> VysionResponse[ImMessageHit]:
+    def get_im_chat_messages(
+        self,
+        platform: str,
+        channelId: str,
+        messageId: str | None,
+        cursor: str | None,
+        limit: str = 30
+    ) -> VysionPaginationResponse:
         url = self._build_api_url__(
-            "im/" + platform + "/chat/" + channelId, gte=gte, lte=lte
+            "im/" + platform + "/chat/" + channelId,
+            messageId=messageId,
+            cursor=cursor,
+            limit=limit
         )
 
-        result = VysionResponse[ImMessageHit].model_validate(self._make_request(url))
+        result = VysionPaginationResponse.model_validate(self._make_request(url))
         return result.data
 
     @vysion_error_manager

{vysion-2.2.0 → vysion-2.2.2}/vysion/dto/dto.py

@@ -411,7 +411,6 @@ class ImProfileHit(BaseModel):
             raise ValueError("DetectionDate field cannot be empty")
         return v
 
-
 class ImChannelHit(BaseModel):
     channelId: Union[int, str]
     channelTitles: Optional[List[str]] = Field(default_factory=lambda: None)
@@ -550,7 +549,6 @@ class CryptoFeedHit(BaseModel):
     id: str
     url: str
     detectionDate: datetime
-    url: str
     network: str
     title: Optional[str] = Field(default_factory=lambda: None)
     tag: List[Tag] = Field(default_factory=lambda: [])
@@ -729,6 +727,31 @@ class Result(BaseModel, Generic[T]):
 
         return data
 
+class ResultPagination(BaseModel, Generic[T]):
+    total: int = 0
+    hits: List[T] = Field(default_factory=lambda: [])
+    prev_cursor: Optional[str] = None
+    next_cursor: Optional[str] = None
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        if self.total <= 0:
+            self.total = len(self.hits)
+
+    def get_type(self) -> type:
+        if len(self.hits) <= 0:
+            return NoneType
+
+        return type(self.hits[0])
+
+    @model_validator(mode="before")
+    @classmethod
+    def check_hits(cls, data):
+        hits = data.get("hits")
+        if not isinstance(hits, list):
+            raise ValueError("hits must be a list")
+
+        return data
 
 class ErrorCode(int, Enum):
     BAD_REQUEST = 400
@@ -760,3 +783,8 @@ class Error(BaseModel):
 class VysionResponse(BaseModel, Generic[T]):
     data: Optional[Result[T]] = None
     error: Optional[Error] = None
+
+
+class VysionPaginationResponse(BaseModel, Generic[T]):
+    data: Optional[ResultPagination[T]] = None
+    error: Optional[Error] = None

{vysion-2.2.0 → vysion-2.2.2}/vysion/version.py

@@ -15,4 +15,4 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-__version__ = "2.2.0"
+__version__ = "2.2.2"