amigo_sdk 0.97.0__tar.gz → 0.98.0__tar.gz

{amigo_sdk-0.97.0 → amigo_sdk-0.98.0}/.github/workflows/test.yml

@@ -83,4 +83,4 @@ jobs:
           AMIGO_ORGANIZATION_ID: ${{ secrets.AMIGO_ORGANIZATION_ID }}
           AMIGO_USER_ID: ${{ secrets.AMIGO_USER_ID }}
           AMIGO_BASE_URL: ${{ secrets.AMIGO_BASE_URL }}
-        run: pytest -m integration -v
+        run: pytest -m integration -v -s

{amigo_sdk-0.97.0 → amigo_sdk-0.98.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amigo_sdk
-Version: 0.97.0
+Version: 0.98.0
 Summary: Amigo AI Python SDK
 Author: Amigo AI
 License-File: LICENSE

{amigo_sdk-0.97.0 → amigo_sdk-0.98.0}/src/amigo_sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.97.0"
+__version__ = "0.98.0"
 from .sdk_client import AmigoClient, AsyncAmigoClient
 
 __all__ = ["__version__", "AmigoClient", "AsyncAmigoClient"]

{amigo_sdk-0.97.0 → amigo_sdk-0.98.0}/src/amigo_sdk/generated/model.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  <stdin>
-#   timestamp: 2026-01-08T19:54:56+00:00
+#   timestamp: 2026-01-08T23:39:28+00:00
 
 from __future__ import annotations
 

{amigo_sdk-0.97.0 → amigo_sdk-0.98.0}/src/amigo_sdk/resources/user.py

@@ -2,6 +2,7 @@ from amigo_sdk.generated.model import (
     GetUsersParametersQuery,
     UserCreateInvitedUserRequest,
     UserCreateInvitedUserResponse,
+    UserGetUserModelResponse,
     UserGetUsersResponse,
     UserUpdateUserInfoRequest,
 )
@@ -54,6 +55,14 @@ class AsyncUserResource:
             json=body.model_dump(mode="json", exclude_none=True),
         )
 
+    async def get_user_model(self, user_id: str) -> UserGetUserModelResponse:
+        """Get the latest user model for a user."""
+        response = await self._http.request(
+            "GET",
+            f"/v1/{self._organization_id}/user/{user_id}/user_model",
+        )
+        return UserGetUserModelResponse.model_validate_json(response.text)
+
 
 class UserResource:
     """User resource (synchronous)."""
@@ -93,3 +102,10 @@ class UserResource:
             f"/v1/{self._organization_id}/user/{user_id}",
             json=body.model_dump(mode="json", exclude_none=True),
         )
+
+    def get_user_model(self, user_id: str) -> UserGetUserModelResponse:
+        response = self._http.request(
+            "GET",
+            f"/v1/{self._organization_id}/user/{user_id}/user_model",
+        )
+        return UserGetUserModelResponse.model_validate_json(response.text)

{amigo_sdk-0.97.0 → amigo_sdk-0.98.0}/tests/integration/test_user_integration.py

@@ -60,6 +60,17 @@ class TestUserIntegration:
             assert by_email is not None
             assert any(u.email == type(self).created_user_email for u in by_email.users)
 
+    # we need to get the user model for the authenticated user because the permission grant for the user model
+    # does not allow access to non-admin user user models
+    async def test_get_user_model(self):
+        user_id = os.environ.get("AMIGO_USER_ID")
+        assert user_id is not None, "AMIGO_USER_ID environment variable must be set"
+        async with AsyncAmigoClient() as client:
+            result = await client.users.get_user_model(user_id)
+            assert result is not None
+            assert isinstance(result.user_models, list)
+            assert isinstance(result.additional_context, list)
+
     async def test_delete_user(self):
         assert type(self).created_user_id is not None
         async with AsyncAmigoClient() as client:
@@ -94,6 +105,10 @@ class TestUserIntegration:
                 with pytest.raises(errors.AuthenticationError):
                     await bad_client.users.get_users()
 
+            # Get user model for non-existent user
+            with pytest.raises(errors.NotFoundError):
+                await client.users.get_user_model("non-existent-id")
+
             # Delete non-existent user
             with pytest.raises(errors.NotFoundError):
                 await client.users.delete_user("non-existent-id")
