refactor: Support different types of aliyun stt model

[shaohuzhang1]

refactor: Support different types of aliyun stt model

[f2c-ci-robot]

Adding the "do-not-merge/release-note-label-needed" label because no release-note block was detected, please follow our release note process to remove it.

Details

Instructions for interacting with me using PR comments are available here. If you have questions or suggestions related to my behavior, please file an issue against the kubernetes-sigs/prow repository.

[f2c-ci-robot]

[APPROVALNOTIFIER] This PR is NOT APPROVED

This pull-request has been approved by:

The full list of commands accepted by this bot can be found here.

Details

Needs approval from an approver in each of these files:

• OWNERS

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

[shaohuzhang1]

The provided code has several issues:

1. The file appears to be incomplete; there should be closing braces (}) after most but not all classes and functions.

2. Some lines with @@ -0,0 +1,68 are placeholder additions for the changes being checked. Ensure that these comments reflect actual added content, especially within classes and methods like speech_to_text, check_auth, etc.

3. Method definitions in Python require a colon at the end of the header (e.g., def check_auth():). You have one colon where it's missing (@staticmethod def is_cache_model()).

4. The imports and class definition are quite extensive without actually implementing any functionality. Consider removing unused import statements and reducing line length.

5. There are repeated instances of importing MaxKBBaseModel although they seem superfluous since the same model base implementation already inherits this class.

Here’s an optimized version while addressing some of those points:

# coding=utf-8

"""
    @project: MaxKB
    @Author：niu
    @file：default_stt.py
    @date：2025/12/5 15:40
    @desc:
"""

import os
from typing import Dict

from models_provider.base_model_provider import MaxKBBaseModel
from models_provider.impl.base_stt import BaseSpeechToText


class AliyunBaiLianDefaultSpeechToText(MaxKBBaseModel, BaseSpeechToText):
    def check_auth(self):
        # Implement authentication logic here
        pass

    def speech_to_text(self, audio_file):
        raise NotImplementedError("This method should be implemented.")

    @staticmethod
    def is_cache_model() -> bool:
        """Check if the model supports caching."""
        return False

    @staticmethod
    def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs) -> 'AliyunBaiLianDefaultSpeechToText':
        from models_provider.impl.aliyun_bai_lian_model_provider.model.stt import (
            AliyunBaiLianOmiSpeechToText,
            AliyunBaiLianSpeechToText,
            AliyunBaiLianAsrSpeechToText,
        )

        stt_type = model_credential.get('type')
        if stt_type == 'qwen':
            return AliyunBaiLianAsrSpeechToText(
                model=model_name,
                api_key=model_credential['api_key'],
                api_url=model_credential['api_url'],
                params=model_kwargs,
                **model_kwargs,
            )
        elif stt_type == 'omni':
            return AliyunBaiLianOmiSpeechToText(
                model=model_name,
                api_key=model_credential['api_key'],
                api_url=model_credential['api_url'],
                params=model_kwargs,
                **model_kwargs,
            )
        else:
            return AliyunBaiLianSpeechToText(
                model=model_name,
                api_key=model_credential['api_key'],
                params=model_kwargs,
                **model_kwargs,
            )


# Missing closing brace for AliyunBaiLianDefaultSpeechToText closure

Ensure that the remaining lines include proper closures (}), particularly after defining the final classes (if any). If additional files or modules are referenced elsewhere, make sure they exist and can be imported correctly.

[shaohuzhang1]

The provided code contains several improvements and corrections:

1. Imports:

   • Corrected AliyunBaiLianDefaultSTTModelCredential to match its implementation (note that it seems there was a typo).
   • Added missing import statement for AliyunBaiLianOriTTIConfig.

2. Initialization of Models:

   • Fixed the constructor call by using correct attribute names (module_type, credential, etc.).

3. Appending Default Model Info:

   • Added an extra default model info with ID 'default' and associated credential and model instance.

4. Module Name Correction:

   • Changed QwenOriTTIConfig to QwenTTIConfig.

Here's the updated code snippet:

from common.config.default_config import QwenTTIConfig, ALIYUN_BAI_LIAN_TTI_MODEL_ID_DEFAULT
...
aliyun_bai_lian_tti_model_configuration = AliyunBaiLianTTIConfiguration(
    module_name=ALIYUN_BAI_LIAN_TTI_MODULE_NAME,
    alias=_('tti'),
    version='1.0'
)

module_info_tti_list = [
    AliyunBaiLianTTIModuleInfo(aliyun_bai_lian_tti_model_configuration.module_id),
]

This should resolve the issues and improve the structure of the model initialization process.