datahub-agent-context 1.3.1.8__py3-none-any.whl

datahub_agent_context/mcp_tools/gql/queries.gql

@@ -0,0 +1,51 @@
+fragment platformFields on DataPlatform {
+  urn
+  name
+  __typename
+}
+
+fragment query on QueryEntity {
+  urn
+  properties {
+    name
+    description
+    source
+    statement {
+      value
+      language
+    }
+    # created {
+    #   time
+    #   actor
+    # }
+    lastModified {
+      # time
+      actor
+    }
+  }
+  platform {
+    ...platformFields
+  }
+  subjects {
+    dataset {
+      urn
+      name
+    }
+    schemaField {
+      urn
+      type
+      fieldPath
+    }
+  }
+}
+
+query listQueries($input: ListQueriesInput!) {
+  listQueries(input: $input) {
+    start
+    total
+    count
+    queries {
+      ...query
+    }
+  }
+}

datahub_agent_context/mcp_tools/gql/query_entity.gql

@@ -0,0 +1,37 @@
+query GetQueryEntity($urn: String!) {
+  entity(urn: $urn) {
+    urn
+    type
+    ... on QueryEntity {
+      urn
+      properties {
+        name
+        description
+        source
+        statement {
+          value
+          language
+        }
+        lastModified {
+          actor
+        }
+      }
+      platform {
+        urn
+        name
+      }
+      subjects {
+        dataset {
+          urn
+          name
+        }
+        schemaField {
+          urn
+          type
+          fieldPath
+        }
+      }
+    }
+  }
+}
+

datahub_agent_context/mcp_tools/gql/read_documents.gql

@@ -0,0 +1,16 @@
+# GraphQL query for fetching document content by URNs
+# Used by grep_documents to search within document content
+
+query readDocuments($urns: [String!]!) {
+  entities(urns: $urns) {
+    urn
+    ... on Document {
+      info {
+        title
+        contents {
+          text
+        }
+      }
+    }
+  }
+}

datahub_agent_context/mcp_tools/gql/search.gql

@@ -0,0 +1,242 @@
+# Simplified platform info for search results
+fragment SimplePlatformFields on DataPlatform {
+  urn
+  type
+  name
+  properties {
+    displayName
+    logoUrl
+  }
+}
+
+fragment SearchEntityInfo on Entity {
+  urn
+
+  # For some entity types, the urns are not human-readable. For those,
+  # we pull the name as well.
+  ... on Dataset {
+    platform {
+      ...SimplePlatformFields
+    }
+    properties {
+      name
+      description
+    }
+  }
+  ... on Chart {
+    platform {
+      ...SimplePlatformFields
+    }
+    properties {
+      name
+      description
+    }
+  }
+  ... on Dashboard {
+    platform {
+      ...SimplePlatformFields
+    }
+    properties {
+      name
+      description
+    }
+  }
+  ... on Container {
+    platform {
+      ...SimplePlatformFields
+    }
+    properties {
+      name
+      description
+    }
+  }
+  ... on GlossaryTerm {
+    properties {
+      name
+      description
+    }
+  }
+  ... on GlossaryNode {
+    properties {
+      name
+      description
+    }
+  }
+  ... on Domain {
+    properties {
+      name
+      description
+    }
+  }
+  ... on DataProduct {
+    properties {
+      name
+      description
+    }
+  }
+  ... on DataJob {
+    properties {
+      name
+      description
+    }
+  }
+  ... on DataFlow {
+    platform {
+      ...SimplePlatformFields
+    }
+    properties {
+      name
+      description
+    }
+  }
+  ... on MLModel {
+    name
+    description
+    platform {
+      ...SimplePlatformFields
+    }
+  }
+  ... on MLModelGroup {
+    name
+    description
+    platform {
+      ...SimplePlatformFields
+    }
+  }
+  ... on MLFeature {
+    name
+    description
+  }
+  ... on MLFeatureTable {
+    name
+    description
+    platform {
+      ...SimplePlatformFields
+    }
+  }
+  ... on MLPrimaryKey {
+    name
+    description
+  }
+  ... on Tag {
+    name
+    description
+    properties {
+      name
+      description
+    }
+  }
+  ... on StructuredPropertyEntity {
+    definition {
+      displayName
+      qualifiedName
+      description
+    }
+  }
+  ... on CorpUser {
+    username
+    properties {
+      displayName
+      fullName
+      title
+      email
+    }
+  }
+  ... on CorpGroup {
+    name
+    info {
+      displayName
+      description
+      email
+    }
+  }
+}
+
+fragment FacetEntityInfo on Entity {
+  ... on Dataset {
+    name
+    properties {
+      name
+    }
+  }
+  ... on Container {
+    subTypes {
+      typeNames
+    }
+    properties {
+      name
+    }
+  }
+  ... on GlossaryTerm {
+    properties {
+      name
+    }
+  }
+  ... on Tag {
+    name
+    properties {
+      name
+    }
+  }
+  ... on CorpUser {
+    username
+    properties {
+      displayName
+      fullName
+      title
+      email
+    }
+  }
+  ... on CorpGroup {
+    name
+    info {
+      displayName
+      description
+      email
+    }
+  }
+}
+
+query search(
+  $types: [EntityType!]
+  $query: String!
+  $orFilters: [AndFilterInput!]
+  $count: Int!
+  $start: Int!
+  $viewUrn: String
+  $sortInput: SearchSortInput
+) {
+  searchAcrossEntities(
+    input: {
+      query: $query
+      count: $count
+      start: $start
+      types: $types
+      orFilters: $orFilters
+      viewUrn: $viewUrn
+      sortInput: $sortInput
+      searchFlags: { skipHighlighting: true, maxAggValues: 5 }
+    }
+  ) {
+    start
+    count
+    total
+    searchResults {
+      entity {
+        ...SearchEntityInfo
+      }
+    }
+    facets {
+      field
+      displayName
+      aggregations {
+        value
+        count
+        displayName
+        entity {
+          ...FacetEntityInfo
+        }
+      }
+    }
+  }
+}