diff --git a/tests/integration_tests/test_moderation.py b/tests/integration_tests/test_moderation.py index cbeb1e8..282c949 100644 --- a/tests/integration_tests/test_moderation.py +++ b/tests/integration_tests/test_moderation.py @@ -13,8 +13,8 @@ def test_completions_temp0(logging_conf): # 生成request_id request_id = time.time() print(f'request_id:{request_id}') - response = client.moderation.moderations.create( - model='moderation', input={'type': 'text', 'text': 'hello world '} + response = client.moderations.create( + model='moderations', input={'type': 'text', 'text': 'hello world '} ) print(response) diff --git a/zhipuai/_client.py b/zhipuai/_client.py index 0c66da1..eb63ce0 100644 --- a/zhipuai/_client.py +++ b/zhipuai/_client.py @@ -67,7 +67,7 @@ def __init__( self.assistant = api_resource.Assistant(self) self.web_search = api_resource.WebSearchApi(self) self.audio = api_resource.audio.Audio(self) - self.moderation = api_resource.moderation.Moderation(self) + self.moderations = api_resource.moderation.Moderations(self) self.agents = api_resource.agents.Agents(self) @property diff --git a/zhipuai/api_resource/__init__.py b/zhipuai/api_resource/__init__.py index 7bc5255..1c73bc3 100644 --- a/zhipuai/api_resource/__init__.py +++ b/zhipuai/api_resource/__init__.py @@ -38,7 +38,7 @@ ) from .moderation import ( - Moderation + Moderations ) from .web_search import ( diff --git a/zhipuai/api_resource/moderation/__init__.py b/zhipuai/api_resource/moderation/__init__.py index ab12acc..855f72a 100644 --- a/zhipuai/api_resource/moderation/__init__.py +++ b/zhipuai/api_resource/moderation/__init__.py @@ -1,7 +1,7 @@ -from .moderation import ( - Moderation +from .moderations import ( + Moderations ) __all__ = [ - 'Moderation' + 'Moderations' ] \ No newline at end of file diff --git a/zhipuai/api_resource/moderation/moderation.py b/zhipuai/api_resource/moderation/moderation.py deleted file mode 100644 index 1fe39e6..0000000 --- a/zhipuai/api_resource/moderation/moderation.py +++ /dev/null @@ -1,10 +0,0 @@ -from typing import TYPE_CHECKING - -from .moderations import Moderations -from ...core import BaseAPI, cached_property - -__all__ = ["Moderation"] -class Moderation(BaseAPI): - @cached_property - def moderations(self) -> Moderations: - return Moderations(self._client) \ No newline at end of file diff --git a/zhipuai/api_resource/moderation/moderations.py b/zhipuai/api_resource/moderation/moderations.py index 1f8bfc3..e31d59c 100644 --- a/zhipuai/api_resource/moderation/moderations.py +++ b/zhipuai/api_resource/moderation/moderations.py @@ -12,7 +12,7 @@ if TYPE_CHECKING: from ..._client import ZhipuAI - +__all__ = ["Moderations"] class Moderations(BaseAPI): def __init__(self, client: "ZhipuAI") -> None: super().__init__(client)