wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

replace os with pathlib.Path in AICFilter for path handling consistency.

Browse Source
This commit is contained in:
Gökmen Görgen
2026-01-20 10:21:37 +01:00
parent 65b8e0e89c
commit 18b3ee743b
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/pipecat/audio/filters/aic_filter.py
View File

@@ -14,7 +14,7 @@ Classes:
AICFilter: For aic-sdk (uses 'aic_sdk' module) AICFilter: For aic-sdk (uses 'aic_sdk' module)
""" """
import os from pathlib import Path
from typing import List, Optional from typing import List, Optional
import numpy as np import numpy as np
@@ -49,7 +49,7 @@ class AICFilter(BaseAudioFilter):
license_key: str = "", license_key: str = "",
model_id: Optional[str] = None, model_id: Optional[str] = None,
model_path: Optional[str] = None, model_path: Optional[str] = None,
model_download_dir: Optional[str] = None, model_download_dir: Optional[Path] = None,
) -> None: ) -> None:
"""Initialize the AIC filter. """Initialize the AIC filter.
@@ -59,8 +59,8 @@ class AICFilter(BaseAudioFilter):
is not provided. See https://artifacts.ai-coustics.io/ for available models. is not provided. See https://artifacts.ai-coustics.io/ for available models.
model_path: Optional path to a local .aicmodel file. If provided, model_path: Optional path to a local .aicmodel file. If provided,
model_id is ignored and no download occurs. model_id is ignored and no download occurs.
model_download_dir: Directory for downloading models. Defaults to model_download_dir: Directory for downloading models as a Path object.
a cache directory in user's home folder. Defaults to a cache directory in user's home folder.
Raises: Raises:
ValueError: If neither model_id nor model_path is provided. ValueError: If neither model_id nor model_path is provided.
@@ -77,8 +77,8 @@ class AICFilter(BaseAudioFilter):
self._license_key = license_key self._license_key = license_key
self._model_id = model_id self._model_id = model_id
self._model_path = model_path self._model_path = model_path
self._model_download_dir = model_download_dir or os.path.expanduser( self._model_download_dir = model_download_dir or (
"~/.cache/pipecat/aic-models" Path.home() / ".cache" / "pipecat" / "aic-models"
) )
self._bypass = False self._bypass = False
@@ -169,8 +169,8 @@ class AICFilter(BaseAudioFilter):
self._model = Model.from_file(self._model_path) self._model = Model.from_file(self._model_path)
else: else:
logger.debug(f"Downloading AIC model: {self._model_id}") logger.debug(f"Downloading AIC model: {self._model_id}")
os.makedirs(self._model_download_dir, exist_ok=True) self._model_download_dir.mkdir(parents=True, exist_ok=True)
model_path = await Model.download_async(self._model_id, self._model_download_dir) model_path = await Model.download_async(self._model_id, str(self._model_download_dir))
logger.debug(f"Model downloaded to: {model_path}") logger.debug(f"Model downloaded to: {model_path}")
self._model = Model.from_file(model_path) self._model = Model.from_file(model_path)