acryl-datahub-cloud 0.3.15rc2__tar.gz → 0.3.15rc4__tar.gz

{acryl_datahub_cloud-0.3.15rc2 → acryl_datahub_cloud-0.3.15rc4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acryl-datahub-cloud
-Version: 0.3.15rc2
+Version: 0.3.15rc4
 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: pyarrow; extra == "datahub-lineage-features"
-Requires-Dist: opensearch-py==2.4.2; extra == "datahub-lineage-features"
 Requires-Dist: tenacity; extra == "datahub-lineage-features"
 Requires-Dist: duckdb; extra == "datahub-lineage-features"
+Requires-Dist: pyarrow; extra == "datahub-lineage-features"
 Requires-Dist: pandas; extra == "datahub-lineage-features"
+Requires-Dist: opensearch-py==2.4.2; extra == "datahub-lineage-features"
 Provides-Extra: datahub-reporting-forms
-Requires-Dist: pyarrow; extra == "datahub-reporting-forms"
-Requires-Dist: boto3; extra == "datahub-reporting-forms"
 Requires-Dist: duckdb; extra == "datahub-reporting-forms"
-Requires-Dist: termcolor==2.5.0; extra == "datahub-reporting-forms"
+Requires-Dist: boto3; extra == "datahub-reporting-forms"
+Requires-Dist: pyarrow; extra == "datahub-reporting-forms"
 Requires-Dist: pandas; extra == "datahub-reporting-forms"
+Requires-Dist: termcolor==2.5.0; extra == "datahub-reporting-forms"
 Provides-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"
-Requires-Dist: boto3; extra == "datahub-reporting-extract-graph"
 Requires-Dist: duckdb; extra == "datahub-reporting-extract-graph"
+Requires-Dist: boto3; extra == "datahub-reporting-extract-graph"
+Requires-Dist: pyarrow; 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"
 Provides-Extra: datahub-reporting-extract-sql
-Requires-Dist: boto3; extra == "datahub-reporting-extract-sql"
 Requires-Dist: pyarrow; extra == "datahub-reporting-extract-sql"
-Requires-Dist: duckdb; 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"
 Provides-Extra: datahub-usage-reporting
-Requires-Dist: pyarrow; extra == "datahub-usage-reporting"
-Requires-Dist: opensearch-py==2.4.2; extra == "datahub-usage-reporting"
+Requires-Dist: pyarrow<=18.0.0; extra == "datahub-usage-reporting"
+Requires-Dist: scipy<=1.14.1; extra == "datahub-usage-reporting"
 Requires-Dist: boto3; extra == "datahub-usage-reporting"
-Requires-Dist: elasticsearch==7.13.4; extra == "datahub-usage-reporting"
-Requires-Dist: numpy<2; extra == "datahub-usage-reporting"
-Requires-Dist: pandas; 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: opensearch-py==2.4.2; extra == "datahub-usage-reporting"
+Requires-Dist: numpy<2; extra == "datahub-usage-reporting"
 Requires-Dist: duckdb; extra == "datahub-usage-reporting"
+Requires-Dist: pyarrow; extra == "datahub-usage-reporting"
+Requires-Dist: pandas; 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: elasticsearch==7.13.4; 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: openai; extra == "acryl-cs-issues"
 Requires-Dist: slack-sdk; extra == "acryl-cs-issues"
-Requires-Dist: jinja2; 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"
 Provides-Extra: datahub-forms-notifications
 Requires-Dist: tenacity; extra == "datahub-forms-notifications"
 Provides-Extra: all
-Requires-Dist: pyarrow; extra == "all"
-Requires-Dist: opensearch-py==2.4.2; extra == "all"
+Requires-Dist: pyarrow<=18.0.0; extra == "all"
 Requires-Dist: tenacity; extra == "all"
+Requires-Dist: scipy<=1.14.1; extra == "all"
 Requires-Dist: boto3; extra == "all"
-Requires-Dist: slack-sdk; extra == "all"
-Requires-Dist: elasticsearch==7.13.4; extra == "all"
+Requires-Dist: polars==1.34.0; extra == "all"
+Requires-Dist: openai; extra == "all"
+Requires-Dist: opensearch-py==2.4.2; extra == "all"
 Requires-Dist: jinja2; extra == "all"
 Requires-Dist: numpy<2; extra == "all"
-Requires-Dist: openai; extra == "all"
-Requires-Dist: pandas; extra == "all"
-Requires-Dist: polars==1.34.0; extra == "all"
-Requires-Dist: scipy<=1.14.1; extra == "all"
+Requires-Dist: slack-sdk; extra == "all"
 Requires-Dist: duckdb; extra == "all"
+Requires-Dist: pyarrow; extra == "all"
+Requires-Dist: pandas; extra == "all"
 Requires-Dist: termcolor==2.5.0; extra == "all"
-Requires-Dist: pyarrow<=18.0.0; extra == "all"
 Requires-Dist: zenpy; extra == "all"
+Requires-Dist: elasticsearch==7.13.4; extra == "all"
 Provides-Extra: dev
-Requires-Dist: pyarrow-stubs; extra == "dev"
-Requires-Dist: scipy-stubs; extra == "dev"
-Requires-Dist: tenacity; extra == "dev"
-Requires-Dist: elasticsearch==7.13.4; extra == "dev"
-Requires-Dist: numpy<2; extra == "dev"
-Requires-Dist: pandas; extra == "dev"
 Requires-Dist: pandas-stubs; extra == "dev"
-Requires-Dist: termcolor==2.5.0; extra == "dev"
+Requires-Dist: boto3; extra == "dev"
+Requires-Dist: openai; extra == "dev"
+Requires-Dist: opensearch-py==2.4.2; extra == "dev"
 Requires-Dist: pyarrow; extra == "dev"
+Requires-Dist: elasticsearch==7.13.4; extra == "dev"
+Requires-Dist: scipy<=1.14.1; extra == "dev"
+Requires-Dist: pyarrow-stubs; extra == "dev"
 Requires-Dist: jinja2; extra == "dev"
+Requires-Dist: numpy<2; extra == "dev"
+Requires-Dist: tenacity; extra == "dev"
+Requires-Dist: duckdb; extra == "dev"
+Requires-Dist: pyarrow<=18.0.0; extra == "dev"
+Requires-Dist: scipy-stubs; extra == "dev"
+Requires-Dist: polars==1.34.0; extra == "dev"
 Requires-Dist: slack-sdk; extra == "dev"
+Requires-Dist: pandas; extra == "dev"
+Requires-Dist: termcolor==2.5.0; extra == "dev"
 Requires-Dist: zenpy; extra == "dev"
 Requires-Dist: acryl-datahub[dev]; extra == "dev"
-Requires-Dist: openai; extra == "dev"
-Requires-Dist: scipy<=1.14.1; extra == "dev"
-Requires-Dist: duckdb; extra == "dev"
-Requires-Dist: opensearch-py==2.4.2; extra == "dev"
-Requires-Dist: boto3; extra == "dev"
-Requires-Dist: polars==1.34.0; extra == "dev"
-Requires-Dist: pyarrow<=18.0.0; extra == "dev"
 Dynamic: provides-extra
 Dynamic: requires-dist
 Dynamic: requires-python

{acryl_datahub_cloud-0.3.15rc2 → acryl_datahub_cloud-0.3.15rc4}/src/acryl_datahub_cloud/_codegen_config.json

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

{acryl_datahub_cloud-0.3.15rc2 → acryl_datahub_cloud-0.3.15rc4}/src/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-0.3.15rc2 → acryl_datahub_cloud-0.3.15rc4}/src/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-0.3.15rc2 → acryl_datahub_cloud-0.3.15rc4}/src/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-0.3.15rc4/src/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-0.3.15rc2 → acryl_datahub_cloud-0.3.15rc4}/src/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-0.3.15rc2 → acryl_datahub_cloud-0.3.15rc4}/src/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": {