Dt8333
7dd95d8a59
* chore: auto fix by ruff * refactor: 统一修正返回类型注解为 None/bool 以匹配实现 * refactor: 将 _get_next_page 改为异步并移除多余的请求错误抛出 * refactor: 将 get_client 的返回类型改为 object * style: 为 LarkMessageEvent 的相关方法添加返回类型注解 None --------- Co-authored-by: Soulter <37870767+Soulter@users.noreply.github.com>
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
from ..register import register_provider_adapter
|
|
from .openai_source import ProviderOpenAIOfficial
|
|
|
|
|
|
@register_provider_adapter(
|
|
"xai_chat_completion", "xAI Chat Completion Provider Adapter"
|
|
)
|
|
class ProviderXAI(ProviderOpenAIOfficial):
|
|
def __init__(
|
|
self,
|
|
provider_config: dict,
|
|
provider_settings: dict,
|
|
) -> None:
|
|
super().__init__(provider_config, provider_settings)
|
|
|
|
def _maybe_inject_xai_search(self, payloads: dict) -> None:
|
|
"""当开启 xAI 原生搜索时，向请求体注入 Live Search 参数。
|
|
|
|
- 仅在 provider_config.xai_native_search 为 True 时生效
|
|
- 默认注入 {"mode": "auto"}
|
|
"""
|
|
if not bool(self.provider_config.get("xai_native_search", False)):
|
|
return
|
|
# OpenAI SDK 不识别的字段会在 _query/_query_stream 中放入 extra_body
|
|
payloads["search_parameters"] = {"mode": "auto"}
|
|
|
|
def _finally_convert_payload(self, payloads: dict) -> None:
|
|
self._maybe_inject_xai_search(payloads)
|
|
super()._finally_convert_payload(payloads)
|