cartography 0.102.0rc1__py3-none-any.whl → 0.103.0__py3-none-any.whl

cartography/models/github/users.py

@@ -27,6 +27,7 @@ The main importance of having two schemas is to allow the two sets of users to b
 an unaffiliated user, but the user already exists in the graph (perhaps they are members of another GitHub orgs for
 example), then loading the unaffiliated user will not blank out the 'has_2fa_enabled' property.
 """
+
 from dataclasses import dataclass
 
 from cartography.models.core.common import PropertyRef
@@ -43,20 +44,20 @@ from cartography.models.core.relationships import TargetNodeMatcher
 @dataclass(frozen=True)
 class BaseGitHubUserNodeProperties(CartographyNodeProperties):
     # core properties in all GitHubUser nodes
-    id: PropertyRef = PropertyRef('url')
-    lastupdated: PropertyRef = PropertyRef('lastupdated', set_in_kwargs=True)
-    fullname: PropertyRef = PropertyRef('name')
-    username: PropertyRef = PropertyRef('login', extra_index=True)
-    is_site_admin: PropertyRef = PropertyRef('isSiteAdmin')
-    is_enterprise_owner: PropertyRef = PropertyRef('isEnterpriseOwner')
-    email: PropertyRef = PropertyRef('email')
-    company: PropertyRef = PropertyRef('company')
+    id: PropertyRef = PropertyRef("url")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    fullname: PropertyRef = PropertyRef("name")
+    username: PropertyRef = PropertyRef("login", extra_index=True)
+    is_site_admin: PropertyRef = PropertyRef("isSiteAdmin")
+    is_enterprise_owner: PropertyRef = PropertyRef("isEnterpriseOwner")
+    email: PropertyRef = PropertyRef("email")
+    company: PropertyRef = PropertyRef("company")
 
 
 @dataclass(frozen=True)
 class GitHubOrganizationUserNodeProperties(BaseGitHubUserNodeProperties):
     # specified for affiliated users only. The GitHub api does not return this property for unaffiliated users.
-    has_2fa_enabled: PropertyRef = PropertyRef('hasTwoFactorEnabled')
+    has_2fa_enabled: PropertyRef = PropertyRef("hasTwoFactorEnabled")
 
 
 @dataclass(frozen=True)
@@ -67,46 +68,54 @@ class GitHubUnaffiliatedUserNodeProperties(BaseGitHubUserNodeProperties):
 
 @dataclass(frozen=True)
 class GitHubUserToOrganizationRelProperties(CartographyRelProperties):
-    lastupdated: PropertyRef = PropertyRef('lastupdated', set_in_kwargs=True)
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
 
 
 @dataclass(frozen=True)
 class GitHubUserMemberOfOrganizationRel(CartographyRelSchema):
-    target_node_label: str = 'GitHubOrganization'
+    target_node_label: str = "GitHubOrganization"
     target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
-        {'id': PropertyRef('MEMBER_OF')},
+        {"id": PropertyRef("MEMBER_OF")},
     )
     direction: LinkDirection = LinkDirection.OUTWARD
     rel_label: str = "MEMBER_OF"
-    properties: GitHubUserToOrganizationRelProperties = GitHubUserToOrganizationRelProperties()
+    properties: GitHubUserToOrganizationRelProperties = (
+        GitHubUserToOrganizationRelProperties()
+    )
 
 
 @dataclass(frozen=True)
 class GitHubUserAdminOfOrganizationRel(CartographyRelSchema):
-    target_node_label: str = 'GitHubOrganization'
+    target_node_label: str = "GitHubOrganization"
     target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
-        {'id': PropertyRef('ADMIN_OF')},
+        {"id": PropertyRef("ADMIN_OF")},
     )
     direction: LinkDirection = LinkDirection.OUTWARD
     rel_label: str = "ADMIN_OF"
-    properties: GitHubUserToOrganizationRelProperties = GitHubUserToOrganizationRelProperties()
+    properties: GitHubUserToOrganizationRelProperties = (
+        GitHubUserToOrganizationRelProperties()
+    )
 
 
 @dataclass(frozen=True)
 class GitHubUserUnaffiliatedOrganizationRel(CartographyRelSchema):
-    target_node_label: str = 'GitHubOrganization'
+    target_node_label: str = "GitHubOrganization"
     target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
-        {'id': PropertyRef('UNAFFILIATED')},
+        {"id": PropertyRef("UNAFFILIATED")},
     )
     direction: LinkDirection = LinkDirection.OUTWARD
     rel_label: str = "UNAFFILIATED"
-    properties: GitHubUserToOrganizationRelProperties = GitHubUserToOrganizationRelProperties()
+    properties: GitHubUserToOrganizationRelProperties = (
+        GitHubUserToOrganizationRelProperties()
+    )
 
 
 @dataclass(frozen=True)
 class GitHubOrganizationUserSchema(CartographyNodeSchema):
-    label: str = 'GitHubUser'
-    properties: GitHubOrganizationUserNodeProperties = GitHubOrganizationUserNodeProperties()
+    label: str = "GitHubUser"
+    properties: GitHubOrganizationUserNodeProperties = (
+        GitHubOrganizationUserNodeProperties()
+    )
     other_relationships: OtherRelationships = OtherRelationships(
         [
             GitHubUserMemberOfOrganizationRel(),
@@ -118,8 +127,10 @@ class GitHubOrganizationUserSchema(CartographyNodeSchema):
 
 @dataclass(frozen=True)
 class GitHubUnaffiliatedUserSchema(CartographyNodeSchema):
-    label: str = 'GitHubUser'
-    properties: GitHubUnaffiliatedUserNodeProperties = GitHubUnaffiliatedUserNodeProperties()
+    label: str = "GitHubUser"
+    properties: GitHubUnaffiliatedUserNodeProperties = (
+        GitHubUnaffiliatedUserNodeProperties()
+    )
     other_relationships: OtherRelationships = OtherRelationships(
         [
             GitHubUserUnaffiliatedOrganizationRel(),

cartography/models/kandji/device.py

@@ -12,37 +12,43 @@ from cartography.models.core.relationships import TargetNodeMatcher
 
 @dataclass(frozen=True)
 class KandjiDeviceNodeProperties(CartographyNodeProperties):
-    id: PropertyRef = PropertyRef('id')
-    lastupdated: PropertyRef = PropertyRef('lastupdated', set_in_kwargs=True)
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
 
-    device_id: PropertyRef = PropertyRef('device_id')
-    device_name: PropertyRef = PropertyRef('device_name')
-    last_check_in: PropertyRef = PropertyRef('last_check_in')
-    model: PropertyRef = PropertyRef('model')
-    os_version: PropertyRef = PropertyRef('os_version')
-    platform: PropertyRef = PropertyRef('platform')
-    serial_number: PropertyRef = PropertyRef('serial_number', extra_index=True)
+    device_id: PropertyRef = PropertyRef("device_id")
+    device_name: PropertyRef = PropertyRef("device_name")
+    last_check_in: PropertyRef = PropertyRef("last_check_in")
+    model: PropertyRef = PropertyRef("model")
+    os_version: PropertyRef = PropertyRef("os_version")
+    platform: PropertyRef = PropertyRef("platform")
+    serial_number: PropertyRef = PropertyRef("serial_number", extra_index=True)
 
 
 @dataclass(frozen=True)
 class KandjiTenantToKandjiDeviceRelProperties(CartographyRelProperties):
-    lastupdated: PropertyRef = PropertyRef('lastupdated', set_in_kwargs=True)
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
 
 
 @dataclass(frozen=True)
 # (:KandjiDevice)-[:ENROLLED_TO]->(:KandjiTenant)
 class KandjiTenantToKandjiDeviceRel(CartographyRelSchema):
-    target_node_label: str = 'KandjiTenant'
+    target_node_label: str = "KandjiTenant"
     target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
-        {'id': PropertyRef('TENANT_ID', set_in_kwargs=True)},
+        {"id": PropertyRef("TENANT_ID", set_in_kwargs=True)},
     )
     direction: LinkDirection = LinkDirection.OUTWARD
     rel_label: str = "ENROLLED_TO"
-    properties: KandjiTenantToKandjiDeviceRelProperties = KandjiTenantToKandjiDeviceRelProperties()
+    properties: KandjiTenantToKandjiDeviceRelProperties = (
+        KandjiTenantToKandjiDeviceRelProperties()
+    )
 
 
 @dataclass(frozen=True)
 class KandjiDeviceSchema(CartographyNodeSchema):
-    label: str = 'KandjiDevice'  # The label of the node
-    properties: KandjiDeviceNodeProperties = KandjiDeviceNodeProperties()  # An object representing all properties
-    sub_resource_relationship: KandjiTenantToKandjiDeviceRel = KandjiTenantToKandjiDeviceRel()
+    label: str = "KandjiDevice"  # The label of the node
+    properties: KandjiDeviceNodeProperties = (
+        KandjiDeviceNodeProperties()
+    )  # An object representing all properties
+    sub_resource_relationship: KandjiTenantToKandjiDeviceRel = (
+        KandjiTenantToKandjiDeviceRel()
+    )

cartography/models/kandji/tenant.py

@@ -7,11 +7,13 @@ from cartography.models.core.nodes import CartographyNodeSchema
 
 @dataclass(frozen=True)
 class KandjiTenantNodeProperties(CartographyNodeProperties):
-    id: PropertyRef = PropertyRef('id')
-    lastupdated: PropertyRef = PropertyRef('lastupdated', set_in_kwargs=True)
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
 
 
 @dataclass(frozen=True)
 class KandjiTenantSchema(CartographyNodeSchema):
-    label: str = 'KandjiTenant'  # The label of the node
-    properties: KandjiTenantNodeProperties = KandjiTenantNodeProperties()  # An object representing all properties
+    label: str = "KandjiTenant"  # The label of the node
+    properties: KandjiTenantNodeProperties = (
+        KandjiTenantNodeProperties()
+    )  # An object representing all properties

cartography/models/lastpass/tenant.py

@@ -7,11 +7,11 @@ from cartography.models.core.nodes import CartographyNodeSchema
 
 @dataclass(frozen=True)
 class LastpassTenantNodeProperties(CartographyNodeProperties):
-    id: PropertyRef = PropertyRef('id')
-    lastupdated: PropertyRef = PropertyRef('lastupdated', set_in_kwargs=True)
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
 
 
 @dataclass(frozen=True)
 class LastpassTenantSchema(CartographyNodeSchema):
-    label: str = 'LastpassTenant'
+    label: str = "LastpassTenant"
     properties: LastpassTenantNodeProperties = LastpassTenantNodeProperties()

cartography/models/lastpass/user.py

@@ -13,38 +13,38 @@ from cartography.models.core.relationships import TargetNodeMatcher
 
 @dataclass(frozen=True)
 class LastpassUserNodeProperties(CartographyNodeProperties):
-    id: PropertyRef = PropertyRef('id')
-    lastupdated: PropertyRef = PropertyRef('lastupdated', set_in_kwargs=True)
-    name: PropertyRef = PropertyRef('fullname')
-    email: PropertyRef = PropertyRef('username', extra_index=True)
-    created: PropertyRef = PropertyRef('created')
-    last_pw_change: PropertyRef = PropertyRef('last_pw_change')
-    last_login: PropertyRef = PropertyRef('last_login')
-    neverloggedin: PropertyRef = PropertyRef('neverloggedin')
-    disabled: PropertyRef = PropertyRef('disabled')
-    admin: PropertyRef = PropertyRef('admin')
-    totalscore: PropertyRef = PropertyRef('totalscore')
-    mpstrength: PropertyRef = PropertyRef('mpstrength')
-    sites: PropertyRef = PropertyRef('sites')
-    notes: PropertyRef = PropertyRef('notes')
-    formfills: PropertyRef = PropertyRef('formfills')
-    applications: PropertyRef = PropertyRef('applications')
-    attachments: PropertyRef = PropertyRef('attachments')
-    password_reset_required: PropertyRef = PropertyRef('password_reset_required')
-    multifactor: PropertyRef = PropertyRef('multifactor')
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+    name: PropertyRef = PropertyRef("fullname")
+    email: PropertyRef = PropertyRef("username", extra_index=True)
+    created: PropertyRef = PropertyRef("created")
+    last_pw_change: PropertyRef = PropertyRef("last_pw_change")
+    last_login: PropertyRef = PropertyRef("last_login")
+    neverloggedin: PropertyRef = PropertyRef("neverloggedin")
+    disabled: PropertyRef = PropertyRef("disabled")
+    admin: PropertyRef = PropertyRef("admin")
+    totalscore: PropertyRef = PropertyRef("totalscore")
+    mpstrength: PropertyRef = PropertyRef("mpstrength")
+    sites: PropertyRef = PropertyRef("sites")
+    notes: PropertyRef = PropertyRef("notes")
+    formfills: PropertyRef = PropertyRef("formfills")
+    applications: PropertyRef = PropertyRef("applications")
+    attachments: PropertyRef = PropertyRef("attachments")
+    password_reset_required: PropertyRef = PropertyRef("password_reset_required")
+    multifactor: PropertyRef = PropertyRef("multifactor")
 
 
 @dataclass(frozen=True)
 class LastpassUserToHumanRelProperties(CartographyRelProperties):
-    lastupdated: PropertyRef = PropertyRef('lastupdated', set_in_kwargs=True)
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
 
 
 @dataclass(frozen=True)
 # (:LastpassUser)<-[:IDENTITY_LASTPASS]-(:Human)
 class LastpassHumanToUserRel(CartographyRelSchema):
-    target_node_label: str = 'Human'
+    target_node_label: str = "Human"
     target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
-        {'email': PropertyRef('username')},
+        {"email": PropertyRef("username")},
     )
     direction: LinkDirection = LinkDirection.INWARD
     rel_label: str = "IDENTITY_LASTPASS"
@@ -53,24 +53,28 @@ class LastpassHumanToUserRel(CartographyRelSchema):
 
 @dataclass(frozen=True)
 class LastpassTenantToLastpassUserRelProperties(CartographyRelProperties):
-    lastupdated: PropertyRef = PropertyRef('lastupdated', set_in_kwargs=True)
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
 
 
 @dataclass(frozen=True)
 # (:LastpassTenant)<-[:RESOURCE]-(:LastpassUser)
 class LastpassTenantToUserRel(CartographyRelSchema):
-    target_node_label: str = 'LastpassTenant'
+    target_node_label: str = "LastpassTenant"
     target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
-        {'id': PropertyRef('TENANT_ID', set_in_kwargs=True)},
+        {"id": PropertyRef("TENANT_ID", set_in_kwargs=True)},
     )
     direction: LinkDirection = LinkDirection.OUTWARD
     rel_label: str = "RESOURCE"
-    properties: LastpassTenantToLastpassUserRelProperties = LastpassTenantToLastpassUserRelProperties()
+    properties: LastpassTenantToLastpassUserRelProperties = (
+        LastpassTenantToLastpassUserRelProperties()
+    )
 
 
 @dataclass(frozen=True)
 class LastpassUserSchema(CartographyNodeSchema):
-    label: str = 'LastpassUser'
+    label: str = "LastpassUser"
     properties: LastpassUserNodeProperties = LastpassUserNodeProperties()
-    other_relationships: OtherRelationships = OtherRelationships(rels=[LastpassHumanToUserRel()])
+    other_relationships: OtherRelationships = OtherRelationships(
+        rels=[LastpassHumanToUserRel()],
+    )
     sub_resource_relationship: LastpassTenantToUserRel = LastpassTenantToUserRel()

cartography/models/openai/adminapikey.py

@@ -0,0 +1,90 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class OpenAIAdminApiKeyNodeProperties(CartographyNodeProperties):
+    object: PropertyRef = PropertyRef("object")
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    created_at: PropertyRef = PropertyRef("created_at")
+    last_used_at: PropertyRef = PropertyRef("last_used_at")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class OpenAIAdminApiKeyToOrganizationRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIOrganization)-[:RESOURCE]->(:OpenAIAdminApiKey)
+class OpenAIAdminApiKeyToOrganizationRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("ORG_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: OpenAIAdminApiKeyToOrganizationRelProperties = (
+        OpenAIAdminApiKeyToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIAdminApiKeyToUserRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIUser)-[:OWNS]->(:OpenAIAdminApiKey)
+class OpenAIAdminApiKeyToUserRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIUser"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("owner_user_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "OWNS"
+    properties: OpenAIAdminApiKeyToUserRelProperties = (
+        OpenAIAdminApiKeyToUserRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIAdminApiKeyToSARelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIServiceAccount)-[:OWNS]->(:OpenAIAdminApiKey)
+class OpenAIAdminApiKeyToSARel(CartographyRelSchema):
+    target_node_label: str = "OpenAIServiceAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("owner_sa_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "OWNS"
+    properties: OpenAIAdminApiKeyToSARelProperties = (
+        OpenAIAdminApiKeyToSARelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIAdminApiKeySchema(CartographyNodeSchema):
+    label: str = "OpenAIAdminApiKey"
+    properties: OpenAIAdminApiKeyNodeProperties = OpenAIAdminApiKeyNodeProperties()
+    sub_resource_relationship: OpenAIAdminApiKeyToOrganizationRel = (
+        OpenAIAdminApiKeyToOrganizationRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [OpenAIAdminApiKeyToUserRel(), OpenAIAdminApiKeyToSARel()],
+    )

cartography/models/openai/apikey.py

@@ -0,0 +1,84 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class OpenAIApiKeyNodeProperties(CartographyNodeProperties):
+    object: PropertyRef = PropertyRef("object")
+    name: PropertyRef = PropertyRef("name")
+    created_at: PropertyRef = PropertyRef("created_at")
+    last_used_at: PropertyRef = PropertyRef("last_used_at")
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class OpenAIApiKeyToProjectRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIApiKey)<-[:RESOURCE]-(:OpenAIProject)
+class OpenAIApiKeyToProjectRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIProject"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("project_id", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: OpenAIApiKeyToProjectRelProperties = (
+        OpenAIApiKeyToProjectRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIApiKeyToUserRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIUser)-[:OWNS]->(:OpenAIApiKey)
+class OpenAIApiKeyToUserRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIUser"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("owner_user_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "OWNS"
+    properties: OpenAIApiKeyToUserRelProperties = OpenAIApiKeyToUserRelProperties()
+
+
+@dataclass(frozen=True)
+class OpenAIApiKeyToSARelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIServiceAccount)-[:OWNS]->(:OpenAIApiKey)
+class OpenAIApiKeyToSARel(CartographyRelSchema):
+    target_node_label: str = "OpenAIServiceAccount"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("owner_sa_id")},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "OWNS"
+    properties: OpenAIApiKeyToSARelProperties = OpenAIApiKeyToSARelProperties()
+
+
+@dataclass(frozen=True)
+class OpenAIApiKeySchema(CartographyNodeSchema):
+    label: str = "OpenAIApiKey"
+    properties: OpenAIApiKeyNodeProperties = OpenAIApiKeyNodeProperties()
+    sub_resource_relationship: OpenAIApiKeyToProjectRel = OpenAIApiKeyToProjectRel()
+    other_relationships: OtherRelationships = OtherRelationships(
+        [OpenAIApiKeyToUserRel(), OpenAIApiKeyToSARel()],
+    )

cartography/models/openai/organization.py

@@ -0,0 +1,17 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+
+
+@dataclass(frozen=True)
+class OpenAIOrganizationNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class OpenAIOrganizationSchema(CartographyNodeSchema):
+    label: str = "OpenAIOrganization"
+    properties: OpenAIOrganizationNodeProperties = OpenAIOrganizationNodeProperties()

cartography/models/openai/project.py

@@ -0,0 +1,70 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import OtherRelationships
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class OpenAIProjectNodeProperties(CartographyNodeProperties):
+    id: PropertyRef = PropertyRef("id")
+    object: PropertyRef = PropertyRef("object")
+    name: PropertyRef = PropertyRef("name")
+    created_at: PropertyRef = PropertyRef("created_at")
+    archived_at: PropertyRef = PropertyRef("archived_at")
+    status: PropertyRef = PropertyRef("status")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class OpenAIProjectToOrganizationRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIOrganization)-[:RESOURCE]->(:OpenAIProject)
+class OpenAIProjectToOrganizationRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIOrganization"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("ORG_ID", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: OpenAIProjectToOrganizationRelProperties = (
+        OpenAIProjectToOrganizationRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIProjectToUserRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIUser)-[:MEMBER_OF]->(:OpenAIProject)
+class OpenAIProjectToUserRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIUser"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("users", one_to_many=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "MEMBER_OF"
+    properties: OpenAIProjectToUserRelProperties = OpenAIProjectToUserRelProperties()
+
+
+@dataclass(frozen=True)
+class OpenAIProjectSchema(CartographyNodeSchema):
+    label: str = "OpenAIProject"
+    properties: OpenAIProjectNodeProperties = OpenAIProjectNodeProperties()
+    sub_resource_relationship: OpenAIProjectToOrganizationRel = (
+        OpenAIProjectToOrganizationRel()
+    )
+    other_relationships: OtherRelationships = OtherRelationships(
+        [OpenAIProjectToUserRel()],
+    )

cartography/models/openai/serviceaccount.py

@@ -0,0 +1,50 @@
+from dataclasses import dataclass
+
+from cartography.models.core.common import PropertyRef
+from cartography.models.core.nodes import CartographyNodeProperties
+from cartography.models.core.nodes import CartographyNodeSchema
+from cartography.models.core.relationships import CartographyRelProperties
+from cartography.models.core.relationships import CartographyRelSchema
+from cartography.models.core.relationships import LinkDirection
+from cartography.models.core.relationships import make_target_node_matcher
+from cartography.models.core.relationships import TargetNodeMatcher
+
+
+@dataclass(frozen=True)
+class OpenAIServiceAccountNodeProperties(CartographyNodeProperties):
+    object: PropertyRef = PropertyRef("object")
+    id: PropertyRef = PropertyRef("id")
+    name: PropertyRef = PropertyRef("name")
+    role: PropertyRef = PropertyRef("role")
+    created_at: PropertyRef = PropertyRef("created_at")
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+class OpenAIServiceAccountToProjectRelProperties(CartographyRelProperties):
+    lastupdated: PropertyRef = PropertyRef("lastupdated", set_in_kwargs=True)
+
+
+@dataclass(frozen=True)
+# (:OpenAIServiceAccount)<-[:RESOURCE]-(:OpenAIProject)
+class OpenAIServiceAccountToProjectRel(CartographyRelSchema):
+    target_node_label: str = "OpenAIProject"
+    target_node_matcher: TargetNodeMatcher = make_target_node_matcher(
+        {"id": PropertyRef("project_id", set_in_kwargs=True)},
+    )
+    direction: LinkDirection = LinkDirection.INWARD
+    rel_label: str = "RESOURCE"
+    properties: OpenAIServiceAccountToProjectRelProperties = (
+        OpenAIServiceAccountToProjectRelProperties()
+    )
+
+
+@dataclass(frozen=True)
+class OpenAIServiceAccountSchema(CartographyNodeSchema):
+    label: str = "OpenAIServiceAccount"
+    properties: OpenAIServiceAccountNodeProperties = (
+        OpenAIServiceAccountNodeProperties()
+    )
+    sub_resource_relationship: OpenAIServiceAccountToProjectRel = (
+        OpenAIServiceAccountToProjectRel()
+    )