amigo_sdk 0.96.0__py3-none-any.whl → 0.98.0__py3-none-any.whl

amigo_sdk/resources/conversation.py

@@ -166,7 +166,7 @@ class AsyncConversationResource:
     ) -> ConversationRecommendResponsesForInteractionResponse:
         """Recommend responses for an interaction."""
         response = await self._http.request(
-            "GET",
+            "POST",
             f"/v1/{self._organization_id}/conversation/{conversation_id}/interaction/{interaction_id}/recommend_responses",
         )
         return ConversationRecommendResponsesForInteractionResponse.model_validate_json(
@@ -321,7 +321,7 @@ class ConversationResource:
         self, conversation_id: str, interaction_id: str
     ) -> ConversationRecommendResponsesForInteractionResponse:
         response = self._http.request(
-            "GET",
+            "POST",
             f"/v1/{self._organization_id}/conversation/{conversation_id}/interaction/{interaction_id}/recommend_responses",
         )
         return ConversationRecommendResponsesForInteractionResponse.model_validate_json(

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.96.0.dist-info → amigo_sdk-0.98.0.dist-info}/METADATA

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

amigo_sdk-0.98.0.dist-info/RECORD

@@ -0,0 +1,22 @@
+scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+scripts/aliases.json,sha256=c1_bkZyvNDQ8iMrKtUuZQqdP87GdLnRPrb3wjX533eQ,322
+scripts/check.py,sha256=ikFavw1IjX51FNjO-kIzXjYdep3on0dKZHYvPElPFtc,2872
+scripts/gen_models.py,sha256=c4J_gHYQIBL3CcKg5z8oUusLvPePytfvLdZ7STHBcNk,3553
+amigo_sdk/__init__.py,sha256=SKegkSGR5Jguo83Iqy3NO1YcX0ctgzamLo5mVpq0ens,139
+amigo_sdk/_retry_utils.py,sha256=VbA_Uwv268J80d11R9XfnCjZiV-NwE1aXRsSlRjjRio,2108
+amigo_sdk/auth.py,sha256=WaM9PcEcnaC6CzNsgRKueHkdSAxNbRylzpR_3Q6guQ0,1765
+amigo_sdk/config.py,sha256=0eZIo-hcJ8ODftKAr-mwB-FGJxGO5PT5T4dRpyWPqAg,1491
+amigo_sdk/errors.py,sha256=Yn12AxIQ6zE9jnGLWO0MUVbBsmw77Kzh82AijyreOkA,4818
+amigo_sdk/http_client.py,sha256=tfMXHqMTmlz2ja7aJhOVgbwAiJNlKCzPcK4UGNqXHO8,13488
+amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
+amigo_sdk/sdk_client.py,sha256=5nj8hzjHSFRs9JMNabdYIhQAcFE2G1jI_r7SHVGr9No,6106
+amigo_sdk/generated/model.py,sha256=TF46V7sugZT_tQcTCYghnb1SjpbCYj74tY__yPQL_Cw,446306
+amigo_sdk/resources/conversation.py,sha256=X47PbPA6NqdkqXgxAPOjlO1E0W7t695tn1tPWSikMOk,14807
+amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
+amigo_sdk/resources/service.py,sha256=t1iA3nS9co-wuR-x5jBhAXXTWfMeGsLwfQcLycwVrCA,1536
+amigo_sdk/resources/user.py,sha256=P14lN-3fmIvELUjn_vpSaYpNsK-xd8zNmUV_KfcZPAU,4138
+amigo_sdk-0.98.0.dist-info/METADATA,sha256=VuYRbvGyFOV6VDvOLKheyRe7m16wXwrY4FLRBBtmL3A,8499
+amigo_sdk-0.98.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+amigo_sdk-0.98.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.98.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.98.0.dist-info/RECORD,,

scripts/aliases.json

@@ -0,0 +1,5 @@
+{
+  "#/components/schemas/amigo_lib__pydantic__base_model__StrippedNonemptyString__1": "NonEmptyString",
+  "#/components/schemas/amigo_lib__pydantic__base_model__StrippedNonemptyString__2": "NonEmptyStringMax256",
+  "#/components/schemas/amigo_lib__pydantic__base_model__StrippedNonemptyString__3": "NonEmptyStringMax3"
+}

scripts/check.py

