aient 1.0.35__tar.gz → 1.0.36__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.
Files changed (72) hide show
  1. {aient-1.0.35/src/aient.egg-info → aient-1.0.36}/PKG-INFO +1 -1
  2. {aient-1.0.35 → aient-1.0.36}/setup.py +1 -1
  3. {aient-1.0.35 → aient-1.0.36}/src/aient/models/chatgpt.py +7 -0
  4. {aient-1.0.35 → aient-1.0.36/src/aient.egg-info}/PKG-INFO +1 -1
  5. {aient-1.0.35 → aient-1.0.36}/LICENSE +0 -0
  6. {aient-1.0.35 → aient-1.0.36}/MANIFEST.in +0 -0
  7. {aient-1.0.35 → aient-1.0.36}/README.md +0 -0
  8. {aient-1.0.35 → aient-1.0.36}/setup.cfg +0 -0
  9. {aient-1.0.35 → aient-1.0.36}/src/aient/__init__.py +0 -0
  10. {aient-1.0.35 → aient-1.0.36}/src/aient/core/.git +0 -0
  11. {aient-1.0.35 → aient-1.0.36}/src/aient/core/__init__.py +0 -0
  12. {aient-1.0.35 → aient-1.0.36}/src/aient/core/log_config.py +0 -0
  13. {aient-1.0.35 → aient-1.0.36}/src/aient/core/models.py +0 -0
  14. {aient-1.0.35 → aient-1.0.36}/src/aient/core/request.py +0 -0
  15. {aient-1.0.35 → aient-1.0.36}/src/aient/core/response.py +0 -0
  16. {aient-1.0.35 → aient-1.0.36}/src/aient/core/test/test_base_api.py +0 -0
  17. {aient-1.0.35 → aient-1.0.36}/src/aient/core/test/test_image.py +0 -0
  18. {aient-1.0.35 → aient-1.0.36}/src/aient/core/test/test_payload.py +0 -0
  19. {aient-1.0.35 → aient-1.0.36}/src/aient/core/utils.py +0 -0
  20. {aient-1.0.35 → aient-1.0.36}/src/aient/models/__init__.py +0 -0
  21. {aient-1.0.35 → aient-1.0.36}/src/aient/models/audio.py +0 -0
  22. {aient-1.0.35 → aient-1.0.36}/src/aient/models/base.py +0 -0
  23. {aient-1.0.35 → aient-1.0.36}/src/aient/models/claude.py +0 -0
  24. {aient-1.0.35 → aient-1.0.36}/src/aient/models/duckduckgo.py +0 -0
  25. {aient-1.0.35 → aient-1.0.36}/src/aient/models/gemini.py +0 -0
  26. {aient-1.0.35 → aient-1.0.36}/src/aient/models/groq.py +0 -0
  27. {aient-1.0.35 → aient-1.0.36}/src/aient/models/vertex.py +0 -0
  28. {aient-1.0.35 → aient-1.0.36}/src/aient/plugins/__init__.py +0 -0
  29. {aient-1.0.35 → aient-1.0.36}/src/aient/plugins/arXiv.py +0 -0
  30. {aient-1.0.35 → aient-1.0.36}/src/aient/plugins/config.py +0 -0
  31. {aient-1.0.35 → aient-1.0.36}/src/aient/plugins/image.py +0 -0
  32. {aient-1.0.35 → aient-1.0.36}/src/aient/plugins/registry.py +0 -0
  33. {aient-1.0.35 → aient-1.0.36}/src/aient/plugins/run_python.py +0 -0
  34. {aient-1.0.35 → aient-1.0.36}/src/aient/plugins/today.py +0 -0
  35. {aient-1.0.35 → aient-1.0.36}/src/aient/plugins/websearch.py +0 -0
  36. {aient-1.0.35 → aient-1.0.36}/src/aient/utils/__init__.py +0 -0
  37. {aient-1.0.35 → aient-1.0.36}/src/aient/utils/prompt.py +0 -0
  38. {aient-1.0.35 → aient-1.0.36}/src/aient/utils/scripts.py +0 -0
  39. {aient-1.0.35 → aient-1.0.36}/src/aient.egg-info/SOURCES.txt +0 -0
  40. {aient-1.0.35 → aient-1.0.36}/src/aient.egg-info/dependency_links.txt +0 -0
  41. {aient-1.0.35 → aient-1.0.36}/src/aient.egg-info/requires.txt +0 -0
  42. {aient-1.0.35 → aient-1.0.36}/src/aient.egg-info/top_level.txt +0 -0
  43. {aient-1.0.35 → aient-1.0.36}/test/test.py +0 -0
  44. {aient-1.0.35 → aient-1.0.36}/test/test_API.py +0 -0
  45. {aient-1.0.35 → aient-1.0.36}/test/test_Deepbricks.py +0 -0
  46. {aient-1.0.35 → aient-1.0.36}/test/test_Web_crawler.py +0 -0
  47. {aient-1.0.35 → aient-1.0.36}/test/test_aiwaves.py +0 -0
  48. {aient-1.0.35 → aient-1.0.36}/test/test_aiwaves_arxiv.py +0 -0
  49. {aient-1.0.35 → aient-1.0.36}/test/test_ask_gemini.py +0 -0
  50. {aient-1.0.35 → aient-1.0.36}/test/test_class.py +0 -0
  51. {aient-1.0.35 → aient-1.0.36}/test/test_claude.py +0 -0
  52. {aient-1.0.35 → aient-1.0.36}/test/test_claude_zh_char.py +0 -0
  53. {aient-1.0.35 → aient-1.0.36}/test/test_ddg_search.py +0 -0
  54. {aient-1.0.35 → aient-1.0.36}/test/test_download_pdf.py +0 -0
  55. {aient-1.0.35 → aient-1.0.36}/test/test_gemini.py +0 -0
  56. {aient-1.0.35 → aient-1.0.36}/test/test_get_token_dict.py +0 -0
  57. {aient-1.0.35 → aient-1.0.36}/test/test_google_search.py +0 -0
  58. {aient-1.0.35 → aient-1.0.36}/test/test_jieba.py +0 -0
  59. {aient-1.0.35 → aient-1.0.36}/test/test_json.py +0 -0
  60. {aient-1.0.35 → aient-1.0.36}/test/test_langchain_search_old.py +0 -0
  61. {aient-1.0.35 → aient-1.0.36}/test/test_logging.py +0 -0
  62. {aient-1.0.35 → aient-1.0.36}/test/test_ollama.py +0 -0
  63. {aient-1.0.35 → aient-1.0.36}/test/test_plugin.py +0 -0
  64. {aient-1.0.35 → aient-1.0.36}/test/test_py_run.py +0 -0
  65. {aient-1.0.35 → aient-1.0.36}/test/test_requests.py +0 -0
  66. {aient-1.0.35 → aient-1.0.36}/test/test_search.py +0 -0
  67. {aient-1.0.35 → aient-1.0.36}/test/test_tikitoken.py +0 -0
  68. {aient-1.0.35 → aient-1.0.36}/test/test_token.py +0 -0
  69. {aient-1.0.35 → aient-1.0.36}/test/test_url.py +0 -0
  70. {aient-1.0.35 → aient-1.0.36}/test/test_whisper.py +0 -0
  71. {aient-1.0.35 → aient-1.0.36}/test/test_wildcard.py +0 -0
  72. {aient-1.0.35 → aient-1.0.36}/test/test_yjh.py +0 -0
{aient-1.0.35/src/aient.egg-info → aient-1.0.36}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aient
3
- Version: 1.0.35
3
+ Version: 1.0.36
4
4
  Summary: Aient: The Awakening of Agent.
