acryl-datahub-cloud 0.3.15rc1__py3-none-any.whl → 0.3.15rc3__py3-none-any.whl

acryl_datahub_cloud/_codegen_config.json

@@ -1,6 +1,6 @@
 {
   "name": "acryl-datahub-cloud",
-  "version": "0.3.15rc1",
+  "version": "0.3.15rc3",
   "install_requires": [
     "avro-gen3==0.7.16",
     "acryl-datahub"

acryl_datahub_cloud/datahub_reporting/datahub_form_reporting.py

@@ -2,10 +2,10 @@ import json
 import logging
 from datetime import date, datetime, timezone
 from enum import Enum
-from typing import Any, Callable, Dict, Iterable, List, Optional
+from typing import Any, Callable, Dict, Iterable, List, Optional, Union
 
 import pandas as pd
-from pydantic import BaseModel
+from pydantic import BaseModel, field_validator
 
 from acryl_datahub_cloud.elasticsearch.graph_service import BaseModelRow
 from datahub.emitter.mcp import MetadataChangeProposalWrapper
@@ -130,6 +130,22 @@ class DataHubFormReportingData(FormData):
         platformInstance: Optional[str] = None
         domains: List[str] = []
 
+        @field_validator(
+            "completedFormsIncompletePromptResponseTimes",
+            "completedFormsCompletedPromptResponseTimes",
+            "incompleteFormsIncompletePromptResponseTimes",
+            "incompleteFormsCompletedPromptResponseTimes",
+            mode="before",
+        )
+        @classmethod
+        def convert_timestamps_to_strings(
+            cls, v: Union[List[int], List[str]]
+        ) -> List[str]:
+            """Convert timestamp integers to strings for compatibility with GMS data."""
+            if not isinstance(v, list):
+                return v
+            return [str(item) for item in v]
+
     def __init__(self, graph: DataHubGraph, allowed_forms: Optional[List[str]] = None):
         self.graph: DataHubGraph = graph
         self.form_registry = FormRegistry(graph)
@@ -430,7 +446,7 @@ class DataHubFormReportingData(FormData):
                             question_status=QuestionStatus.COMPLETED,
                             question_completed_date=datetime.fromtimestamp(
                                 float(prompt_response_time) / 1000, tz=timezone.utc
-                            ),
+                            ).date(),
                             snapshot_date=self.snapshot_date,
                         )
             complete_forms = (
@@ -532,7 +548,7 @@ class DataHubFormReportingData(FormData):
                             question_status=QuestionStatus.COMPLETED,
                             question_completed_date=datetime.fromtimestamp(
                                 float(prompt_response_time) / 1000, tz=timezone.utc
-                            ),
+                            ).date(),
                             snapshot_date=self.snapshot_date,
                         )
 

acryl_datahub_cloud/elasticsearch/graph_service.py

@@ -1,5 +1,5 @@
 import datetime
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, get_args, get_origin
 
 import pyarrow as pa
 from pydantic import BaseModel
@@ -15,7 +15,37 @@ class BaseModelRow(BaseModel):
         return self.__dict__
 
     @staticmethod
-    def pydantic_type_to_pyarrow(type_: Any) -> pa.DataType:
+    def pydantic_type_to_pyarrow(type_: Any) -> pa.DataType:  # noqa: C901
+        # Handle generic types (List, Optional, Union, etc.)
+        origin = get_origin(type_)
+
+        if origin is list:
+            # List[X] -> list(X)
+            args = get_args(type_)
+            if args:
+                inner_type = BaseModelRow.pydantic_type_to_pyarrow(args[0])
+                return pa.list_(inner_type)
+            return pa.list_(pa.string())  # Default to list of strings
+
+        if origin is type(None):
+            # Just None type
+            return pa.null()
+
+        # Optional[X] is Union[X, None]
+        args = get_args(type_)
+        if args:
+            # Check if this is Optional (Union with None)
+            non_none_types = [arg for arg in args if arg is not type(None)]
+            if len(non_none_types) < len(args):  # Had None in the union
+                if non_none_types:
+                    return BaseModelRow.pydantic_type_to_pyarrow(non_none_types[0])
+                return pa.null()
+
+        # Handle simple types - check if it's actually a class first
+        if not isinstance(type_, type):
+            # If it's not a type/class, default to string
+            return pa.string()
+
         if issubclass(type_, bool):
             return pa.bool_()
         elif issubclass(type_, int):
@@ -50,16 +80,34 @@ class BaseModelRow(BaseModel):
         )  # Default to str for unknown types
         return BaseModelRow.pydantic_type_to_pyarrow(python_type)
 
+    @staticmethod
+    def _is_optional_type(type_: Any) -> bool:
+        """Check if a type annotation is Optional (i.e., Union with None)."""
+        origin = get_origin(type_)
+        if origin is type(None):
+            return True
+
+        args = get_args(type_)
+        if args:
+            # Check if None is in the Union args (Optional is Union[X, None])
+            return type(None) in args
+
+        return False
+
     @classmethod
     def arrow_schema(cls) -> pa.Schema:
         fields = []
         for field_name, field_model in cls.model_fields.items():
             pyarrow_type = BaseModelRow.pydantic_type_to_pyarrow(field_model.annotation)
             pyarrow_field = pa.field(field_name, pyarrow_type)
-            if not field_model.is_required():
-                pyarrow_field = pyarrow_field.with_nullable(True)
-            else:
-                pyarrow_field = pyarrow_field.with_nullable(False)
+
+            # Check if the type is Optional (Union with None) OR if the field has a default value
+            is_nullable = (
+                BaseModelRow._is_optional_type(field_model.annotation)
+                or not field_model.is_required()
+            )
+
+            pyarrow_field = pyarrow_field.with_nullable(is_nullable)
             fields.append(pyarrow_field)
         return pa.schema(fields)
 

acryl_datahub_cloud/metadata/_urns/urn_defs.py

@@ -1511,6 +1511,62 @@ class MlPrimaryKeyUrn(_SpecificUrn):
     def name(self) -> str:
         return self._entity_ids[1]
 
+if TYPE_CHECKING:
+    from datahub.metadata.schema_classes import DataHubAiConversationKeyClass
+
+class DataHubAiConversationUrn(_SpecificUrn):
+    ENTITY_TYPE: ClassVar[Literal["dataHubAiConversation"]] = "dataHubAiConversation"
+    _URN_PARTS: ClassVar[int] = 1
+
+    def __init__(self, id: Union["DataHubAiConversationUrn", str], *, _allow_coercion: bool = True) -> None:
+        if _allow_coercion:
+            # Field coercion logic (if any is required).
+            if isinstance(id, str):
+                if id.startswith('urn:li:'):
+                    try:
+                        id = DataHubAiConversationUrn.from_string(id)
+                    except InvalidUrnError:
+                        raise InvalidUrnError(f'Expecting a DataHubAiConversationUrn but got {id}')
+                else:
+                    id = UrnEncoder.encode_string(id)
+
+        # Validation logic.
+        if not id:
+            raise InvalidUrnError("DataHubAiConversationUrn id cannot be empty")
+        if isinstance(id, DataHubAiConversationUrn):
+            id = id.id
+        elif isinstance(id, Urn):
+            raise InvalidUrnError(f'Expecting a DataHubAiConversationUrn but got {id}')
+        if UrnEncoder.contains_reserved_char(id):
+            raise InvalidUrnError(f'DataHubAiConversationUrn id contains reserved characters')
+
+        super().__init__(self.ENTITY_TYPE, [id])
+
+    @classmethod
+    def _parse_ids(cls, entity_ids: List[str]) -> "DataHubAiConversationUrn":
+        if len(entity_ids) != cls._URN_PARTS:
+            raise InvalidUrnError(f"DataHubAiConversationUrn should have {cls._URN_PARTS} parts, got {len(entity_ids)}: {entity_ids}")
+        return cls(id=entity_ids[0], _allow_coercion=False)
+
+    @classmethod
+    def underlying_key_aspect_type(cls) -> Type["DataHubAiConversationKeyClass"]:
+        from datahub.metadata.schema_classes import DataHubAiConversationKeyClass
+
+        return DataHubAiConversationKeyClass
+
+    def to_key_aspect(self) -> "DataHubAiConversationKeyClass":
+        from datahub.metadata.schema_classes import DataHubAiConversationKeyClass
+
+        return DataHubAiConversationKeyClass(id=self.id)
+
+    @classmethod
+    def from_key_aspect(cls, key_aspect: "DataHubAiConversationKeyClass") -> "DataHubAiConversationUrn":
+        return cls(id=key_aspect.id)
+
+    @property
+    def id(self) -> str:
+        return self._entity_ids[0]
+
 if TYPE_CHECKING:
     from datahub.metadata.schema_classes import TestKeyClass
 

acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/conversation/__init__.py

@@ -0,0 +1,27 @@
+# mypy: ignore-errors
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# pylint: skip-file
+# fmt: off
+# isort: skip_file
+from .....schema_classes import DataHubAiConversationActorClass
+from .....schema_classes import DataHubAiConversationActorTypeClass
+from .....schema_classes import DataHubAiConversationInfoClass
+from .....schema_classes import DataHubAiConversationMessageClass
+from .....schema_classes import DataHubAiConversationMessageContentClass
+from .....schema_classes import DataHubAiConversationMessageTypeClass
+from .....schema_classes import DataHubAiConversationOriginTypeClass
+
+
+DataHubAiConversationActor = DataHubAiConversationActorClass
+DataHubAiConversationActorType = DataHubAiConversationActorTypeClass
+DataHubAiConversationInfo = DataHubAiConversationInfoClass
+DataHubAiConversationMessage = DataHubAiConversationMessageClass
+DataHubAiConversationMessageContent = DataHubAiConversationMessageContentClass
+DataHubAiConversationMessageType = DataHubAiConversationMessageTypeClass
+DataHubAiConversationOriginType = DataHubAiConversationOriginTypeClass
+
+# fmt: on

acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/metadata/key/__init__.py

@@ -21,6 +21,7 @@ from ......schema_classes import DataContractKeyClass
 from ......schema_classes import DataFlowKeyClass
 from ......schema_classes import DataHubAccessTokenKeyClass
 from ......schema_classes import DataHubActionKeyClass
+from ......schema_classes import DataHubAiConversationKeyClass
 from ......schema_classes import DataHubConnectionKeyClass
 from ......schema_classes import DataHubFileKeyClass
 from ......schema_classes import DataHubIngestionSourceKeyClass
@@ -92,6 +93,7 @@ DataContractKey = DataContractKeyClass
 DataFlowKey = DataFlowKeyClass
 DataHubAccessTokenKey = DataHubAccessTokenKeyClass
 DataHubActionKey = DataHubActionKeyClass
+DataHubAiConversationKey = DataHubAiConversationKeyClass
 DataHubConnectionKey = DataHubConnectionKeyClass
 DataHubFileKey = DataHubFileKeyClass
 DataHubIngestionSourceKey = DataHubIngestionSourceKeyClass

acryl_datahub_cloud/metadata/schema.avsc

@@ -18807,6 +18807,161 @@
     "doc": "Properties associated with a MLPrimaryKey editable from the UI"
   },
   "com.linkedin.pegasus2avro.ml.metadata.EvaluationData",
+  {
+    "type": "record",
+    "Aspect": {
+      "name": "dataHubAiConversationInfo"
+    },
+    "name": "DataHubAiConversationInfo",
+    "namespace": "com.linkedin.pegasus2avro.conversation",
+    "fields": [
+      {
+        "type": [
+          "null",
+          "string"
+        ],
+        "name": "title",
+        "default": null,
+        "doc": "Optional title for the conversation. "
+      },
+      {
+        "type": {
+          "type": "array",
+          "items": {
+            "type": "record",
+            "name": "DataHubAiConversationMessage",
+            "namespace": "com.linkedin.pegasus2avro.conversation",
+            "fields": [
+              {
+                "type": {
+                  "type": "enum",
+                  "name": "DataHubAiConversationMessageType",
+                  "namespace": "com.linkedin.pegasus2avro.conversation",
+                  "symbols": [
+                    "TEXT",
+                    "TOOL_CALL",
+                    "TOOL_RESULT",
+                    "THINKING"
+                  ]
+                },
+                "name": "type",
+                "doc": "The type of message"
+              },
+              {
+                "type": "long",
+                "name": "time",
+                "doc": "Timestamp in milliseconds"
+              },
+              {
+                "type": {
+                  "type": "record",
+                  "name": "DataHubAiConversationActor",
+                  "namespace": "com.linkedin.pegasus2avro.conversation",
+                  "fields": [
+                    {
+                      "type": {
+                        "type": "enum",
+                        "symbolDocs": {
+                          "AGENT": "An agent initiated message. ",
+                          "USER": "A user initiated message. "
+                        },
+                        "name": "DataHubAiConversationActorType",
+                        "namespace": "com.linkedin.pegasus2avro.conversation",
+                        "symbols": [
+                          "USER",
+                          "AGENT"
+                        ]
+                      },
+                      "name": "type",
+                      "doc": "The type of actor "
+                    },
+                    {
+                      "Relationship": {
+                        "entityTypes": [
+                          "corpuser",
+                          "corpGroup"
+                        ],
+                        "name": "AuthoredBy"
+                      },
+                      "java": {
+                        "class": "com.linkedin.pegasus2avro.common.urn.Urn"
+                      },
+                      "Urn": "Urn",
+                      "entityTypes": [
+                        "corpuser",
+                        "corpGroup"
+                      ],
+                      "type": [
+                        "null",
+                        "string"
+                      ],
+                      "name": "actor",
+                      "default": null,
+                      "doc": "The actor that has sent the message.\nRequired if type is not AGENT."
+                    }
+                  ]
+                },
+                "name": "actor",
+                "doc": "Messages in the conversation"
+              },
+              {
+                "type": {
+                  "type": "record",
+                  "name": "DataHubAiConversationMessageContent",
+                  "namespace": "com.linkedin.pegasus2avro.conversation",
+                  "fields": [
+                    {
+                      "type": "string",
+                      "name": "text",
+                      "doc": "Raw message text"
+                    }
+                  ],
+                  "doc": "Conversation message content info "
+                },
+                "name": "content",
+                "doc": "The contents of the message. "
+              }
+            ],
+            "doc": "Conversation info "
+          }
+        },
+        "name": "messages",
+        "doc": "Messages in the conversation"
+      },
+      {
+        "Searchable": {
+          "/actor": {
+            "fieldName": "creator",
+            "fieldType": "URN"
+          },
+          "/time": {
+            "fieldName": "createdAt",
+            "fieldType": "DATETIME"
+          }
+        },
+        "type": "com.linkedin.pegasus2avro.common.AuditStamp",
+        "name": "created",
+        "doc": "The time and actor who initiated the conversation"
+      },
+      {
+        "type": {
+          "type": "enum",
+          "symbolDocs": {
+            "DATAHUB_UI": "Chat session originated in DataHub UI "
+          },
+          "name": "DataHubAiConversationOriginType",
+          "namespace": "com.linkedin.pegasus2avro.conversation",
+          "symbols": [
+            "DATAHUB_UI"
+          ]
+        },
+        "name": "originType",
+        "default": "DATAHUB_UI",
+        "doc": "Origin type for the chat session"
+      }
+    ],
+    "doc": "Conversation info "
+  },
   "com.linkedin.pegasus2avro.datajob.DataJobInfo",
   "com.linkedin.pegasus2avro.datajob.DataFlowInfo",
   {
@@ -25172,6 +25327,28 @@
     "doc": "Key for a Version Set entity"
   },
   "com.linkedin.pegasus2avro.metadata.key.MLPrimaryKeyKey",
