Fix user (#46)

* 增加用户id

* 单元测试

Author: glide-the

tests/integration_tests/test_charglm3.py

@@ -25,6 +25,7 @@ def test_completions_charglm(logging_conf):
                 "bot_name": "苏梦远",
                 "user_name": "陆星辰"
             },
+            user_id="12345678"
         )
         print(response)
 

tests/integration_tests/test_chat.py

@@ -27,6 +27,7 @@ def test_completions_temp0(logging_conf):
             top_p=0.7,
             temperature=0,
             max_tokens=2000,
+            user_id="12345678"
         )
         print(response)
 
@@ -57,6 +58,7 @@ def test_completions_temp1(logging_conf):
             top_p=0.7,
             temperature=1,
             max_tokens=2000,
+            user_id="12345678"
         )
         print(response)
 
@@ -89,6 +91,7 @@ def test_completions_top0(logging_conf):
             top_p=0,
             temperature=0.9,
             max_tokens=2000,
+            user_id="12345678"
         )
         print(response)
 
@@ -121,6 +124,7 @@ def test_completions_top1(logging_conf):
             top_p=1,
             temperature=0.9,
             max_tokens=2000,
+            user_id="12345678"
         )
         print(response)
 
@@ -160,6 +164,7 @@ def test_completions(logging_conf):
                     }
                 }
             ],
+            user_id="12345678",
             extra_body={"temperature": 0.5, "max_tokens": 50},
         )
         print(response)
@@ -201,6 +206,7 @@ def test_completions_disenable_web_search(logging_conf):
                     }
                 }
             ],
+            user_id="12345678",
             extra_body={"temperature": 0.5, "max_tokens": 50},
         )
         print(response)
@@ -243,6 +249,7 @@ def test_completions_enable_web_search(logging_conf):
                 }
             ],
             extra_body={"temperature": 0.5, "max_tokens": 50},
+            user_id="12345678"
         )
         print(response)
 
@@ -275,6 +282,7 @@ def test_completions_stream(logging_conf):
                 {"role": "user", "content": "创造一个更精准、吸引人的slogan"}
             ],
             extra_body={"temperature": 0.5, "max_tokens": 50},
+            user_id="12345678"
         )
         for item in response:
             print(item)
@@ -323,7 +331,9 @@ def test_completions_vis(logging_conf):
                         }
                     ]
                 }
-            ]
+            ],
+
+            user_id="12345678"
         )
         print(response)
 
@@ -371,7 +381,9 @@ def test_completions_vis_base64(test_file_path, logging_conf):
                         }
                     ]
                 }
-            ]
+            ],
+
+            user_id="12345678"
         )
         print(response)
 
@@ -411,6 +423,8 @@ def test_async_completions(logging_conf):
                 }
             ],
             extra_body={"temperature": 0.5, "max_tokens": 50},
+
+            user_id="12345678"
         )
         print(response)
 

tests/integration_tests/test_images.py

@@ -13,6 +13,7 @@ def test_images(logging_conf):
             model="cogview-3", #填写需要调用的模型名称
             prompt="一只可爱的小猫咪",
             extra_body={"user_id": "1222212"},
+            user_id="12345678"
         )
         print(response)
 

tests/integration_tests/test_videos.py

@@ -11,7 +11,9 @@ def test_videos(logging_conf):
     try:
         response = client.videos.generations(
             model="cogvideo",
-            prompt="一些相信光的人，举着奥特曼"
+            prompt="一个开船的人",
+
+            user_id="1212222"
         )
         print(response)
 
@@ -32,7 +34,8 @@ def test_videos_image_url(logging_conf):
         response = client.videos.generations(
             model="cogvideo",
             image_url="https://cdn.bigmodel.cn/static/platform/images/solutions/car/empowerment/icon-metric.png",
-            prompt="一些相信光的人，举着奥特曼"
+            prompt="一些相信光的人，举着奥特曼",
+            user_id="12222211"
         )
         print(response)
 

zhipuai/api_resource/chat/async_completions.py

@@ -27,6 +27,7 @@ def create(
             *,
             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,
@@ -72,6 +73,7 @@ def create(
         body = {
             "model": model,
             "request_id": request_id,
+            "user_id": user_id,
             "temperature": temperature,
             "top_p": top_p,
             "do_sample": do_sample,

zhipuai/api_resource/chat/completions.py

@@ -29,6 +29,7 @@ def create(
             *,
             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,
@@ -74,6 +75,7 @@ def create(
         body = deepcopy_minimal({
             "model": model,
             "request_id": request_id,
+            "user_id": user_id,
             "temperature": temperature,
             "top_p": top_p,
             "do_sample": do_sample,

zhipuai/api_resource/images.py

@@ -29,6 +29,7 @@ def generations(
             style: Optional[str] | NotGiven = NOT_GIVEN,
             user: str | NotGiven = NOT_GIVEN,
             request_id: Optional[str] | NotGiven = NOT_GIVEN,
+            user_id: Optional[str] | NotGiven = NOT_GIVEN,
             extra_headers: Headers | None = None,
             extra_body: Body | None = None,
             disable_strict_validation: Optional[bool] | None = None,
@@ -48,6 +49,7 @@ def generations(
                 "size": size,
                 "style": style,
                 "user": user,
+                "user_id": user_id,
                 "request_id": request_id,
             },
             options=make_request_options(