fix: export audio intelligence types

GitOrigin-RevId: 2b503fcca2631976f86807808f56befffb8a220d

Author: dmccrystal

README.md

@@ -222,7 +222,7 @@ questions = [
 
 results = transcript.lemur.question(questions)
 
-for result in results:
+for result in result:
     print(f"Question: {result.question}")
     print(f"Answer: {result.answer}")
 ```

assemblyai/__init__.py

@@ -3,6 +3,18 @@
 from .transcriber import Transcriber, Transcript, TranscriptGroup
 from .types import (
     AssemblyAIError,
+    AutohighlightResponse,
+    AutohighlightResult,
+    Chapter,
+    ContentSafetyLabel,
+    ContentSafetyLabelResult,
+    ContentSafetyResponse,
+    ContentSafetySeverityScore,
+    Entity,
+    EntityType,
+    IABLabelResult,
+    IABResponse,
+    IABResult,
     LanguageCode,
     LemurError,
     LemurModel,
@@ -13,7 +25,10 @@
     PIISubstitutionPolicy,
     RawTranscriptionConfig,
     Sentence,
+    Sentiment,
+    SentimentType,
     Settings,
+    StatusResult,
     SummarizationModel,
     SummarizationType,
     Timestamp,
@@ -34,15 +49,32 @@
 __all__ = [
     # types
     "AssemblyAIError",
+    "AutohighlightResponse",
+    "AutohighlightResult",
+    "Chapter",
     "Client",
+    "ContentSafetyLabel",
+    "ContentSafetyLabelResult",
+    "ContentSafetyResponse",
+    "ContentSafetySeverityScore",
+    "Entity",
+    "EntityType",
+    "IABLabelResult",
+    "IABResponse",
+    "IABResult",
     "LanguageCode",
     "Lemur",
     "LemurError",
     "LemurModel",
     "LemurQuestion",
     "LemurQuestionResult",
+    "PIIRedactionPolicy",
+    "PIISubstitutionPolicy",
     "Sentence",
+    "Sentiment",
+    "SentimentType",
     "Settings",
+    "StatusResult",
     "SummarizationModel",
     "SummarizationType",
     "Timestamp",

assemblyai/transcriber.py

@@ -77,7 +77,7 @@ def wait_for_completion(self) -> Self:
             ):
                 break
 
-            time.sleep(self._client.settings.polling_interval)
+            time.sleep(3)
 
         return self
 

assemblyai/types.py

@@ -37,9 +37,6 @@ class Settings(BaseSettings):
     base_url: str = "https://api.assemblyai.com/v2"
     "The base URL for the AssemblyAI API"
 
-    polling_interval: float = 1.0
-    "The default polling interval for long-running requests (e.g. polling the `Transcript`'s status)"
-
     class Config:
         env_prefix = "assemblyai_"
 

setup.py

@@ -7,7 +7,7 @@
 
 setup(
     name="assemblyai",
-    version="0.8.1",
+    version="0.7.1",
     description="AssemblyAI Python SDK",
     author="AssemblyAI",
     author_email="[email protected]",

tests/unit/BUILD.bazel

@@ -0,0 +1,50 @@
+load("@rules_python//python:defs.bzl", "py_library", "py_test")
+load("@aspect_rules_py//py:defs.bzl", "py_pytest_main")
+
+py_pytest_main(name = "__test__")
+
+py_library(
+    name = "unit",
+    testonly = True,
+    srcs = [
+        "__init__.py",  # keep
+        "conftest.py",
+        "factories.py",  # keep
+    ],
+    imports = ["../../../../../.."],
+    visibility = ["//:__subpackages__"],
+    deps = [
+        "@assemblyai_deps_factory_boy//:pkg",  # keep
+        "@assemblyai_deps_pytest//:pkg",
+    ],
+)
+
+py_test(
+    name = "unit_test",
+    srcs = [
+        "test_auto_chapters.py",
+        "test_client.py",
+        "test_config.py",
+        "test_content_safety.py",
+        "test_domains.py",
+        "test_entity_detection.py",
+        "test_lemur.py",
+        "test_sentiment_analysis.py",
+        "test_summarization.py",
+        "test_transcriber.py",
+        "test_transcript.py",
+        ":__test__",
+    ],
+    imports = ["../../../../../.."],
+    main = ":__test__.py",
+    deps = [
+        ":__test__",
+        ":unit",
+        "//assemblyai/developer_tools/python/sdk:assemblyai_sdk",
+        "@assemblyai_deps_factory_boy//:pkg",
+        "@assemblyai_deps_faker//:pkg",
+        "@assemblyai_deps_httpx//:pkg",
+        "@assemblyai_deps_pytest//:pkg",
+        "@assemblyai_deps_pytest_httpx//:pkg",
+    ],
+)