Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion scripts/bidi/test_bidi_openai.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from strands_tools import calculator

from strands.experimental.bidi.agent.agent import BidiAgent
from strands.experimental.bidi.models.openai import BidiOpenAIRealtimeModel
from strands.experimental.bidi.models.openai_realtime import BidiOpenAIRealtimeModel


async def play(context):
Expand Down
10 changes: 2 additions & 8 deletions src/strands/experimental/bidi/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,8 @@
from .io.audio import BidiAudioIO

# Model interface (for custom implementations)
from .models.bidi_model import BidiModel

# Model providers - What users need to create models
from .models.gemini_live import BidiGeminiLiveModel
from .models.novasonic import BidiNovaSonicModel
from .models.openai import BidiOpenAIRealtimeModel
from .models.model import BidiModel
from .models.nova_sonic import BidiNovaSonicModel

# Built-in tools
from .tools import stop_conversation
Expand Down Expand Up @@ -53,9 +49,7 @@
# IO channels
"BidiAudioIO",
# Model providers
"BidiGeminiLiveModel",
"BidiNovaSonicModel",
"BidiOpenAIRealtimeModel",
# Built-in tools
"stop_conversation",
# Input Event types
Expand Down
4 changes: 2 additions & 2 deletions src/strands/experimental/bidi/agent/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,8 @@
from ...hooks.events import BidiAgentInitializedEvent
from ...tools import ToolProvider
from .._async import stop_all
from ..models.bidi_model import BidiModel
from ..models.novasonic import BidiNovaSonicModel
from ..models.model import BidiModel
from ..models.nova_sonic import BidiNovaSonicModel
from ..types.agent import BidiAgentInput
from ..types.events import (
BidiAudioInputEvent,
Expand Down
8 changes: 2 additions & 6 deletions src/strands/experimental/bidi/models/__init__.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,10 @@
"""Bidirectional model interfaces and implementations."""

from .bidi_model import BidiModel, BidiModelTimeoutError
from .gemini_live import BidiGeminiLiveModel
from .novasonic import BidiNovaSonicModel
from .openai import BidiOpenAIRealtimeModel
from .model import BidiModel, BidiModelTimeoutError
from .nova_sonic import BidiNovaSonicModel

__all__ = [
"BidiModel",
"BidiModelTimeoutError",
"BidiGeminiLiveModel",
"BidiNovaSonicModel",
"BidiOpenAIRealtimeModel",
]
4 changes: 2 additions & 2 deletions src/strands/experimental/bidi/models/gemini_live.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
from ....types.content import Messages
from ....types.tools import ToolResult, ToolSpec, ToolUse
from .._async import stop_all
from ..types.bidi_model import AudioConfig
from ..types.model import AudioConfig
from ..types.events import (
AudioChannel,
AudioSampleRate,
Expand All @@ -41,7 +41,7 @@
BidiUsageEvent,
ModalityUsage,
)
from .bidi_model import BidiModel, BidiModelTimeoutError
from .model import BidiModel, BidiModelTimeoutError

logger = logging.getLogger(__name__)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@
from ....types.content import Messages
from ....types.tools import ToolResult, ToolSpec, ToolUse
from .._async import stop_all
from ..types.bidi_model import AudioConfig
from ..types.model import AudioConfig
from ..types.events import (
AudioChannel,
AudioSampleRate,
Expand All @@ -53,7 +53,7 @@
BidiTranscriptStreamEvent,
BidiUsageEvent,
)
from .bidi_model import BidiModel, BidiModelTimeoutError
from .model import BidiModel, BidiModelTimeoutError

logger = logging.getLogger(__name__)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from ....types.content import Messages
from ....types.tools import ToolResult, ToolSpec, ToolUse
from .._async import stop_all
from ..types.bidi_model import AudioConfig
from ..types.model import AudioConfig
from ..types.events import (
AudioSampleRate,
BidiAudioInputEvent,
Expand All @@ -37,7 +37,7 @@
Role,
StopReason,
)
from .bidi_model import BidiModel, BidiModelTimeoutError
from .model import BidiModel, BidiModelTimeoutError

logger = logging.getLogger(__name__)

Expand Down
2 changes: 1 addition & 1 deletion src/strands/experimental/bidi/types/events.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@
from ....types.streaming import ContentBlockDelta

if TYPE_CHECKING:
from ..models.bidi_model import BidiModelTimeoutError
from ..models.model import BidiModelTimeoutError

AudioChannel = Literal[1, 2]
"""Number of audio channels.
Expand Down
2 changes: 1 addition & 1 deletion tests/strands/experimental/bidi/models/test_gemini_live.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
import pytest
from google.genai import types as genai_types

from strands.experimental.bidi.models.bidi_model import BidiModelTimeoutError
from strands.experimental.bidi.models.model import BidiModelTimeoutError
from strands.experimental.bidi.models.gemini_live import BidiGeminiLiveModel
from strands.experimental.bidi.types.events import (
BidiAudioInputEvent,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,10 @@
import pytest_asyncio
from aws_sdk_bedrock_runtime.models import ModelTimeoutException, ValidationException

from strands.experimental.bidi.models.novasonic import (
from strands.experimental.bidi.models.nova_sonic import (
BidiNovaSonicModel,
)
from strands.experimental.bidi.models.bidi_model import BidiModelTimeoutError
from strands.experimental.bidi.models.model import BidiModelTimeoutError
from strands.experimental.bidi.types.events import (
BidiAudioInputEvent,
BidiAudioStreamEvent,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@

import pytest

from strands.experimental.bidi.models.bidi_model import BidiModelTimeoutError
from strands.experimental.bidi.models.openai import BidiOpenAIRealtimeModel
from strands.experimental.bidi.models.model import BidiModelTimeoutError
from strands.experimental.bidi.models.openai_realtime import BidiOpenAIRealtimeModel
from strands.experimental.bidi.types.events import (
BidiAudioInputEvent,
BidiAudioStreamEvent,
Expand Down
Loading