@@ -0,0 +1,102 @@
+#!/usr/bin/env python3
+"""
+Development check script - runs formatting, linting, and tests.
+
+Usage:
+    python scripts/check.py           # Run all checks
+    python scripts/check.py --fix     # Run all checks and auto-fix issues
+    python scripts/check.py --fast    # Skip tests (format + lint only)
+"""
+
+import argparse
+import subprocess
+import sys
+from pathlib import Path
+
+
+def run_command(cmd: list[str], description: str) -> bool:
+    """Run a command and return True if successful."""
+    print(f"🔍 {description}...")
+
+    try:
+        result = subprocess.run(
+            cmd, capture_output=True, text=True, cwd=Path(__file__).parent.parent
+        )
+
+        if result.returncode == 0:
+            print(f"✅ {description} - PASSED")
+            if result.stdout:
+                print(f"   {result.stdout.strip()}")
+            return True
+        else:
+            print(f"❌ {description} - FAILED")
+            if result.stdout:
+                print(f"   stdout: {result.stdout.strip()}")
+            if result.stderr:
+                print(f"   stderr: {result.stderr.strip()}")
+            return False
+
+    except Exception as e:
+        print(f"❌ {description} - ERROR: {e}")
+        return False
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Run development checks")
+    parser.add_argument(
+        "--fix", action="store_true", help="Auto-fix linting and formatting issues"
+    )
+    parser.add_argument(
+        "--fast", action="store_true", help="Skip tests (format + lint only)"
+    )
+    args = parser.parse_args()
+
+    print("🚀 Running development checks...\n")
+
+    all_passed = True
+
+    # 1. Format check or fix
+    if args.fix:
+        success = run_command(["ruff", "format", "."], "Code formatting (auto-fix)")
+    else:
+        success = run_command(
+            ["ruff", "format", "--check", "."], "Code formatting check"
+        )
+    all_passed &= success
+
+    print()
+
+    # 2. Lint check or fix
+    if args.fix:
+        success = run_command(["ruff", "check", ".", "--fix"], "Linting (auto-fix)")
+    else:
+        success = run_command(["ruff", "check", "."], "Linting check")
+    all_passed &= success
+
+    print()
+
+    # 3. Tests (unless --fast)
+    if not args.fast:
+        success = run_command(
+            ["pytest", "--cov=src", "--cov-report=term", "-v"],
+            "Test suite with coverage",
+        )
+        all_passed &= success
+        print()
+
+    # Summary
+    print("=" * 50)
+    if all_passed:
+        print("🎉 All checks PASSED! Ready to commit.")
+        if args.fast:
+            print("💡 Run without --fast to include tests.")
+        sys.exit(0)
+    else:
+        print("💥 Some checks FAILED!")
+        if not args.fix:
+            print("💡 Try running with --fix to auto-fix formatting/linting issues.")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

scripts/gen_models.py

