convoviz 0.1.3__tar.gz → 0.1.5__tar.gz

{convoviz-0.1.3 → convoviz-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: convoviz
-Version: 0.1.3
+Version: 0.1.5
 Summary: Get analytics and visualizations on your ChatGPT data !
 Home-page: https://github.com/mohamed-chs/chatgpt-history-export-to-md
 License: MIT

{convoviz-0.1.3 → convoviz-0.1.5}/convoviz/models/_conversation.py

@@ -5,7 +5,7 @@ object path : conversations.json -> conversation (one of the list items)
 
 from __future__ import annotations
 
-from datetime import datetime, timedelta, timezone
+from datetime import datetime, timedelta
 from os import utime as os_utime
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, ClassVar
@@ -46,7 +46,7 @@ class Conversation(BaseModel):
     plugin_ids: list[str] | None = None
     conversation_id: str
     conversation_template_id: str | None = None
-    id: str | None = None
+    id: str | None = None  # noqa: A003
 
     @classmethod
     def update_configs(cls, configs: ConversationConfigs) -> None:
@@ -81,7 +81,7 @@ class Conversation(BaseModel):
         return [
             node
             for node in self._all_message_nodes
-            if node.message and node.message.author_role in authors
+            if node.message and node.message.author.role in authors
         ]
 
     @property
@@ -99,7 +99,7 @@ class Conversation(BaseModel):
         """List of all content types in the conversation (all branches)."""
         return list(
             {
-                node.message.content_type
+                node.message.content.content_type
                 for node in self._all_message_nodes
                 if node.message
             },
@@ -123,7 +123,7 @@ class Conversation(BaseModel):
 
         message = assistant_nodes[0].message
 
-        return message.model if message else None
+        return message.metadata.model_slug if message else None
 
     @property
     def used_plugins(self) -> list[str]:
@@ -268,14 +268,22 @@ class Conversation(BaseModel):
     @property
     def month_start(self) -> datetime:
         """Return the first of the month the conversation was created in."""
-        return datetime(
-            self.create_time.year,
-            self.create_time.month,
-            1,
-            tzinfo=timezone.utc,
+        return self.create_time.replace(
+            day=1,
+            hour=0,
+            minute=0,
+            second=0,
+            microsecond=0,
         )
 
     @property
     def year_start(self) -> datetime:
         """Return the first of January of the year the conversation was created in."""
-        return datetime(self.create_time.year, 1, 1, tzinfo=timezone.utc)
+        return self.create_time.replace(
+            month=1,
+            day=1,
+            hour=0,
+            minute=0,
+            second=0,
+            microsecond=0,
+        )

{convoviz-0.1.3 → convoviz-0.1.5}/convoviz/models/_message.py

@@ -69,15 +69,10 @@ class Message(BaseModel):
         """Set the configuration for all messages."""
         cls.__configs.update(configs)
 
-    @property
-    def author_role(self) -> AuthorRole:
-        """Return the role of the author of the message."""
-        return self.author.role
-
     @property
     def header(self) -> str:
         """Get the title header of the message based on the configs."""
-        return self.__configs["author_headers"][self.author_role]
+        return self.__configs["author_headers"][self.author.role]
 
     @property
     def text(self) -> str:
@@ -92,13 +87,3 @@ class Message(BaseModel):
         # this error caught some hidden bugs in the data. need more of these
         err_msg = f"No valid content found in message: {self.id}"
         raise ValueError(err_msg)
-
-    @property
-    def content_type(self) -> str:
-        """Get the content type of the message."""
-        return self.content.content_type
-
-    @property
-    def model(self) -> str | None:
-        """Get the model used for the message."""
-        return self.metadata.model_slug

{convoviz-0.1.3 → convoviz-0.1.5}/pyproject.toml

@@ -21,7 +21,7 @@ enable_incomplete_feature = ["Unpack"]
 
 [tool.poetry]
 name = "convoviz"
-version = "0.1.3"
+version = "0.1.5"
 description = "Get analytics and visualizations on your ChatGPT data !"
 authors = ["Mohamed Cheikh Sidiya <mohamedcheikhsidiya77@gmail.com>"]
 license = "MIT"