chore: Add combined flag and adjust for Python (box/box-codegen#836)

Author: box-sdk-build

.codegen.json

@@ -1 +1 @@
-{ "engineHash": "0271743", "specHash": "1ed059a", "version": "0.1.0" }
+{ "engineHash": "cedd6e8", "specHash": "1ed059a", "version": "0.1.0" }

test/box_sdk_gen/__init__.py

@@ -0,0 +1,92 @@
+from box_sdk_gen.internal.utils import to_string
+
+from typing import Optional
+
+from box_sdk_gen.schemas.ai_single_agent_response_full import AiSingleAgentResponseFull
+
+from box_sdk_gen.schemas.ai_studio_agent_ask import AiStudioAgentAsk
+
+from box_sdk_gen.schemas.ai_multiple_agent_response import AiMultipleAgentResponse
+
+from box_sdk_gen.schemas.file_full import FileFull
+
+from box_sdk_gen.schemas.ai_response_full import AiResponseFull
+
+from box_sdk_gen.managers.ai import CreateAiAskMode
+
+from box_sdk_gen.schemas.ai_item_ask import AiItemAsk
+
+from box_sdk_gen.schemas.ai_item_ask import AiItemAskTypeField
+
+from box_sdk_gen.box.developer_token_auth import BoxDeveloperTokenAuth
+
+from box_sdk_gen.client import BoxClient
+
+from test.box_sdk_gen.test.commons import get_default_client
+
+from test.box_sdk_gen.test.commons import upload_new_file
+
+from box_sdk_gen.internal.utils import get_uuid
+
+from box_sdk_gen.schemas.ai_agent_reference import AiAgentReference
+
+client: BoxClient = get_default_client()
+
+
+def testAiStudioCRUD():
+    agent_name: str = get_uuid()
+    created_agent: AiSingleAgentResponseFull = client.ai_studio.create_ai_agent(
+        agent_name,
+        'enabled',
+        ask=AiStudioAgentAsk(access_state='enabled', description='desc1'),
+    )
+    assert created_agent.name == agent_name
+    agents: AiMultipleAgentResponse = client.ai_studio.get_ai_agents()
+    num_agents: int = len(agents.entries)
+    assert to_string(agents.entries[0].type) == 'ai_agent'
+    retrieved_agent: AiSingleAgentResponseFull = client.ai_studio.get_ai_agent_by_id(
+        created_agent.id, fields=['ask']
+    )
+    assert retrieved_agent.name == agent_name
+    assert to_string(retrieved_agent.access_state) == 'enabled'
+    assert to_string(retrieved_agent.ask.access_state) == 'enabled'
+    assert retrieved_agent.ask.description == 'desc1'
+    updated_agent: AiSingleAgentResponseFull = client.ai_studio.update_ai_agent_by_id(
+        created_agent.id,
+        agent_name,
+        'enabled',
+        ask=AiStudioAgentAsk(access_state='disabled', description='desc2'),
+    )
+    assert to_string(updated_agent.access_state) == 'enabled'
+    assert to_string(updated_agent.ask.access_state) == 'disabled'
+    assert updated_agent.ask.description == 'desc2'
+    client.ai_studio.delete_ai_agent_by_id(created_agent.id)
+    agents_after_delete: AiMultipleAgentResponse = client.ai_studio.get_ai_agents()
+    assert len(agents_after_delete.entries) == num_agents - 1
+
+
+def testUseAIAgentReferenceInAIAsk():
+    agent_name: str = get_uuid()
+    created_agent: AiSingleAgentResponseFull = client.ai_studio.create_ai_agent(
+        agent_name,
+        'enabled',
+        ask=AiStudioAgentAsk(access_state='enabled', description='desc1'),
+    )
+    file_to_ask: FileFull = upload_new_file()
+    response: Optional[AiResponseFull] = client.ai.create_ai_ask(
+        CreateAiAskMode.SINGLE_ITEM_QA,
+        'which direction sun rises',
+        [
+            AiItemAsk(
+                id=file_to_ask.id,
+                type=AiItemAskTypeField.FILE,
+                content='Sun rises in the East',
+            )
+        ],
+        ai_agent=AiAgentReference(id=created_agent.id),
+    )
+    assert 'East' in response.answer
+    assert response.completion_reason == 'done'
+    assert len(response.ai_agent_info.models) > 0
+    client.files.delete_file_by_id(file_to_ask.id)
+    client.ai_studio.delete_ai_agent_by_id(created_agent.id)

test/box_sdk_gen/test/__init__.py

@@ -0,0 +1,55 @@
+from box_sdk_gen.internal.utils import to_string
+
+from box_sdk_gen.client import BoxClient
+
+from box_sdk_gen.schemas.app_item_associations import AppItemAssociations
+
+from box_sdk_gen.schemas.app_item_association import AppItemAssociation
+
+from box_sdk_gen.schemas.file_full import FileFull
+
+from box_sdk_gen.schemas.folder_full import FolderFull
+
+from box_sdk_gen.internal.utils import get_uuid
+
+from box_sdk_gen.internal.utils import get_env_var
+
+from test.box_sdk_gen.test.commons import get_default_client_with_user_subject
+
+
+def testListFileAppItemAssocations():
+    client: BoxClient = get_default_client_with_user_subject(get_env_var('USER_ID'))
+    file_id: str = get_env_var('APP_ITEM_ASSOCIATION_FILE_ID')
+    file_app_item_associations: AppItemAssociations = (
+        client.app_item_associations.get_file_app_item_associations(file_id)
+    )
+    assert len(file_app_item_associations.entries) == 1
+    association: AppItemAssociation = file_app_item_associations.entries[0]
+    assert not association.id == ''
+    assert to_string(association.app_item.application_type) == 'hubs'
+    assert to_string(association.app_item.type) == 'app_item'
+    assert to_string(association.item.type) == 'file'
+    assert association.item.id == file_id
+    file: FileFull = client.files.get_file_by_id(
+        file_id, fields=['is_associated_with_app_item']
+    )
+    assert file.is_associated_with_app_item == True
+
+
+def testListFolderAppItemAssocations():
+    client: BoxClient = get_default_client_with_user_subject(get_env_var('USER_ID'))
+    folder_id: str = get_env_var('APP_ITEM_ASSOCIATION_FOLDER_ID')
+    folder_app_item_associations: AppItemAssociations = (
+        client.app_item_associations.get_folder_app_item_associations(folder_id)
+    )
+    assert len(folder_app_item_associations.entries) == 1
+    association: AppItemAssociation = folder_app_item_associations.entries[0]
+    assert not association.id == ''
+    assert to_string(association.app_item.application_type) == 'hubs'
+    assert to_string(association.app_item.type) == 'app_item'
+    assert to_string(association.item.type) == 'folder'
+    assert association.item.id == folder_id
+    folder: FolderFull = client.folders.get_folder_by_id(
+        folder_id, fields=['is_associated_with_app_item']
+    )
+    assert folder.is_associated_with_app_item == True

test/box_sdk_gen/test/ai.py

@@ -0,0 +1,31 @@
+from box_sdk_gen.internal.utils import to_string
+
+import pytest
+
+from box_sdk_gen.client import BoxClient
+
+from box_sdk_gen.schemas.v2025_r0.archive_v2025_r0 import ArchiveV2025R0
+
+from box_sdk_gen.schemas.v2025_r0.archives_v2025_r0 import ArchivesV2025R0
+
+from box_sdk_gen.internal.utils import get_uuid
+
+from box_sdk_gen.internal.utils import get_env_var
+
+from test.box_sdk_gen.test.commons import get_default_client_with_user_subject
+
+user_id: str = get_env_var('USER_ID')
+
+client: BoxClient = get_default_client_with_user_subject(user_id)
+
+
+def testArchivesCreateListDelete():
+    archive_name: str = get_uuid()
+    archive: ArchiveV2025R0 = client.archives.create_archive_v2025_r0(archive_name)
+    assert to_string(archive.type) == 'archive'
+    assert archive.name == archive_name
+    archives: ArchivesV2025R0 = client.archives.get_archives_v2025_r0(limit=100)
+    assert len(archives.entries) > 0
+    client.archives.delete_archive_by_id_v2025_r0(archive.id)
+    with pytest.raises(Exception):
+        client.archives.delete_archive_by_id_v2025_r0(archive.id)