@@ -0,0 +1,122 @@
+import json
+from pathlib import Path
+
+import httpx
+from datamodel_code_generator import (
+    DataModelType,
+    InputFileType,
+    OpenAPIScope,
+    generate,
+)
+
+# Prefixes to strip from schema names (API-specific internal paths)
+STRIP_PREFIXES = [
+    "src__app__endpoints__",
+    "src__app__amigo__",
+    "amigo_lib__",
+]
+
+
+def strip_prefixes_from_schema(spec: dict) -> dict:
+    """
+    Pre-process the OpenAPI spec to strip internal prefixes from schema names.
+    This allows the code generator's built-in name sanitization to work correctly.
+    """
+    schemas = spec.get("components", {}).get("schemas", {})
+    if not schemas:
+        return spec
+
+    # Build a mapping of old names to new names
+    rename_map: dict[str, str] = {}
+    for name in schemas:
+        new_name = name
+        for prefix in STRIP_PREFIXES:
+            if new_name.startswith(prefix):
+                new_name = new_name[len(prefix) :]
+                break
+        if new_name != name:
+            rename_map[name] = new_name
+
+    if not rename_map:
+        return spec
+
+    # Rename schemas
+    new_schemas = {}
+    for name, schema in schemas.items():
+        new_name = rename_map.get(name, name)
+        new_schemas[new_name] = schema
+    spec["components"]["schemas"] = new_schemas
+
+    # Update all $ref pointers throughout the spec
+    def update_refs(obj):
+        if isinstance(obj, dict):
+            if "$ref" in obj:
+                ref = obj["$ref"]
+                for old, new in rename_map.items():
+                    old_ref = f"#/components/schemas/{old}"
+                    new_ref = f"#/components/schemas/{new}"
+                    if ref == old_ref:
+                        obj["$ref"] = new_ref
+                        break
+            for value in obj.values():
+                update_refs(value)
+        elif isinstance(obj, list):
+            for item in obj:
+                update_refs(item)
+
+    update_refs(spec)
+    return spec
+
+
+def main() -> None:
+    schema_url = "https://api.amigo.ai/v1/openapi.json"
+    root = Path(__file__).parent.parent
+    out_dir = root / "src" / "amigo_sdk" / "generated"
+    output_file = out_dir / "model.py"
+    aliases_path = root / "scripts" / "aliases.json"
+
+    # Create the generated directory if it doesn't exist
+    out_dir.mkdir(parents=True, exist_ok=True)
+
+    # Remove existing model.py if it exists
+    if output_file.exists():
+        output_file.unlink()
+
+    # Fetch the OpenAPI schema from the remote URL
+    print(f"Fetching OpenAPI schema from {schema_url}...")
+    response = httpx.get(schema_url)
+    response.raise_for_status()
+    spec = response.json()
+
+    # Pre-process: strip internal prefixes from schema names
+    spec = strip_prefixes_from_schema(spec)
+
+    # Load aliases as a mapping (Python API expects a dict)
+    aliases: dict[str, str] = {}
+    if aliases_path.exists():
+        aliases = json.loads(aliases_path.read_text())
+
+    generate(
+        json.dumps(spec),
+        input_file_type=InputFileType.OpenAPI,
+        output=output_file,
+        output_model_type=DataModelType.PydanticV2BaseModel,
+        openapi_scopes=[
+            OpenAPIScope.Schemas,
+            OpenAPIScope.Parameters,
+            OpenAPIScope.Paths,
+            OpenAPIScope.Tags,
+        ],
+        snake_case_field=True,
+        field_constraints=True,
+        use_operation_id_as_name=True,
+        reuse_model=True,
+        aliases=aliases,
+        collapse_root_models=True,
+    )
+
+    print(f"✅ Models regenerated → {output_file}")
+
+
+if __name__ == "__main__":
+    main()

amigo_sdk-0.96.0.dist-info/RECORD

@@ -1,18 +0,0 @@
-amigo_sdk/__init__.py,sha256=sF3mjc_jLenRVw_Lr-cNIbXd2JWMQ1OMx_-MV72EB7A,139
-amigo_sdk/_retry_utils.py,sha256=VbA_Uwv268J80d11R9XfnCjZiV-NwE1aXRsSlRjjRio,2108
-amigo_sdk/auth.py,sha256=WaM9PcEcnaC6CzNsgRKueHkdSAxNbRylzpR_3Q6guQ0,1765
-amigo_sdk/config.py,sha256=0eZIo-hcJ8ODftKAr-mwB-FGJxGO5PT5T4dRpyWPqAg,1491
-amigo_sdk/errors.py,sha256=Yn12AxIQ6zE9jnGLWO0MUVbBsmw77Kzh82AijyreOkA,4818
-amigo_sdk/http_client.py,sha256=tfMXHqMTmlz2ja7aJhOVgbwAiJNlKCzPcK4UGNqXHO8,13488
-amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
-amigo_sdk/sdk_client.py,sha256=5nj8hzjHSFRs9JMNabdYIhQAcFE2G1jI_r7SHVGr9No,6106
-amigo_sdk/generated/model.py,sha256=2BP-cnNJ7NxNptpVcZt0iyIqT3NEwV17UFQZONKgGUs,424265
-amigo_sdk/resources/conversation.py,sha256=u4kOx5LPbhX92L41UnmPmaF9TkQA5bGev8N00dMoP9I,14805
-amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
-amigo_sdk/resources/service.py,sha256=t1iA3nS9co-wuR-x5jBhAXXTWfMeGsLwfQcLycwVrCA,1536
-amigo_sdk/resources/user.py,sha256=zikijiuGXgmkBVtrldor6XQGQK6jWXonDvNFuQs6XK4,3472
-amigo_sdk-0.96.0.dist-info/METADATA,sha256=cgjyoSbW_5RTUWDBWCjqt6nlB1lTm8Xg2H64fulhn2g,8499
-amigo_sdk-0.96.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-amigo_sdk-0.96.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.96.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.96.0.dist-info/RECORD,,