fix a type typo

jm12138 · Aug 22, 2023 · a97c92f · a97c92f
1 parent 4a7035b
commit a97c92f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/wenxinworkshop/apis.py b/wenxinworkshop/apis.py
@@ -115,13 +115,13 @@ class LLMAPI:
         penalty_score: Optional[float] = None,
         stream: Optional[bool] = None,
         user_id: Optional[str] = None,
-        chunk_size: Optional[int] = 512
+        chunk_size: int = 512
     ) -> Union[str, Generator[str, None, None]]:
         Get response from LLM API.
 
     stream_response(
         response: requests.Response,
-        chunk_size: Optional[int] = 512
+        chunk_size: int = 512
     ) -> Generator[str, None, None]:
         Stream response from LLM API.
 
@@ -203,7 +203,7 @@ def __call__(
         penalty_score: Optional[float] = None,
         stream: Optional[bool] = None,
         user_id: Optional[str] = None,
-        chunk_size: Optional[int] = 512
+        chunk_size: int = 512
     ) -> Union[str, Generator[str, None, None]]:
         '''
         Get response from LLM API.
@@ -228,7 +228,7 @@ def __call__(
         user_id : Optional[str], optional
             User ID of LLM API, by default None.
 
-        chunk_size : Optional[int], optional
+        chunk_size : int, optional
             Chunk size of LLM API, by default 512.
 
         Returns
@@ -317,7 +317,7 @@ def __call__(
     @staticmethod
     def stream_response(
         response: requests.Response,
-        chunk_size: Optional[int] = 512
+        chunk_size: int = 512
     ) -> Generator[str, None, None]:
         '''
         Stream response from LLM API.
@@ -327,7 +327,7 @@ def stream_response(
         response : requests.Response
             Response from LLM API.
 
-        chunk_size : Optional[int], optional
+        chunk_size : int, optional
             Chunk size of LLM API, by default 512.
 
         Yields