@@ -148,6 +163,15 @@ class TestUserIntegrationSync:
             assert by_email is not None
             assert any(u.email == type(self).created_user_email for u in by_email.users)
 
+    def test_get_user_model(self):
+        user_id = os.environ.get("AMIGO_USER_ID")
+        assert user_id is not None, "AMIGO_USER_ID environment variable not set"
+        with AmigoClient() as client:
+            result = client.users.get_user_model(user_id)
+            assert result is not None
+            assert isinstance(result.user_models, list)
+            assert isinstance(result.additional_context, list)
+
     def test_delete_user(self):
         assert type(self).created_user_id is not None
         with AmigoClient() as client:
@@ -177,5 +201,8 @@ class TestUserIntegrationSync:
                 with pytest.raises(errors.AuthenticationError):
                     bad_client.users.get_users()
 
+            with pytest.raises(errors.NotFoundError):
+                client.users.get_user_model("non-existent-id")
+
             with pytest.raises(errors.NotFoundError):
                 client.users.delete_user("non-existent-id")

{amigo_sdk-0.97.0 → amigo_sdk-0.98.0}/tests/resources/test_user.py

@@ -4,9 +4,12 @@ from amigo_sdk.config import AmigoConfig
 from amigo_sdk.errors import NotFoundError, ValidationError
 from amigo_sdk.generated.model import (
     GetUsersParametersQuery,
+    MongoCollectionsUserUserUserModelUserDimension,
     UserCreateInvitedUserRequest,
     UserCreateInvitedUserResponse,
+    UserGetUserModelResponse,
     UserGetUsersResponse,
+    UserModel,
     UserUpdateUserInfoRequest,
 )
 from amigo_sdk.http_client import AmigoAsyncHttpClient, AmigoHttpClient
@@ -122,6 +125,34 @@ class TestUserResource:
             with pytest.raises(ValidationError):
                 await user_resource.update_user("u-1", body)
 
+    @pytest.mark.asyncio
+    async def test_get_user_model_returns_data(self, user_resource):
+        mock_response = UserGetUserModelResponse(
+            user_models=[
+                UserModel(
+                    content="model-content",
+                    insight_ids=["insight-1"],
+                    dimensions=[
+                        MongoCollectionsUserUserUserModelUserDimension(
+                            description="detail", tags=["tag"]
+                        )
+                    ],
+                )
+            ],
+            additional_context=["context"],
+        )
+
+        async with mock_http_request(mock_response):
+            result = await user_resource.get_user_model("u-1")
+            assert isinstance(result, UserGetUserModelResponse)
+            assert result.user_models[0].content == "model-content"
+
+    @pytest.mark.asyncio
+    async def test_get_user_model_not_found_raises(self, user_resource):
+        async with mock_http_request("{}", status_code=404):
+            with pytest.raises(NotFoundError):
+                await user_resource.get_user_model("missing")
+
 
 @pytest.mark.unit
 class TestUserResourceSync:
@@ -211,3 +242,30 @@ class TestUserResourceSync:
         with mock_http_request_sync({"detail": "bad"}, status_code=422):
             with pytest.raises(ValidationError):
                 res.update_user("u-1", body)
+
+    def test_get_user_model_returns_data_sync(self, mock_config):
+        res = self._resource(mock_config)
+        mock_response = UserGetUserModelResponse(
+            user_models=[
+                UserModel(
+                    content="model-content",
+                    insight_ids=["insight-1"],
+                    dimensions=[
+                        MongoCollectionsUserUserUserModelUserDimension(
+                            description="detail", tags=["tag"]
+                        )
+                    ],
+                )
+            ],
+            additional_context=["context"],
+        )
+        with mock_http_request_sync(mock_response):
+            result = res.get_user_model("u-1")
+            assert isinstance(result, UserGetUserModelResponse)
+            assert result.user_models[0].content == "model-content"
+
+    def test_get_user_model_not_found_raises_sync(self, mock_config):
+        res = self._resource(mock_config)
+        with mock_http_request_sync("{}", status_code=404):
+            with pytest.raises(NotFoundError):
+                res.get_user_model("missing")