remove gitignore

.venv/lib/python3.10/site-packages/zhipuai/api_resource/chat/__init__.py

@@ -0,0 +1,17 @@
+from .async_completions import (
+    AsyncCompletions
+)
+
+from .chat import (
+    Chat
+)
+
+from .completions import (
+    Completions
+)
+
+__all__ = [
+    'AsyncCompletions'
+    'Chat'
+    'Completions'
+]

.venv/lib/python3.10/site-packages/zhipuai/api_resource/chat/async_completions.py

@@ -0,0 +1,120 @@
+from __future__ import annotations
+
+from typing import Union, List, Optional, TYPE_CHECKING, Dict
+
+import httpx
+import logging
+from typing_extensions import Literal
+
+from ...core import BaseAPI, maybe_transform, drop_prefix_image_data
+from ...core import NotGiven, NOT_GIVEN, Headers, Body
+from ...core import make_request_options
+from ...types.chat.async_chat_completion import AsyncTaskStatus, AsyncCompletion
+from ...types.chat.code_geex import code_geex_params
+from ...types.sensitive_word_check import SensitiveWordCheckRequest
+
+logger = logging.getLogger(__name__)
+
+if TYPE_CHECKING:
+    from ..._client import ZhipuAI
+
+
+class AsyncCompletions(BaseAPI):
+    def __init__(self, client: "ZhipuAI") -> None:
+        super().__init__(client)
+
+    def create(
+            self,
+            *,
+            model: str,
+            request_id: Optional[str] | NotGiven = NOT_GIVEN,
+            user_id: Optional[str] | NotGiven = NOT_GIVEN,
+            do_sample: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
+            temperature: Optional[float] | NotGiven = NOT_GIVEN,
+            top_p: Optional[float] | NotGiven = NOT_GIVEN,
+            max_tokens: int | NotGiven = NOT_GIVEN,
+            seed: int | NotGiven = NOT_GIVEN,
+            messages: Union[str, List[str], List[int], List[List[int]], None],
+            stop: Optional[Union[str, List[str], None]] | NotGiven = NOT_GIVEN,
+            sensitive_word_check: Optional[SensitiveWordCheckRequest] | NotGiven = NOT_GIVEN,
+            tools: Optional[object] | NotGiven = NOT_GIVEN,
+            tool_choice: str | NotGiven = NOT_GIVEN,
+            meta: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
+            extra: Optional[code_geex_params.CodeGeexExtra] | NotGiven = NOT_GIVEN,
+            extra_headers: Headers | None = None,
+            extra_body: Body | None = None,
+            timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+            response_format: object | None = None,
+            thinking: object | None = None,
+    ) -> AsyncTaskStatus:
+        _cast_type = AsyncTaskStatus
+        logger.debug(f"temperature:{temperature}, top_p:{top_p}")
+        if temperature is not None and temperature != NOT_GIVEN:
+
+            if temperature <= 0:
+                do_sample = False
+                temperature = 0.01
+                # logger.warning("temperature:取值范围是：(0.0, 1.0) 开区间，do_sample重写为:false（参数top_p temperture不生效）")
+            if temperature >= 1:
+                temperature = 0.99
+                # logger.warning("temperature:取值范围是：(0.0, 1.0) 开区间")
+        if top_p is not None and top_p != NOT_GIVEN:
+
+            if top_p >= 1:
+                top_p = 0.99
+                # logger.warning("top_p:取值范围是：(0.0, 1.0) 开区间，不能等于 0 或 1")
+            if top_p <= 0:
+                top_p = 0.01
+                # logger.warning("top_p:取值范围是：(0.0, 1.0) 开区间，不能等于 0 或 1")
+
+        logger.debug(f"temperature:{temperature}, top_p:{top_p}")
+        if isinstance(messages, List):
+            for item in messages:
+                if item.get('content'):
+                    item['content'] = drop_prefix_image_data(item['content'])
+
+        body = {
+            "model": model,
+            "request_id": request_id,
+            "user_id": user_id,
+            "temperature": temperature,
+            "top_p": top_p,
+            "do_sample": do_sample,
+            "max_tokens": max_tokens,
+            "seed": seed,
+            "messages": messages,
+            "stop": stop,
+            "sensitive_word_check": sensitive_word_check,
+            "tools": tools,
+            "tool_choice": tool_choice,
+            "meta": meta,
+            "extra": maybe_transform(extra, code_geex_params.CodeGeexExtra),
+            "response_format": response_format,
+            "thinking": thinking
+        }
+        return self._post(
+            "/async/chat/completions",
+
+            body=body,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_body=extra_body, timeout=timeout
+            ),
+            cast_type=_cast_type,
+            stream=False,
+        )
+
+    def retrieve_completion_result(
+            self,
+            id: str,
+            extra_headers: Headers | None = None,
+            extra_body: Body | None = None,
+            timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Union[AsyncCompletion, AsyncTaskStatus]:
+        _cast_type = Union[AsyncCompletion, AsyncTaskStatus]
+        return self._get(
+            path=f"/async-result/{id}",
+            cast_type=_cast_type,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_body=extra_body, timeout=timeout
+            ),
+        )