5
5
  Description-Content-Type: text/markdown
6
6
  License-File: LICENSE
{aient-1.0.35 → aient-1.0.36}/setup.py RENAMED
@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
4
4
 
5
5
  setup(
6
6
  name="aient",
7
- version="1.0.35",
7
+ version="1.0.36",
8
8
  description="Aient: The Awakening of Agent.",
9
9
  long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
10
10
  long_description_content_type="text/markdown",
{aient-1.0.35 → aient-1.0.36}/src/aient/models/chatgpt.py RENAMED
@@ -61,6 +61,7 @@ class chatgpt(BaseLLM):
61
61
  print_log: bool = False,
62
62
  tools: Optional[Union[list, str, Callable]] = [],
63
63
  function_call_max_loop: int = 3,
64
+ cut_history_by_function_name: str = "",
64
65
  ) -> None:
65
66
  """
66
67
  Initialize Chatbot with API key (from https://platform.openai.com/account/api-keys)
@@ -76,6 +77,7 @@ class chatgpt(BaseLLM):
76
77
  }
77
78
  self.function_calls_counter = {}
78
79
  self.function_call_max_loop = function_call_max_loop
80
+ self.cut_history_by_function_name = cut_history_by_function_name
79
81
 
80
82
 
81
83
  # 注册和处理传入的工具
@@ -124,6 +126,11 @@ class chatgpt(BaseLLM):
124
126
  if function_name == "" and message and message != None:
125
127
  self.conversation[convo_id].append({"role": role, "content": message})
126
128
  elif function_name != "" and message and message != None:
129
+ # 删除从 cut_history_by_function_name 以后的所有历史记录
130
+ if function_name == self.cut_history_by_function_name:
131
+ matching_message = next(filter(lambda x: safe_get(x, "tool_calls", 0, "function", "name", default="") == 'get_next_pdf', self.conversation[convo_id]), None)
132
+ if matching_message is not None:
133
+ self.conversation[convo_id] = self.conversation[convo_id][:self.conversation[convo_id].index(matching_message)]
127
134
  self.conversation[convo_id].append({
128
135
  "role": "assistant",
129
136
  "tool_calls": [
{aient-1.0.35 → aient-1.0.36/src/aient.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aient
3
- Version: 1.0.35
3
+ Version: 1.0.36
4
4
  Summary: Aient: The Awakening of Agent.
5
5
  Description-Content-Type: text/markdown
6
6
  License-File: LICENSE
{aient-1.0.35 → aient-1.0.36}/LICENSE RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/MANIFEST.in RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/README.md RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/setup.cfg RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/src/aient/__init__.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/src/aient/core/.git RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/src/aient/core/models.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/src/aient/core/request.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/src/aient/core/utils.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/src/aient/models/audio.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/src/aient/models/base.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/src/aient/models/groq.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/src/aient/utils/prompt.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_API.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_Deepbricks.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_Web_crawler.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_aiwaves.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_ask_gemini.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_class.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_claude.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_ddg_search.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_download_pdf.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_gemini.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_jieba.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_json.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_logging.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_ollama.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_plugin.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_py_run.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_requests.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_search.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_tikitoken.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_token.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_url.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_whisper.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_wildcard.py RENAMED
File without changes
{aient-1.0.35 → aient-1.0.36}/test/test_yjh.py RENAMED
File without changes