aient 1.1.88__tar.gz → 1.1.89__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {aient-1.1.88 → aient-1.1.89}/PKG-INFO +1 -1
- {aient-1.1.88 → aient-1.1.89}/aient/models/chatgpt.py +6 -1
- {aient-1.1.88 → aient-1.1.89}/aient.egg-info/PKG-INFO +1 -1
- {aient-1.1.88 → aient-1.1.89}/pyproject.toml +1 -1
- {aient-1.1.88 → aient-1.1.89}/LICENSE +0 -0
- {aient-1.1.88 → aient-1.1.89}/README.md +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/__init__.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/__init__.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/log_config.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/models.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/request.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/response.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/test/test_base_api.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/test/test_geminimask.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/test/test_image.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/test/test_payload.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/core/utils.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/models/__init__.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/models/audio.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/models/base.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/__init__.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/arXiv.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/config.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/excute_command.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/get_time.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/image.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/list_directory.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/read_file.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/read_image.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/readonly.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/registry.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/run_python.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/websearch.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/plugins/write_file.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/utils/__init__.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/utils/prompt.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient/utils/scripts.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient.egg-info/SOURCES.txt +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient.egg-info/dependency_links.txt +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient.egg-info/requires.txt +0 -0
- {aient-1.1.88 → aient-1.1.89}/aient.egg-info/top_level.txt +0 -0
- {aient-1.1.88 → aient-1.1.89}/setup.cfg +0 -0
- {aient-1.1.88 → aient-1.1.89}/test/test_Web_crawler.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/test/test_ddg_search.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/test/test_google_search.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/test/test_ollama.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/test/test_plugin.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/test/test_search.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/test/test_url.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/test/test_whisper.py +0 -0
- {aient-1.1.88 → aient-1.1.89}/test/test_yjh.py +0 -0
@@ -757,6 +757,7 @@ class chatgpt(BaseLLM):
|
|
757
757
|
|
758
758
|
# 发送请求并处理响应
|
759
759
|
retry_times = 0
|
760
|
+
error_to_raise = None
|
760
761
|
while True:
|
761
762
|
tmp_post_json = copy.deepcopy(json_post)
|
762
763
|
if need_done_prompt:
|
@@ -840,7 +841,8 @@ class chatgpt(BaseLLM):
|
|
840
841
|
continue
|
841
842
|
except InputTokenCountExceededError as e:
|
842
843
|
self.logger.error(f"The request body is too long: {e}")
|
843
|
-
|
844
|
+
error_to_raise = e
|
845
|
+
break
|
844
846
|
except BadRequestError as e:
|
845
847
|
self.logger.error(f"Bad request error: {e}")
|
846
848
|
raise
|
@@ -873,6 +875,9 @@ class chatgpt(BaseLLM):
|
|
873
875
|
if retry_times == 9:
|
874
876
|
raise RetryFailedError(str(e))
|
875
877
|
|
878
|
+
if error_to_raise:
|
879
|
+
raise error_to_raise
|
880
|
+
|
876
881
|
def ask_stream(
|
877
882
|
self,
|
878
883
|
prompt: list,
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|