convoviz 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

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/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.dist-info → convoviz-0.1.5.dist-info}/METADATA

@@ -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.dist-info → convoviz-0.1.5.dist-info}/RECORD

@@ -37,13 +37,13 @@ convoviz/configuration.py,sha256=gsabIQt_oiLCGUFYaftcM55oIhgWm6O6PbUBwPu6B94,390
 convoviz/data_analysis.py,sha256=OBnElm61pclK6DTFCKdrJHRoSiXTMyL7S8h_3TZ_0Ek,3551
 convoviz/long_runs.py,sha256=lViUuD2aC-djfyzaw2iFprIiEhLaOhAcQQA1E7OD-EM,2676
 convoviz/models/__init__.py,sha256=ui47MNe4qRhx2EFZpaeU3Qo32hFVrTOtMcUvKXbv4X0,261
-convoviz/models/_conversation.py,sha256=63nnKhO5oI3lhTAF7zMGUXIIRCiOFoj9l70DDx85SNk,8874
+convoviz/models/_conversation.py,sha256=avhvc4FHex4s4jQ_A_c78MbqbtYSsUb-iLVjHvW1YtE,9015
 convoviz/models/_conversation_set.py,sha256=0ngb3fcsy89YkYjRNIQSFpM326ZUu2u7gikPed91VC8,6600
-convoviz/models/_message.py,sha256=qihdl8K4D7ofJnso5dqGbiFHa2m5s3up2OzMYjMW3nk,3030
+convoviz/models/_message.py,sha256=fqwGE4fQDvBfoFBWLK0_ITo5tOXFt7oDhvcmpQHqJag,2600
 convoviz/models/_node.py,sha256=TEWtqIvfkTJA3uE9exwfO1ow7MeYGK8bzSTorXbigsM,2214
 convoviz/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 convoviz/utils.py,sha256=EkEMS6y7OZEvREpzwjdq04PQm1wDEszk41gnuugCk6A,7078
-convoviz-0.1.3.dist-info/LICENSE,sha256=LO6tqJjRNZIT4w4btugJrWzmwsxeZZcL3j7KV7qK_tM,1099
-convoviz-0.1.3.dist-info/METADATA,sha256=0UEd4mVAA7S3xPvmKsfTIMQVoT5ZJVJqzBLRvY7qe5Q,5458
-convoviz-0.1.3.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-convoviz-0.1.3.dist-info/RECORD,,
+convoviz-0.1.5.dist-info/LICENSE,sha256=LO6tqJjRNZIT4w4btugJrWzmwsxeZZcL3j7KV7qK_tM,1099
+convoviz-0.1.5.dist-info/METADATA,sha256=mM3LFGolDM-_C2ue-0kQXV2wed15kH7AGuWggumr6_4,5458
+convoviz-0.1.5.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+convoviz-0.1.5.dist-info/RECORD,,