.venv/lib/python3.10/site-packages/zhipuai/api_resource/chat/chat.py

@@ -0,0 +1,18 @@
+from typing import TYPE_CHECKING
+from .completions import Completions
+from .async_completions import AsyncCompletions
+from ...core import BaseAPI, cached_property
+
+if TYPE_CHECKING:
+    from ..._client import ZhipuAI
+
+
+class Chat(BaseAPI):
+
+    @cached_property
+    def completions(self) -> Completions:
+        return Completions(self._client)
+
+    @cached_property
+    def asyncCompletions(self) -> AsyncCompletions:
+        return AsyncCompletions(self._client)

.venv/lib/python3.10/site-packages/zhipuai/api_resource/chat/completions.py

@@ -0,0 +1,109 @@
+from __future__ import annotations
+
+from typing import Union, List, Optional, TYPE_CHECKING, Dict
+
+import httpx
+import logging
+from typing_extensions import Literal
+
+from ...core import BaseAPI, deepcopy_minimal, maybe_transform, drop_prefix_image_data
+from ...core import NotGiven, NOT_GIVEN, Headers, Query, Body
+from ...core import make_request_options
+from ...core import StreamResponse
+from ...types.chat.chat_completion import Completion
+from ...types.chat.chat_completion_chunk import ChatCompletionChunk
+from ...types.chat.code_geex import code_geex_params
+from ...types.sensitive_word_check import SensitiveWordCheckRequest
+
+logger = logging.getLogger(__name__)
+
+if TYPE_CHECKING:
+    from ..._client import ZhipuAI
+
+
+class Completions(BaseAPI):
+    def __init__(self, client: "ZhipuAI") -> None:
+        super().__init__(client)
+
+    def create(
+            self,
+            *,
+            model: str,
+            request_id: Optional[str] | NotGiven = NOT_GIVEN,
+            user_id: Optional[str] | NotGiven = NOT_GIVEN,
+            do_sample: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
+            stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
+            temperature: Optional[float] | NotGiven = NOT_GIVEN,
+            top_p: Optional[float] | NotGiven = NOT_GIVEN,
+            max_tokens: int | NotGiven = NOT_GIVEN,
+            seed: int | NotGiven = NOT_GIVEN,
+            messages: Union[str, List[str], List[int], object, None],
+            stop: Optional[Union[str, List[str], None]] | NotGiven = NOT_GIVEN,
+            sensitive_word_check: Optional[SensitiveWordCheckRequest] | NotGiven = NOT_GIVEN,
+            tools: Optional[object] | NotGiven = NOT_GIVEN,
+            tool_choice: str | NotGiven = NOT_GIVEN,
+            meta: Optional[Dict[str,str]] | NotGiven = NOT_GIVEN,
+            extra: Optional[code_geex_params.CodeGeexExtra] | NotGiven = NOT_GIVEN,
+            extra_headers: Headers | None = None,
+            extra_body: Body | None = None,
+            timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+            response_format : object | None = None,
+            thinking :  object | None = None,
+    ) -> Completion | StreamResponse[ChatCompletionChunk]:
+        logger.debug(f"temperature:{temperature}, top_p:{top_p}")
+        if temperature is not None and temperature != NOT_GIVEN:
+
+            if temperature <= 0:
+                do_sample = False
+                temperature = 0.01
+                # logger.warning("temperature:取值范围是：(0.0, 1.0) 开区间，do_sample重写为:false（参数top_p temperture不生效）")
+            if temperature >= 1:
+                temperature = 0.99
+                # logger.warning("temperature:取值范围是：(0.0, 1.0) 开区间")
+        if top_p is not None and top_p != NOT_GIVEN:
+
+            if top_p >= 1:
+                top_p = 0.99
+                # logger.warning("top_p:取值范围是：(0.0, 1.0) 开区间，不能等于 0 或 1")
+            if top_p <= 0:
+                top_p = 0.01
+                # logger.warning("top_p:取值范围是：(0.0, 1.0) 开区间，不能等于 0 或 1")
+
+        logger.debug(f"temperature:{temperature}, top_p:{top_p}")
+        if isinstance(messages, List):
+            for item in messages:
+                if item.get('content'):
+                    item['content'] = drop_prefix_image_data(item['content'])
+
+        body = deepcopy_minimal({
+            "model": model,
+            "request_id": request_id,
+            "user_id": user_id,
+            "temperature": temperature,
+            "top_p": top_p,
+            "do_sample": do_sample,
+            "max_tokens": max_tokens,
+            "seed": seed,
+            "messages": messages,
+            "stop": stop,
+            "sensitive_word_check": sensitive_word_check,
+            "stream": stream,
+            "tools": tools,
+            "tool_choice": tool_choice,
+            "meta": meta,
+            "extra": maybe_transform(extra, code_geex_params.CodeGeexExtra),
+            "response_format": response_format,
+            "thinking": thinking
+        })
+        return self._post(
+            "/chat/completions",
+            body=body,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_body=extra_body, timeout=timeout
+            ),
+            cast_type=Completion,
+            stream=stream or False,
+            stream_cls=StreamResponse[ChatCompletionChunk],
+        )
+
+