+  {
+    "type": "record",
+    "Aspect": {
+      "name": "dataHubAiConversationKey",
+      "keyForEntity": "dataHubAiConversation",
+      "entityCategory": "core",
+      "entityAspects": [
+        "dataHubAiConversationInfo"
+      ]
+    },
+    "name": "DataHubAiConversationKey",
+    "namespace": "com.linkedin.pegasus2avro.metadata.key",
+    "fields": [
+      {
+        "Searchable": {},
+        "type": "string",
+        "name": "id",
+        "doc": "Unique identifier for the conversation"
+      }
+    ],
+    "doc": "Key for a DataHub conversation"
+  },
   {
     "type": "record",
     "Aspect": {

acryl_datahub_cloud/metadata/schema_classes.py

@@ -12088,6 +12088,234 @@ class EditableContainerPropertiesClass(_Aspect):
         self._inner_dict['description'] = value
     
     
+class DataHubAiConversationActorClass(DictWrapper):
+    # No docs available.
+    
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.conversation.DataHubAiConversationActor")
+    def __init__(self,
+        type: Union[str, "DataHubAiConversationActorTypeClass"],
+        actor: Union[None, str]=None,
+    ):
+        super().__init__()
+        
+        self.type = type
+        self.actor = actor
+    
+    def _restore_defaults(self) -> None:
+        self.type = DataHubAiConversationActorTypeClass.USER
+        self.actor = self.RECORD_SCHEMA.fields_dict["actor"].default
+    
+    
+    @property
+    def type(self) -> Union[str, "DataHubAiConversationActorTypeClass"]:
+        """The type of actor """
+        return self._inner_dict.get('type')  # type: ignore
+    
+    @type.setter
+    def type(self, value: Union[str, "DataHubAiConversationActorTypeClass"]) -> None:
+        self._inner_dict['type'] = value
+    
+    
+    @property
+    def actor(self) -> Union[None, str]:
+        """The actor that has sent the message.
+    Required if type is not AGENT."""
+        return self._inner_dict.get('actor')  # type: ignore
+    
+    @actor.setter
+    def actor(self, value: Union[None, str]) -> None:
+        self._inner_dict['actor'] = value
+    
+    
+class DataHubAiConversationActorTypeClass(object):
+    # No docs available.
+    
+    USER = "USER"
+    """A user initiated message. """
+    
+    AGENT = "AGENT"
+    """An agent initiated message. """
+    
+    
+    
+class DataHubAiConversationInfoClass(_Aspect):
+    """Conversation info """
+
+
+    ASPECT_NAME = 'dataHubAiConversationInfo'
+    ASPECT_INFO = {}
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.conversation.DataHubAiConversationInfo")
+
+    def __init__(self,
+        messages: List["DataHubAiConversationMessageClass"],
+        created: "AuditStampClass",
+        title: Union[None, str]=None,
+        originType: Optional[Union[str, "DataHubAiConversationOriginTypeClass"]]=None,
+    ):
+        super().__init__()
+        
+        self.title = title
+        self.messages = messages
+        self.created = created
+        if originType is None:
+            # default: 'DATAHUB_UI'
+            self.originType = self.RECORD_SCHEMA.fields_dict["originType"].default
+        else:
+            self.originType = originType
+    
+    def _restore_defaults(self) -> None:
+        self.title = self.RECORD_SCHEMA.fields_dict["title"].default
+        self.messages = list()
+        self.created = AuditStampClass._construct_with_defaults()
+        self.originType = self.RECORD_SCHEMA.fields_dict["originType"].default
+    
+    
+    @property
+    def title(self) -> Union[None, str]:
+        """Optional title for the conversation. """
+        return self._inner_dict.get('title')  # type: ignore
+    
+    @title.setter
+    def title(self, value: Union[None, str]) -> None:
+        self._inner_dict['title'] = value
+    
+    
+    @property
+    def messages(self) -> List["DataHubAiConversationMessageClass"]:
+        """Messages in the conversation"""
+        return self._inner_dict.get('messages')  # type: ignore
+    
+    @messages.setter
+    def messages(self, value: List["DataHubAiConversationMessageClass"]) -> None:
+        self._inner_dict['messages'] = value
+    
+    
+    @property
+    def created(self) -> "AuditStampClass":
+        """The time and actor who initiated the conversation"""
+        return self._inner_dict.get('created')  # type: ignore
+    
+    @created.setter
+    def created(self, value: "AuditStampClass") -> None:
+        self._inner_dict['created'] = value
+    
+    
+    @property
+    def originType(self) -> Union[str, "DataHubAiConversationOriginTypeClass"]:
+        """Origin type for the chat session"""
+        return self._inner_dict.get('originType')  # type: ignore
+    
+    @originType.setter
+    def originType(self, value: Union[str, "DataHubAiConversationOriginTypeClass"]) -> None:
+        self._inner_dict['originType'] = value
+    
+    
+class DataHubAiConversationMessageClass(DictWrapper):
+    """Conversation info """
+    
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.conversation.DataHubAiConversationMessage")
+    def __init__(self,
+        type: Union[str, "DataHubAiConversationMessageTypeClass"],
+        time: int,
+        actor: "DataHubAiConversationActorClass",
+        content: "DataHubAiConversationMessageContentClass",
+    ):
+        super().__init__()
+        
+        self.type = type
+        self.time = time
+        self.actor = actor
+        self.content = content
+    
+    def _restore_defaults(self) -> None:
+        self.type = DataHubAiConversationMessageTypeClass.TEXT
+        self.time = int()
+        self.actor = DataHubAiConversationActorClass._construct_with_defaults()
+        self.content = DataHubAiConversationMessageContentClass._construct_with_defaults()
+    
+    
+    @property
+    def type(self) -> Union[str, "DataHubAiConversationMessageTypeClass"]:
+        """The type of message"""
+        return self._inner_dict.get('type')  # type: ignore
+    
+    @type.setter
+    def type(self, value: Union[str, "DataHubAiConversationMessageTypeClass"]) -> None:
+        self._inner_dict['type'] = value
+    
+    
+    @property
+    def time(self) -> int:
+        """Timestamp in milliseconds"""
+        return self._inner_dict.get('time')  # type: ignore
+    
+    @time.setter
+    def time(self, value: int) -> None:
+        self._inner_dict['time'] = value
+    
+    
+    @property
+    def actor(self) -> "DataHubAiConversationActorClass":
+        """Messages in the conversation"""
+        return self._inner_dict.get('actor')  # type: ignore
+    
+    @actor.setter
+    def actor(self, value: "DataHubAiConversationActorClass") -> None:
+        self._inner_dict['actor'] = value
+    
+    
+    @property
+    def content(self) -> "DataHubAiConversationMessageContentClass":
+        """The contents of the message. """
+        return self._inner_dict.get('content')  # type: ignore
+    
+    @content.setter
+    def content(self, value: "DataHubAiConversationMessageContentClass") -> None:
+        self._inner_dict['content'] = value
+    
+    
+class DataHubAiConversationMessageContentClass(DictWrapper):
+    """Conversation message content info """
+    
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.conversation.DataHubAiConversationMessageContent")
+    def __init__(self,
+        text: str,
+    ):
+        super().__init__()
+        
+        self.text = text
+    
+    def _restore_defaults(self) -> None:
+        self.text = str()
+    
+    
+    @property
+    def text(self) -> str:
+        """Raw message text"""
+        return self._inner_dict.get('text')  # type: ignore
+    
+    @text.setter
+    def text(self, value: str) -> None:
+        self._inner_dict['text'] = value
+    
+    
+class DataHubAiConversationMessageTypeClass(object):
+    # No docs available.
+    
+    TEXT = "TEXT"
+    TOOL_CALL = "TOOL_CALL"
+    TOOL_RESULT = "TOOL_RESULT"
+    THINKING = "THINKING"
+    
+    
+class DataHubAiConversationOriginTypeClass(object):
+    # No docs available.
+    
+    DATAHUB_UI = "DATAHUB_UI"
+    """Chat session originated in DataHub UI """
+    
+    
+    
 class DashboardInfoClass(_Aspect):
     """Information about a dashboard"""
 
@@ -23968,6 +24196,35 @@ class DataHubActionKeyClass(_Aspect):
         self._inner_dict['id'] = value
     
     
+class DataHubAiConversationKeyClass(_Aspect):
+    """Key for a DataHub conversation"""
+
+
+    ASPECT_NAME = 'dataHubAiConversationKey'
+    ASPECT_INFO = {'keyForEntity': 'dataHubAiConversation', 'entityCategory': 'core', 'entityAspects': ['dataHubAiConversationInfo']}
+    RECORD_SCHEMA = get_schema_type("com.linkedin.pegasus2avro.metadata.key.DataHubAiConversationKey")
+
+    def __init__(self,
+        id: str,
+    ):
+        super().__init__()
+        
+        self.id = id
+    
+    def _restore_defaults(self) -> None:
+        self.id = str()
+    
+    
+    @property
+    def id(self) -> str:
+        """Unique identifier for the conversation"""
+        return self._inner_dict.get('id')  # type: ignore
+    
+    @id.setter
+    def id(self, value: str) -> None:
+        self._inner_dict['id'] = value
+    
+    
 class DataHubConnectionKeyClass(_Aspect):
     """Key for a Connection"""
 
@@ -41303,6 +41560,13 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.container.Container': ContainerClass,
     'com.linkedin.pegasus2avro.container.ContainerProperties': ContainerPropertiesClass,
     'com.linkedin.pegasus2avro.container.EditableContainerProperties': EditableContainerPropertiesClass,
+    'com.linkedin.pegasus2avro.conversation.DataHubAiConversationActor': DataHubAiConversationActorClass,
+    'com.linkedin.pegasus2avro.conversation.DataHubAiConversationActorType': DataHubAiConversationActorTypeClass,
+    'com.linkedin.pegasus2avro.conversation.DataHubAiConversationInfo': DataHubAiConversationInfoClass,
+    'com.linkedin.pegasus2avro.conversation.DataHubAiConversationMessage': DataHubAiConversationMessageClass,
+    'com.linkedin.pegasus2avro.conversation.DataHubAiConversationMessageContent': DataHubAiConversationMessageContentClass,
+    'com.linkedin.pegasus2avro.conversation.DataHubAiConversationMessageType': DataHubAiConversationMessageTypeClass,
+    'com.linkedin.pegasus2avro.conversation.DataHubAiConversationOriginType': DataHubAiConversationOriginTypeClass,
     'com.linkedin.pegasus2avro.dashboard.DashboardInfo': DashboardInfoClass,
     'com.linkedin.pegasus2avro.dashboard.DashboardUsageStatistics': DashboardUsageStatisticsClass,
     'com.linkedin.pegasus2avro.dashboard.DashboardUserUsageCounts': DashboardUserUsageCountsClass,
@@ -41499,6 +41763,7 @@ __SCHEMA_TYPES = {
     'com.linkedin.pegasus2avro.metadata.key.DataFlowKey': DataFlowKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubAccessTokenKey': DataHubAccessTokenKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubActionKey': DataHubActionKeyClass,
+    'com.linkedin.pegasus2avro.metadata.key.DataHubAiConversationKey': DataHubAiConversationKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubConnectionKey': DataHubConnectionKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubFileKey': DataHubFileKeyClass,
     'com.linkedin.pegasus2avro.metadata.key.DataHubIngestionSourceKey': DataHubIngestionSourceKeyClass,
@@ -42091,6 +42356,13 @@ __SCHEMA_TYPES = {
     'Container': ContainerClass,
     'ContainerProperties': ContainerPropertiesClass,
     'EditableContainerProperties': EditableContainerPropertiesClass,
+    'DataHubAiConversationActor': DataHubAiConversationActorClass,
+    'DataHubAiConversationActorType': DataHubAiConversationActorTypeClass,
+    'DataHubAiConversationInfo': DataHubAiConversationInfoClass,
+    'DataHubAiConversationMessage': DataHubAiConversationMessageClass,
+    'DataHubAiConversationMessageContent': DataHubAiConversationMessageContentClass,
+    'DataHubAiConversationMessageType': DataHubAiConversationMessageTypeClass,
+    'DataHubAiConversationOriginType': DataHubAiConversationOriginTypeClass,
     'DashboardInfo': DashboardInfoClass,
     'DashboardUsageStatistics': DashboardUsageStatisticsClass,
     'DashboardUserUsageCounts': DashboardUserUsageCountsClass,
@@ -42287,6 +42559,7 @@ __SCHEMA_TYPES = {
     'DataFlowKey': DataFlowKeyClass,
     'DataHubAccessTokenKey': DataHubAccessTokenKeyClass,
     'DataHubActionKey': DataHubActionKeyClass,
+    'DataHubAiConversationKey': DataHubAiConversationKeyClass,
     'DataHubConnectionKey': DataHubConnectionKeyClass,
     'DataHubFileKey': DataHubFileKeyClass,
     'DataHubIngestionSourceKey': DataHubIngestionSourceKeyClass,
@@ -42770,6 +43043,7 @@ ASPECT_CLASSES: List[Type[_Aspect]] = [
     MLTrainingRunPropertiesClass,
     EditableMLPrimaryKeyPropertiesClass,
     EvaluationDataClass,
+    DataHubAiConversationInfoClass,
     DataJobInfoClass,
     DataFlowInfoClass,
     VersionInfoClass,
@@ -42832,6 +43106,7 @@ ASPECT_CLASSES: List[Type[_Aspect]] = [
     ConstraintKeyClass,
     VersionSetKeyClass,
     MLPrimaryKeyKeyClass,
+    DataHubAiConversationKeyClass,
     AnomalyKeyClass,
     TestKeyClass,
     MLModelGroupKeyClass,
@@ -43070,6 +43345,7 @@ class AspectBag(TypedDict, total=False):
     mlTrainingRunProperties: MLTrainingRunPropertiesClass
     editableMlPrimaryKeyProperties: EditableMLPrimaryKeyPropertiesClass
     mlModelEvaluationData: EvaluationDataClass
+    dataHubAiConversationInfo: DataHubAiConversationInfoClass
     dataJobInfo: DataJobInfoClass
     dataFlowInfo: DataFlowInfoClass
     versionInfo: VersionInfoClass
@@ -43132,6 +43408,7 @@ class AspectBag(TypedDict, total=False):
     constraintKey: ConstraintKeyClass
     versionSetKey: VersionSetKeyClass
     mlPrimaryKeyKey: MLPrimaryKeyKeyClass
+    dataHubAiConversationKey: DataHubAiConversationKeyClass
     anomalyKey: AnomalyKeyClass
     testKey: TestKeyClass
     mlModelGroupKey: MLModelGroupKeyClass
@@ -43275,6 +43552,7 @@ KEY_ASPECTS: Dict[str, Type[_Aspect]] = {
     'constraint': ConstraintKeyClass,
     'versionSet': VersionSetKeyClass,
     'mlPrimaryKey': MLPrimaryKeyKeyClass,
+    'dataHubAiConversation': DataHubAiConversationKeyClass,
     'test': TestKeyClass,
     'mlModelGroup': MLModelGroupKeyClass,
     'dataHubSecret': DataHubSecretKeyClass,
@@ -43354,6 +43632,7 @@ ENTITY_TYPE_NAMES: List[str] = [
     'constraint',
     'versionSet',
     'mlPrimaryKey',
+    'dataHubAiConversation',
     'test',
     'mlModelGroup',
     'dataHubSecret',
@@ -43430,6 +43709,7 @@ EntityTypeName = Literal[
     'constraint',
     'versionSet',
     'mlPrimaryKey',
+    'dataHubAiConversation',
     'test',
     'mlModelGroup',
     'dataHubSecret',

acryl_datahub_cloud/metadata/schemas/DataHubAiConversationInfo.avsc

@@ -0,0 +1,198 @@
+{
+  "type": "record",
+  "Aspect": {
+    "name": "dataHubAiConversationInfo"
+  },
+  "name": "DataHubAiConversationInfo",
+  "namespace": "com.linkedin.pegasus2avro.conversation",
+  "fields": [
+    {
+      "type": [
+        "null",
+        "string"
+      ],
+      "name": "title",
+      "default": null,
+      "doc": "Optional title for the conversation. "
+    },
+    {
+      "type": {
+        "type": "array",
+        "items": {
+          "type": "record",
+          "name": "DataHubAiConversationMessage",
+          "namespace": "com.linkedin.pegasus2avro.conversation",
+          "fields": [
+            {
+              "type": {
+                "type": "enum",
+                "name": "DataHubAiConversationMessageType",
+                "namespace": "com.linkedin.pegasus2avro.conversation",
+                "symbols": [
+                  "TEXT",
+                  "TOOL_CALL",
+                  "TOOL_RESULT",
+                  "THINKING"
+                ]
+              },
+              "name": "type",
+              "doc": "The type of message"
+            },
+            {
+              "type": "long",
+              "name": "time",
+              "doc": "Timestamp in milliseconds"
+            },
+            {
+              "type": {
+                "type": "record",
+                "name": "DataHubAiConversationActor",
+                "namespace": "com.linkedin.pegasus2avro.conversation",
+                "fields": [
+                  {
+                    "type": {
+                      "type": "enum",
+                      "symbolDocs": {
+                        "AGENT": "An agent initiated message. ",
+                        "USER": "A user initiated message. "
+                      },
+                      "name": "DataHubAiConversationActorType",
+                      "namespace": "com.linkedin.pegasus2avro.conversation",
+                      "symbols": [
+                        "USER",
+                        "AGENT"
+                      ]
+                    },
+                    "name": "type",
+                    "doc": "The type of actor "
+                  },
+                  {
+                    "Relationship": {
+                      "entityTypes": [
+                        "corpuser",
+                        "corpGroup"
+                      ],
+                      "name": "AuthoredBy"
+                    },
+                    "java": {
+                      "class": "com.linkedin.pegasus2avro.common.urn.Urn"
+                    },
+                    "type": [
+                      "null",
+                      "string"
+                    ],
+                    "name": "actor",
+                    "default": null,
+                    "doc": "The actor that has sent the message.\nRequired if type is not AGENT.",
+                    "Urn": "Urn",
+                    "entityTypes": [
+                      "corpuser",
+                      "corpGroup"
+                    ]
+                  }
+                ]
+              },
+              "name": "actor",
+              "doc": "Messages in the conversation"
+            },
+            {
+              "type": {
+                "type": "record",
+                "name": "DataHubAiConversationMessageContent",
+                "namespace": "com.linkedin.pegasus2avro.conversation",
+                "fields": [
+                  {
+                    "type": "string",
+                    "name": "text",
+                    "doc": "Raw message text"
+                  }
+                ],
+                "doc": "Conversation message content info "
+              },
+              "name": "content",
+              "doc": "The contents of the message. "
+            }
+          ],
+          "doc": "Conversation info "
+        }
+      },
+      "name": "messages",
+      "doc": "Messages in the conversation"
+    },
+    {
+      "Searchable": {
+        "/actor": {
+          "fieldName": "creator",
+          "fieldType": "URN"
+        },
+        "/time": {
+          "fieldName": "createdAt",
+          "fieldType": "DATETIME"
+        }
+      },
+      "type": {
+        "type": "record",
+        "name": "AuditStamp",
+        "namespace": "com.linkedin.pegasus2avro.common",
+        "fields": [
+          {
+            "type": "long",
+            "name": "time",
+            "doc": "When did the resource/association/sub-resource move into the specific lifecycle stage represented by this AuditEvent."
+          },
+          {
+            "java": {
+              "class": "com.linkedin.pegasus2avro.common.urn.Urn"
+            },
+            "type": "string",
+            "name": "actor",
+            "doc": "The entity (e.g. a member URN) which will be credited for moving the resource/association/sub-resource into the specific lifecycle stage. It is also the one used to authorize the change.",
+            "Urn": "Urn"
+          },
+          {
+            "java": {
+              "class": "com.linkedin.pegasus2avro.common.urn.Urn"
+            },
+            "type": [
+              "null",
+              "string"
+            ],
+            "name": "impersonator",
+            "default": null,
+            "doc": "The entity (e.g. a service URN) which performs the change on behalf of the Actor and must be authorized to act as the Actor.",
+            "Urn": "Urn"
+          },
+          {
+            "type": [
+              "null",
+              "string"
+            ],
+            "name": "message",
+            "default": null,
+            "doc": "Additional context around how DataHub was informed of the particular change. For example: was the change created by an automated process, or manually."
+          }
+        ],
+        "doc": "Data captured on a resource/association/sub-resource level giving insight into when that resource/association/sub-resource moved into a particular lifecycle stage, and who acted to move it into that specific lifecycle stage."
+      },
+      "name": "created",
+      "doc": "The time and actor who initiated the conversation"
+    },
+    {
+      "type": {
+        "type": "enum",
+        "symbolDocs": {
+          "DATAHUB_UI": "Chat session originated in DataHub UI "
+        },
+        "name": "DataHubAiConversationOriginType",
+        "namespace": "com.linkedin.pegasus2avro.conversation",
+        "symbols": [
+          "DATAHUB_UI"
+        ]
+      },
+      "name": "originType",
+      "default": "DATAHUB_UI",
+      "doc": "Origin type for the chat session"
+    }
+  ],
+  "doc": "Conversation info "
+}

acryl_datahub_cloud/metadata/schemas/DataHubAiConversationKey.avsc

@@ -0,0 +1,22 @@
+{
+  "type": "record",
+  "Aspect": {
+    "name": "dataHubAiConversationKey",
+    "keyForEntity": "dataHubAiConversation",
+    "entityCategory": "core",
+    "entityAspects": [
+      "dataHubAiConversationInfo"
+    ]
+  },
+  "name": "DataHubAiConversationKey",
+  "namespace": "com.linkedin.pegasus2avro.metadata.key",
+  "fields": [
+    {
+      "Searchable": {},
+      "type": "string",
+      "name": "id",
+      "doc": "Unique identifier for the conversation"
+    }
+  ],
+  "doc": "Key for a DataHub conversation"
+}

{acryl_datahub_cloud-0.3.15rc1.dist-info → acryl_datahub_cloud-0.3.15rc3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acryl-datahub-cloud
-Version: 0.3.15rc1
+Version: 0.3.15rc3
 Requires-Python: >=3.10
 Requires-Dist: avro-gen3==0.7.16
 Requires-Dist: acryl-datahub
@@ -10,89 +10,89 @@ Requires-Dist: pytz
 Requires-Dist: types-croniter
 Requires-Dist: tzlocal
 Provides-Extra: datahub-lineage-features
-Requires-Dist: pandas; extra == "datahub-lineage-features"
 Requires-Dist: tenacity; extra == "datahub-lineage-features"
-Requires-Dist: opensearch-py==2.4.2; extra == "datahub-lineage-features"
-Requires-Dist: pyarrow; extra == "datahub-lineage-features"
+Requires-Dist: pandas; extra == "datahub-lineage-features"
 Requires-Dist: duckdb; extra == "datahub-lineage-features"
+Requires-Dist: pyarrow; extra == "datahub-lineage-features"
+Requires-Dist: opensearch-py==2.4.2; extra == "datahub-lineage-features"
 Provides-Extra: datahub-reporting-forms
 Requires-Dist: boto3; extra == "datahub-reporting-forms"
-Requires-Dist: pandas; extra == "datahub-reporting-forms"
 Requires-Dist: termcolor==2.5.0; extra == "datahub-reporting-forms"
-Requires-Dist: pyarrow; extra == "datahub-reporting-forms"
+Requires-Dist: pandas; extra == "datahub-reporting-forms"
 Requires-Dist: duckdb; extra == "datahub-reporting-forms"
+Requires-Dist: pyarrow; extra == "datahub-reporting-forms"
 Provides-Extra: datahub-reporting-extract-graph
 Requires-Dist: boto3; extra == "datahub-reporting-extract-graph"
 Requires-Dist: pandas; extra == "datahub-reporting-extract-graph"
-Requires-Dist: opensearch-py==2.4.2; extra == "datahub-reporting-extract-graph"
-Requires-Dist: pyarrow; extra == "datahub-reporting-extract-graph"
 Requires-Dist: duckdb; extra == "datahub-reporting-extract-graph"
+Requires-Dist: pyarrow; extra == "datahub-reporting-extract-graph"
+Requires-Dist: opensearch-py==2.4.2; extra == "datahub-reporting-extract-graph"
 Provides-Extra: datahub-reporting-extract-sql
-Requires-Dist: pyarrow; extra == "datahub-reporting-extract-sql"
-Requires-Dist: boto3; extra == "datahub-reporting-extract-sql"
 Requires-Dist: pandas; extra == "datahub-reporting-extract-sql"
 Requires-Dist: duckdb; extra == "datahub-reporting-extract-sql"
+Requires-Dist: boto3; extra == "datahub-reporting-extract-sql"
+Requires-Dist: pyarrow; extra == "datahub-reporting-extract-sql"
 Provides-Extra: datahub-usage-reporting
-Requires-Dist: elasticsearch==7.13.4; extra == "datahub-usage-reporting"
+Requires-Dist: numpy<2; extra == "datahub-usage-reporting"
 Requires-Dist: termcolor==2.5.0; extra == "datahub-usage-reporting"
+Requires-Dist: pyarrow<=18.0.0; extra == "datahub-usage-reporting"
+Requires-Dist: duckdb; extra == "datahub-usage-reporting"
+Requires-Dist: elasticsearch==7.13.4; extra == "datahub-usage-reporting"
+Requires-Dist: boto3; extra == "datahub-usage-reporting"
 Requires-Dist: polars==1.34.0; extra == "datahub-usage-reporting"
 Requires-Dist: scipy<=1.14.1; extra == "datahub-usage-reporting"
-Requires-Dist: pyarrow; extra == "datahub-usage-reporting"
-Requires-Dist: boto3; extra == "datahub-usage-reporting"
-Requires-Dist: duckdb; extra == "datahub-usage-reporting"
-Requires-Dist: pyarrow<=18.0.0; extra == "datahub-usage-reporting"
 Requires-Dist: pandas; extra == "datahub-usage-reporting"
+Requires-Dist: pyarrow; extra == "datahub-usage-reporting"
 Requires-Dist: opensearch-py==2.4.2; extra == "datahub-usage-reporting"
-Requires-Dist: numpy<2; extra == "datahub-usage-reporting"
 Provides-Extra: datahub-metadata-sharing
 Requires-Dist: tenacity; extra == "datahub-metadata-sharing"
 Provides-Extra: datahub-action-request-owner
 Requires-Dist: tenacity; extra == "datahub-action-request-owner"
 Provides-Extra: acryl-cs-issues
-Requires-Dist: jinja2; extra == "acryl-cs-issues"
-Requires-Dist: openai; extra == "acryl-cs-issues"
 Requires-Dist: zenpy; extra == "acryl-cs-issues"
+Requires-Dist: openai; extra == "acryl-cs-issues"
+Requires-Dist: jinja2; extra == "acryl-cs-issues"
 Requires-Dist: slack-sdk; extra == "acryl-cs-issues"
 Provides-Extra: datahub-forms-notifications
 Requires-Dist: tenacity; extra == "datahub-forms-notifications"
 Provides-Extra: all
-Requires-Dist: elasticsearch==7.13.4; extra == "all"
-Requires-Dist: tenacity; extra == "all"
-Requires-Dist: openai; extra == "all"
+Requires-Dist: zenpy; extra == "all"
 Requires-Dist: termcolor==2.5.0; extra == "all"
+Requires-Dist: numpy<2; extra == "all"
+Requires-Dist: pyarrow<=18.0.0; extra == "all"
+Requires-Dist: tenacity; extra == "all"
+Requires-Dist: duckdb; extra == "all"
+Requires-Dist: slack-sdk; extra == "all"
+Requires-Dist: elasticsearch==7.13.4; extra == "all"
+Requires-Dist: boto3; extra == "all"
 Requires-Dist: polars==1.34.0; extra == "all"
 Requires-Dist: scipy<=1.14.1; extra == "all"
-Requires-Dist: pyarrow; extra == "all"
-Requires-Dist: boto3; extra == "all"
 Requires-Dist: jinja2; extra == "all"
-Requires-Dist: duckdb; extra == "all"
-Requires-Dist: pyarrow<=18.0.0; extra == "all"
 Requires-Dist: pandas; extra == "all"
+Requires-Dist: openai; extra == "all"
+Requires-Dist: pyarrow; extra == "all"
 Requires-Dist: opensearch-py==2.4.2; extra == "all"
-Requires-Dist: zenpy; extra == "all"
-Requires-Dist: slack-sdk; extra == "all"
-Requires-Dist: numpy<2; extra == "all"
 Provides-Extra: dev
-Requires-Dist: openai; extra == "dev"
-Requires-Dist: polars==1.34.0; extra == "dev"
 Requires-Dist: pyarrow-stubs; extra == "dev"
+Requires-Dist: openai; extra == "dev"
+Requires-Dist: zenpy; extra == "dev"
 Requires-Dist: numpy<2; extra == "dev"
-Requires-Dist: duckdb; extra == "dev"
+Requires-Dist: polars==1.34.0; extra == "dev"
+Requires-Dist: pyarrow; extra == "dev"
+Requires-Dist: termcolor==2.5.0; extra == "dev"
+Requires-Dist: acryl-datahub[dev]; extra == "dev"
 Requires-Dist: slack-sdk; extra == "dev"
 Requires-Dist: tenacity; extra == "dev"
-Requires-Dist: scipy<=1.14.1; extra == "dev"
+Requires-Dist: duckdb; extra == "dev"
+Requires-Dist: scipy-stubs; extra == "dev"
 Requires-Dist: pandas-stubs; extra == "dev"
+Requires-Dist: jinja2; extra == "dev"
+Requires-Dist: scipy<=1.14.1; extra == "dev"
 Requires-Dist: pyarrow<=18.0.0; extra == "dev"
-Requires-Dist: pandas; extra == "dev"
-Requires-Dist: scipy-stubs; extra == "dev"
-Requires-Dist: zenpy; extra == "dev"
 Requires-Dist: elasticsearch==7.13.4; extra == "dev"
-Requires-Dist: termcolor==2.5.0; extra == "dev"
-Requires-Dist: jinja2; extra == "dev"
 Requires-Dist: boto3; extra == "dev"
-Requires-Dist: pyarrow; extra == "dev"
+Requires-Dist: pandas; extra == "dev"
 Requires-Dist: opensearch-py==2.4.2; extra == "dev"
-Requires-Dist: acryl-datahub[dev]; extra == "dev"
 Dynamic: provides-extra
 Dynamic: requires-dist
 Dynamic: requires-python

{acryl_datahub_cloud-0.3.15rc1.dist-info → acryl_datahub_cloud-0.3.15rc3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 acryl_datahub_cloud/__init__.py,sha256=axrMXkn0RW80YmuZgwUP_YQImcv6L28duZLWnW-gaNM,521
-acryl_datahub_cloud/_codegen_config.json,sha256=9_Hr77yGGuoyJhwT9Ov6lXlvk-nkSkYDXIkLswDvKlA,556
+acryl_datahub_cloud/_codegen_config.json,sha256=utyHcTdJiMn_sLpkOpUH4dYx5Oxb6oVZJ-Gy1ecdsRo,556
 acryl_datahub_cloud/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 acryl_datahub_cloud/acryl_cs_issues/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 acryl_datahub_cloud/acryl_cs_issues/acryl_customer.py,sha256=uqYPmluXYdlgyq3C09gxIU5nEkKiqHoZ53h2oN5etj0,25227
@@ -26,7 +26,7 @@ acryl_datahub_cloud/datahub_metadata_sharing/scroll_shared_entities.gql,sha256=N
 acryl_datahub_cloud/datahub_metadata_sharing/share_entity.gql,sha256=tJ0VkAekRQCxZ3TkaC0nVqMHQoILqf2J6J-HfbSRL0U,286
 acryl_datahub_cloud/datahub_reporting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 acryl_datahub_cloud/datahub_reporting/datahub_dataset.py,sha256=y3mlLbtg9VxrfyzUUxwRXIYFQ-2K0YUlcosNEFWsMeE,20015
-acryl_datahub_cloud/datahub_reporting/datahub_form_reporting.py,sha256=qXUgJHH2n85K7BPBDCPOoXuWCExYeEZcxLmILJk_MKQ,22608
+acryl_datahub_cloud/datahub_reporting/datahub_form_reporting.py,sha256=lXjEGAswg5SVHFxIYz2m0NU_tF-6_tpWpRTBsrQ3ZFA,23281
 acryl_datahub_cloud/datahub_reporting/extract_graph.py,sha256=n8DXMbGlAllWh5FcQ4bnUf3HYOiNYsSodWomigtmfWA,7896
 acryl_datahub_cloud/datahub_reporting/extract_sql.py,sha256=H5y3e35RuwXk23trhcCKsjFkStGlXZiYzdIYf0213Hw,9537
 acryl_datahub_cloud/datahub_reporting/forms.py,sha256=WUmJ3DLcKn4VcSplZFxmzdPrxQY2mYKqKc7hk8XZ9_Q,5939
@@ -41,14 +41,14 @@ acryl_datahub_cloud/datahub_usage_reporting/usage_feature_patch_builder.py,sha25
 acryl_datahub_cloud/datahub_usage_reporting/usage_feature_reporter.py,sha256=H9bs9Yssa90eMAC1Qlg5lzwkbddaVVdjKgsWsKsxbIU,83289
 acryl_datahub_cloud/elasticsearch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 acryl_datahub_cloud/elasticsearch/config.py,sha256=6QNBOmoQZu1cJrDIBZyvZgdQt0QLfP82hdQkPtP-4HE,1220
-acryl_datahub_cloud/elasticsearch/graph_service.py,sha256=Bpx_BaMcNSzsLtDes6Jej9VZuFF2iJRQBcBUCVE6sxg,3404
+acryl_datahub_cloud/elasticsearch/graph_service.py,sha256=nvs8rfjT-3wMr193X9b98tUxQYh2IXh_pWeO6NcERmw,5141
 acryl_datahub_cloud/lineage_features/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 acryl_datahub_cloud/lineage_features/source.py,sha256=fGtnIM4PsablJleNPZA_sz0uHEy0Ij2jfm1Le2t8E0Q,16638
 acryl_datahub_cloud/metadata/__init__.py,sha256=AjhXPjI6cnpdcrBRrE5gOWo15vv2TTl2ctU4UAnUN7A,238
-acryl_datahub_cloud/metadata/schema.avsc,sha256=VXKV0OCp7Lx8jSyqwQ7HFHzBAcOVxZ5xUpeGXrxnR80,1239850
-acryl_datahub_cloud/metadata/schema_classes.py,sha256=A6sP_k_qZ5nENsjisYjonm-zayNN3yWID0c5c7Kw8FQ,1631525
+acryl_datahub_cloud/metadata/schema.avsc,sha256=Ynlk6QU6E9Z9j_N26dnSg56xDKqvyCOjiO80VWGyLWM,1245399
+acryl_datahub_cloud/metadata/schema_classes.py,sha256=Z3-DXjtNo3xqRIxvxnSKPrKy-iDcLk8KYXFkOICvyks,1641283
 acryl_datahub_cloud/metadata/_urns/__init__.py,sha256=cOF3GHMDgPhmbLKbN02NPpuLGHSu0qNgQyBRv08eqF0,243
-acryl_datahub_cloud/metadata/_urns/urn_defs.py,sha256=k2fYG8RGGpwyuy-Xb8E_ZDm4T6H1T3aVDIiY5sFIpbM,172971
+acryl_datahub_cloud/metadata/_urns/urn_defs.py,sha256=Bij7q4iXp0EugcF3xZg-JrrSlvpavUBogUdCeCDJKWs,175353
 acryl_datahub_cloud/metadata/com/__init__.py,sha256=gsAIuTxzfJdI7a9ybZlgMIHMAYksM1SxGxXjtySgKSc,202
 acryl_datahub_cloud/metadata/com/linkedin/__init__.py,sha256=gsAIuTxzfJdI7a9ybZlgMIHMAYksM1SxGxXjtySgKSc,202
 acryl_datahub_cloud/metadata/com/linkedin/events/__init__.py,sha256=s_dR0plZF-rOxxIbE8ojekJqwiHzl2WYR-Z3kW6kKS0,298
@@ -69,6 +69,7 @@ acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/common/fieldtransformer/_
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/connection/__init__.py,sha256=qRtw-dB14pzVzgQ0pDK8kyBplNdpRxVKNj4D70e_FqI,564
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/constraint/__init__.py,sha256=cTdQMR-vfPXYcvlp1nqIQEL-YWhxkzKyGJXjte3fR5I,519
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/container/__init__.py,sha256=3yWt36KqDKFhRc9pzvt0AMnbMTlhKurGvT3BUvc25QU,510
+acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/conversation/__init__.py,sha256=pDMO-76YFHGpn2dee2kyHUS0G2grvf_RzQylmGiBQ-8,1154
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/dashboard/__init__.py,sha256=spcGVIipGkPWr6orwFXsOMBkAo3xo5eNiPxtA-Amhgw,656
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/datacontract/__init__.py,sha256=zgCDXvT8RXLBGJTO6WA4nPMRaHYNQnqjYdPaR9CGQdI,887
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/datahub/__init__.py,sha256=BnjtqbqyQIbROlo77oNsl5uZLjmBSQ6mVz9wQyf9otU,400
@@ -105,7 +106,7 @@ acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/ingestion/__init__.py,sha
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/link/__init__.py,sha256=4DfT4T_I6dh-iGk9LXYjrp98L9D66xZzM7Boqc7jmNg,388
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/logical/__init__.py,sha256=7SHiR-KzV1CkAimFy94SkcY0Xg0RlsIlLTUTGmGAW_U,290
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/metadata/__init__.py,sha256=gsAIuTxzfJdI7a9ybZlgMIHMAYksM1SxGxXjtySgKSc,202
-acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/metadata/key/__init__.py,sha256=7N620d7c-BPAXnIsVI5xQ5zpfjXFIJ65XIy2QRZ0ns0,6621
+acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/metadata/key/__init__.py,sha256=gJaTiK7iyh-WQ9vGGeShII4_fvjC7hgOc8dbzCNpu6I,6741
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/metadata/query/__init__.py,sha256=gsAIuTxzfJdI7a9ybZlgMIHMAYksM1SxGxXjtySgKSc,202
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/metadata/query/filter/__init__.py,sha256=DBP_QtxkFmC5q_kuk4dGjb4uOKbB4xKgqTWXGxmNbBQ,532
 acryl_datahub_cloud/metadata/com/linkedin/pegasus2avro/metadata/recommendation/__init__.py,sha256=6XhFJ-Qf_H8RkEG_kZV6TcUWa0z-RXNlze6MLhV85l4,927
@@ -216,6 +217,8 @@ acryl_datahub_cloud/metadata/schemas/DataHubAccessTokenKey.avsc,sha256=3EspNIxgb
 acryl_datahub_cloud/metadata/schemas/DataHubActionInfo.avsc,sha256=4sJNBSRJKTXxaofgDGHR04juUFmPy40Y8mbfgkwqAts,3168
 acryl_datahub_cloud/metadata/schemas/DataHubActionKey.avsc,sha256=7WGnI_vOvRKzwEASLRz_zltOwyOtLL-rsXkyToBPz30,508
 acryl_datahub_cloud/metadata/schemas/DataHubActionStatus.avsc,sha256=9OrdSvY6lUGr_srXyBynFk4JYguFbckbX0xJv6QH4Jo,6854
+acryl_datahub_cloud/metadata/schemas/DataHubAiConversationInfo.avsc,sha256=88eOu52rIleUrUxkx08K4nKgNhv9k3HFi6dbolCY5f8,6457
+acryl_datahub_cloud/metadata/schemas/DataHubAiConversationKey.avsc,sha256=kxnJInOOD_bLMFf76w_oqvDWuzo5uCv0Wo-_nIQI7hg,510
 acryl_datahub_cloud/metadata/schemas/DataHubConnectionDetails.avsc,sha256=IvZj6OA7HRvy-ZIIn0UbXdJNnyt_oTn16XIe5ZlcqGk,1661
 acryl_datahub_cloud/metadata/schemas/DataHubConnectionKey.avsc,sha256=n9pykqL4EBTLuFoVasxF-4a1v_5g578IrgXLPSLDxGQ,538
 acryl_datahub_cloud/metadata/schemas/DataHubFileInfo.avsc,sha256=tylUV_qx3kmoqEjQX747FUD0ZP-65kXBM7VidtJ9km8,6374
@@ -479,8 +482,8 @@ acryl_datahub_cloud/sdk/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 acryl_datahub_cloud/sdk/entities/assertion.py,sha256=-OILvHyKAI4-5mS2bb_P44Fvk6rBOOcvaxSMXfEYvRw,15077
 acryl_datahub_cloud/sdk/entities/monitor.py,sha256=NMrhJrWYNPvorxA33S_5FOl8YCtSmmeAavTzFLtWcOo,9665
 acryl_datahub_cloud/sdk/entities/subscription.py,sha256=0zl3LxkFM_hXBE8iCy8g6Nen-udJF784RmtPZ73QCb8,2801
-acryl_datahub_cloud-0.3.15rc1.dist-info/METADATA,sha256=FQ2yNJVQ-8ijeV0jqpOvn2hdrkG-naIqWQ7USKAWKeU,4615
-acryl_datahub_cloud-0.3.15rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-acryl_datahub_cloud-0.3.15rc1.dist-info/entry_points.txt,sha256=veuyIaEzm7JF2q-C8Q-RcSV6V5Y9LvnVvIhTjiT5WUs,1342
-acryl_datahub_cloud-0.3.15rc1.dist-info/top_level.txt,sha256=EwgCxfX-DzJANwxj-Mx_j4TOfAFhmc_FgMbRPzWsoZs,20
-acryl_datahub_cloud-0.3.15rc1.dist-info/RECORD,,
+acryl_datahub_cloud-0.3.15rc3.dist-info/METADATA,sha256=K695K8adiqaXWDZOSxWIz1xJ0f2X_nyWQ3AEGCsmzvg,4615
+acryl_datahub_cloud-0.3.15rc3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+acryl_datahub_cloud-0.3.15rc3.dist-info/entry_points.txt,sha256=veuyIaEzm7JF2q-C8Q-RcSV6V5Y9LvnVvIhTjiT5WUs,1342
+acryl_datahub_cloud-0.3.15rc3.dist-info/top_level.txt,sha256=EwgCxfX-DzJANwxj-Mx_j4TOfAFhmc_FgMbRPzWsoZs,20
+acryl_datahub_cloud-0.3.15rc3.dist-info/RECORD,,