agentUniverse 0.0.11.dev1__tar.gz → 0.0.12.dev1__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 (233) hide show
  1. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/PKG-INFO +2 -1
  2. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/knowledge.py +2 -0
  3. agentuniverse-0.0.12.dev1/agentuniverse/agent/action/knowledge/reader/file/txt_reader.py +39 -0
  4. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/store/chroma_store.py +1 -1
  5. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/tool/tool.py +3 -0
  6. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/executing_agent/executing_agent.py +36 -3
  7. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/message.py +1 -1
  8. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/peer_planner/peer_planner.py +36 -19
  9. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/planner.py +9 -9
  10. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/react_planner/stream_callback.py +21 -1
  11. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/dal/request_library.py +3 -4
  12. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/request_task.py +4 -1
  13. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/thread_with_result.py +13 -0
  14. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/web_util.py +20 -0
  15. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/agentuniverse.py +6 -5
  16. agentuniverse-0.0.12.dev1/agentuniverse/base/annotation/trace.py +300 -0
  17. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/component/component_base.py +3 -1
  18. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/component/component_enum.py +9 -0
  19. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/application_configer/app_configer.py +6 -0
  20. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/context/framework_context_manager.py +12 -1
  21. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/memory_util.py +4 -2
  22. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/monitor/monitor.py +79 -0
  23. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/database/sqldb_wrapper.py +2 -2
  24. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/prompt/prompt_manager.py +1 -1
  25. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/pyproject.toml +2 -1
  26. agentuniverse-0.0.11.dev1/agentuniverse/base/annotation/trace.py +0 -144
  27. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/LICENSE +0 -0
  28. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/README_PYPI.md +0 -0
  29. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/__init__.py +0 -0
  30. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/__init__.py +0 -0
  31. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/__init__.py +0 -0
  32. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/__init__.py +0 -0
  33. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/embedding/__init__.py +0 -0
  34. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/embedding/dashscope_embedding.py +0 -0
  35. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/embedding/embedding.py +0 -0
  36. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/embedding/openai_embedding.py +0 -0
  37. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/knowledge_manager.py +0 -0
  38. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/reader/__init__.py +0 -0
  39. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/reader/file/__init__.py +0 -0
  40. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/reader/file/docx_reader.py +0 -0
  41. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/reader/file/file_reader.py +0 -0
  42. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/reader/file/pdf_reader.py +0 -0
  43. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/reader/file/pptx_reader.py +0 -0
  44. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/reader/file/web_pdf_reader.py +0 -0
  45. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/reader/reader.py +0 -0
  46. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/store/__init__.py +0 -0
  47. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/store/document.py +0 -0
  48. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/store/milvus_store.py +0 -0
  49. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/store/query.py +0 -0
  50. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/knowledge/store/store.py +0 -0
  51. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/tool/__init__.py +0 -0
  52. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/tool/enum.py +0 -0
  53. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/action/tool/tool_manager.py +0 -0
  54. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/agent.py +0 -0
  55. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/agent_manager.py +0 -0
  56. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/agent_model.py +0 -0
  57. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/__init__.py +0 -0
  58. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/executing_agent/__init__.py +0 -0
  59. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/executing_agent/default_cn_prompt.yaml +0 -0
  60. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/executing_agent/default_en_prompt.yaml +0 -0
  61. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/executing_agent/executing_agent.yaml +0 -0
  62. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/expressing_agent/__init__.py +0 -0
  63. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/expressing_agent/default_cn_prompt.yaml +0 -0
  64. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/expressing_agent/default_en_prompt.yaml +0 -0
  65. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/expressing_agent/expressing_agent.py +0 -0
  66. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/expressing_agent/expressing_agent.yaml +0 -0
  67. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/nl2api_agent/__init__.py +0 -0
  68. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/nl2api_agent/default_cn_prompt.yaml +0 -0
  69. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/nl2api_agent/default_en_prompt.yaml +0 -0
  70. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/nl2api_agent/nl2api_agent.py +0 -0
  71. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/nl2api_agent/nl2api_agent.yaml +0 -0
  72. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/peer_agent/__init__.py +0 -0
  73. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/peer_agent/peer_agent.py +0 -0
  74. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/peer_agent/peer_agent.yaml +0 -0
  75. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/planning_agent/__init__.py +0 -0
  76. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/planning_agent/default_cn_prompt.yaml +0 -0
  77. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/planning_agent/default_en_prompt.yaml +0 -0
  78. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/planning_agent/planning_agent.py +0 -0
  79. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/planning_agent/planning_agent.yaml +0 -0
  80. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/rag_agent/__init__.py +0 -0
  81. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/rag_agent/default_cn_prompt.yaml +0 -0
  82. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/rag_agent/default_en_prompt.yaml +0 -0
  83. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/rag_agent/rag_agent.py +0 -0
  84. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/rag_agent/rag_agent.yaml +0 -0
  85. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/react_agent/__init__.py +0 -0
  86. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/react_agent/default_cn_prompt.yaml +0 -0
  87. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/react_agent/default_en_prompt.yaml +0 -0
  88. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/react_agent/react_agent.py +0 -0
  89. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/react_agent/react_agent.yaml +0 -0
  90. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/reviewing_agent/__init__.py +0 -0
  91. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/reviewing_agent/default_cn_prompt.yaml +0 -0
  92. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/reviewing_agent/default_en_prompt.yaml +0 -0
  93. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/reviewing_agent/reviewing_agent.py +0 -0
  94. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/default/reviewing_agent/reviewing_agent.yaml +0 -0
  95. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/input_object.py +0 -0
  96. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/__init__.py +0 -0
  97. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/chat_memory.py +0 -0
  98. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/default/__init__.py +0 -0
  99. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/default/default_memory.py +0 -0
  100. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/default/default_memory.yaml +0 -0
  101. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/enum.py +0 -0
  102. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/langchain_instance.py +0 -0
  103. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/memory.py +0 -0
  104. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/memory_manager.py +0 -0
  105. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/summarizer_cn_prompt.yaml +0 -0
  106. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/memory/summarizer_en_prompt.yaml +0 -0
  107. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/output_object.py +0 -0
  108. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/__init__.py +0 -0
  109. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/__init__.py +0 -0
  110. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/executing_planner/__init__.py +0 -0
  111. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/executing_planner/executing_planner.py +0 -0
  112. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/executing_planner/executing_planner.yaml +0 -0
  113. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/expressing_planner/__init__.py +0 -0
  114. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/expressing_planner/expressing_planner.py +0 -0
  115. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/expressing_planner/expressing_planner.yaml +0 -0
  116. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/nl2api_planner/__init__.py +0 -0
  117. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/nl2api_planner/nl2api_planner.py +0 -0
  118. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/nl2api_planner/nl2api_planner.yaml +0 -0
  119. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/peer_planner/__init__.py +0 -0
  120. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/peer_planner/peer_planner.yaml +0 -0
  121. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/planner_manager.py +0 -0
  122. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/planning_planner/__init__.py +0 -0
  123. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/planning_planner/planning_planner.py +0 -0
  124. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/planning_planner/planning_planner.yaml +0 -0
  125. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/rag_planner/__init__.py +0 -0
  126. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/rag_planner/rag_planner.py +0 -0
  127. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/rag_planner/rag_planner.yaml +0 -0
  128. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/react_planner/__init__.py +0 -0
  129. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/react_planner/react_planner.py +0 -0
  130. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/react_planner/react_planner.yaml +0 -0
  131. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/reviewing_planner/__init__.py +0 -0
  132. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/reviewing_planner/reviewing_planner.py +0 -0
  133. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent/plan/planner/reviewing_planner/reviewing_planner.yaml +0 -0
  134. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/__init__.py +0 -0
  135. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/service.py +0 -0
  136. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/service_configer.py +0 -0
  137. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/service_instance.py +0 -0
  138. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/service_manager.py +0 -0
  139. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/__init__.py +0 -0
  140. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/dal/__init__.py +0 -0
  141. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/dal/entity/__init__.py +0 -0
  142. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/dal/entity/request_do.py +0 -0
  143. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/flask_server.py +0 -0
  144. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/gunicorn_server.py +0 -0
  145. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/post_fork_queue.py +0 -0
  146. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/rpc/__init__.py +0 -0
  147. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/rpc/grpc/__init__.py +0 -0
  148. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/rpc/grpc/agentuniverse_service.proto +0 -0
  149. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/rpc/grpc/agentuniverse_service_pb2.py +0 -0
  150. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/rpc/grpc/agentuniverse_service_pb2_grpc.py +0 -0
  151. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/rpc/grpc/grpc_server_booster.py +0 -0
  152. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/rpc/rpc_server.py +0 -0
  153. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/agent_serve/web/web_booster.py +0 -0
  154. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/__init__.py +0 -0
  155. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/annotation/__init__.py +0 -0
  156. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/annotation/singleton.py +0 -0
  157. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/component/__init__.py +0 -0
  158. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/component/application_component_manager.py +0 -0
  159. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/component/component_configer_util.py +0 -0
  160. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/component/component_manager_base.py +0 -0
  161. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/__init__.py +0 -0
  162. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/application_configer/__init__.py +0 -0
  163. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/application_configer/application_config_manager.py +0 -0
  164. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/__init__.py +0 -0
  165. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/component_configer.py +0 -0
  166. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/configers/__init__.py +0 -0
  167. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/configers/agent_configer.py +0 -0
  168. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/configers/knowledge_configer.py +0 -0
  169. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/configers/llm_configer.py +0 -0
  170. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/configers/memory_configer.py +0 -0
  171. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/configers/planner_configer.py +0 -0
  172. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/configers/prompt_configer.py +0 -0
  173. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/configers/sqldb_wrapper_config.py +0 -0
  174. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/component_configer/configers/tool_configer.py +0 -0
  175. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/config_type_enum.py +0 -0
  176. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/configer.py +0 -0
  177. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/custom_configer/__init__.py +0 -0
  178. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/config/custom_configer/custom_key_configer.py +0 -0
  179. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/context/__init__.py +0 -0
  180. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/context/framework_context.py +0 -0
  181. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/__init__.py +0 -0
  182. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/env_util.py +0 -0
  183. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/logging/__init__.py +0 -0
  184. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/logging/general_logger.py +0 -0
  185. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/logging/logging_config.py +0 -0
  186. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/logging/logging_util.py +0 -0
  187. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/monitor/__init__.py +0 -0
  188. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/prompt/__init__.py +0 -0
  189. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/prompt/combine_cn_prompt.yaml +0 -0
  190. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/prompt/combine_en_prompt.yaml +0 -0
  191. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/prompt/summary_cn_prompt.yaml +0 -0
  192. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/prompt/summary_en_prompt.yaml +0 -0
  193. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/prompt_util.py +0 -0
  194. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/base/util/system_util.py +0 -0
  195. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/database/__init__.py +0 -0
  196. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/database/sqldb_wrapper_manager.py +0 -0
  197. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/__init__.py +0 -0
  198. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/claude_langchain_instance.py +0 -0
  199. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/__init__.py +0 -0
  200. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/baichuan_openai_style_llm.py +0 -0
  201. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/baichuan_openai_style_llm.yaml +0 -0
  202. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/claude_llm.py +0 -0
  203. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/claude_llm.yaml +0 -0
  204. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/deep_seek_openai_style_llm.py +0 -0
  205. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/deep_seek_openai_style_llm.yaml +0 -0
  206. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/default_ollama_llm.py +0 -0
  207. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/default_ollama_llm.yaml +0 -0
  208. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/default_openai_llm.py +0 -0
  209. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/default_openai_llm.yaml +0 -0
  210. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/kimi_openai_style_llm.py +0 -0
  211. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/kimi_openai_style_llm.yaml +0 -0
  212. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/qwen_openai_style_llm.py +0 -0
  213. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/qwen_openai_style_llm.yaml +0 -0
  214. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/wenxin_llm.py +0 -0
  215. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/default/wenxin_llm.yaml +0 -0
  216. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/langchain_instance.py +0 -0
  217. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/llm.py +0 -0
  218. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/llm_manager.py +0 -0
  219. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/llm_output.py +0 -0
  220. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/ollama_langchain_instance.py +0 -0
  221. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/openai_llm.py +0 -0
  222. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/openai_style_langchain_instance.py +0 -0
  223. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/openai_style_llm.py +0 -0
  224. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/llm/wenxin_langchain_instance.py +0 -0
  225. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/prompt/__init__.py +0 -0
  226. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/prompt/chat_prompt.py +0 -0
  227. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/prompt/enum.py +0 -0
  228. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/prompt/prompt.py +0 -0
  229. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse/prompt/prompt_model.py +0 -0
  230. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse_connector/__init__.py +0 -0
  231. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse_extension/__init__.py +0 -0
  232. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse_extension/logger/__init__.py +0 -0
  233. {agentuniverse-0.0.11.dev1 → agentuniverse-0.0.12.dev1}/agentuniverse_extension/logger/sls_sink.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: agentUniverse
3
- Version: 0.0.11.dev1
3
+ Version: 0.0.12.dev1
4
4
  Summary: agentUniverse is a framework for developing applications powered by multi-agent base on large language model.
5
5
  Home-page: https://github.com/alipay/agentUniverse
6
6
  Author: AntGroup
@@ -36,6 +36,7 @@ Requires-Dist: myst-parser (>=2.0.0,<3.0.0)
36
36
  Requires-Dist: numpy (>=1.26.0,<2.0.0)
37
37
  Requires-Dist: ollama (>=0.2.1,<0.3.0)
38
38
  Requires-Dist: openai (==1.13.3)
39
+ Requires-Dist: openpyxl (>=3.1.5,<4.0.0)
39
40
  Requires-Dist: opentelemetry-exporter-otlp-proto-grpc (>=1.25.0,<2.0.0)
40
41
  Requires-Dist: pandas (>=2.2.2,<3.0.0)
41
42
  Requires-Dist: pyarrow (>=16.1.0,<17.0.0)
@@ -14,6 +14,7 @@ from agentuniverse.agent.action.knowledge.reader.reader import Reader
14
14
  from agentuniverse.agent.action.knowledge.store.document import Document
15
15
  from agentuniverse.agent.action.knowledge.store.query import Query
16
16
  from agentuniverse.agent.action.knowledge.store.store import Store
17
+ from agentuniverse.base.annotation.trace import trace_knowledge
17
18
  from agentuniverse.base.component.component_base import ComponentBase
18
19
  from agentuniverse.base.component.component_enum import ComponentEnum
19
20
  from agentuniverse.base.config.application_configer.application_config_manager import ApplicationConfigManager
@@ -53,6 +54,7 @@ class Knowledge(ComponentBase):
53
54
  document_list: List[Document] = self.reader.load_data()
54
55
  self.store.insert_documents(document_list, **kwargs)
55
56
 
57
+ @trace_knowledge
56
58
  def query_knowledge(self, **kwargs) -> List[Document]:
57
59
  """Query the knowledge.
58
60
 
@@ -0,0 +1,39 @@
1
+
2
+ from pathlib import Path
3
+ from typing import List, Optional, Dict
4
+
5
+ from agentuniverse.agent.action.knowledge.reader.reader import Reader
6
+ from agentuniverse.agent.action.knowledge.store.document import Document
7
+
8
+
9
+ class LineTxtReader(Reader):
10
+
11
+ def load_data(self, fpath: Path, ext_info: Optional[Dict] = None) -> List[Document]:
12
+ dlist = []
13
+
14
+ with open(fpath, 'r', encoding='utf-8') as file:
15
+
16
+ metadata = {"file_name": file.name}
17
+ if ext_info is not None:
18
+ metadata.update(ext_info)
19
+
20
+ for line in file:
21
+ dlist.append(Document(text=line, metadata=metadata or {}))
22
+
23
+ return dlist
24
+
25
+
26
+ class TxtReader(Reader):
27
+ """Txt reader."""
28
+
29
+ def load_data(self, fpath: Path, ext_info: Optional[Dict] = None) -> List[Document]:
30
+
31
+ with open(fpath, 'r', encoding='utf-8') as file:
32
+
33
+ metadata = {"file_name": file.name}
34
+ if ext_info is not None:
35
+ metadata.update(ext_info)
36
+
37
+ txt = file.read()
38
+
39
+ return [Document(text=txt, metadata=metadata or {})]
@@ -96,7 +96,7 @@ class ChromaStore(Store):
96
96
  self.collection.add(
97
97
  documents=[document.text],
98
98
  metadatas=[document.metadata],
99
- embeddings=[embedding] if embedding is not None else None,
99
+ embeddings=[embedding] if len(embedding) > 0 else None,
100
100
  ids=[document.id]
101
101
  )
102
102
 
@@ -13,6 +13,7 @@ from pydantic import BaseModel
13
13
  from langchain.tools import Tool as LangchainTool
14
14
 
15
15
  from agentuniverse.agent.action.tool.enum import ToolTypeEnum
16
+ from agentuniverse.base.annotation.trace import trace_tool
16
17
  from agentuniverse.base.component.component_base import ComponentBase
17
18
  from agentuniverse.base.component.component_enum import ComponentEnum
18
19
  from agentuniverse.base.config.application_configer.application_config_manager import ApplicationConfigManager
@@ -61,6 +62,7 @@ class Tool(ComponentBase):
61
62
  def __init__(self, **kwargs):
62
63
  super().__init__(component_type=ComponentEnum.TOOL, **kwargs)
63
64
 
65
+ @trace_tool
64
66
  def run(self, **kwargs):
65
67
  """The callable method that runs the tool."""
66
68
  self.input_check(kwargs)
@@ -73,6 +75,7 @@ class Tool(ComponentBase):
73
75
  if key not in kwargs.keys():
74
76
  raise Exception(f'{self.get_instance_code()} - The input must include key: {key}.')
75
77
 
78
+ @trace_tool
76
79
  def langchain_run(self, *args, callbacks=None, **kwargs):
77
80
  """The callable method that runs the tool."""
78
81
  kwargs["callbacks"] = callbacks
@@ -5,21 +5,23 @@
5
5
  # @Email : lc299034@antgroup.com
6
6
  # @FileName: executing_agent.py
7
7
  """Executing Agent module."""
8
- import copy
9
8
  from concurrent.futures import ThreadPoolExecutor, wait, ALL_COMPLETED
10
9
  from typing import Optional, Any
11
10
 
12
11
  from agentuniverse.agent.action.tool.tool_manager import ToolManager
13
12
  from agentuniverse.agent.agent import Agent
13
+ from agentuniverse.agent.agent_model import AgentModel
14
14
  from agentuniverse.agent.input_object import InputObject
15
15
  from agentuniverse.agent.plan.planner.planner import Planner
16
16
  from agentuniverse.agent.plan.planner.planner_manager import PlannerManager
17
+ from agentuniverse.base.context.framework_context_manager import FrameworkContextManager
17
18
 
18
19
 
19
20
  class ExecutingAgent(Agent):
20
21
  """Executing Agent class."""
21
22
 
22
23
  executor: Optional[Any] = ThreadPoolExecutor(max_workers=10, thread_name_prefix="executing_agent")
24
+ _context_values: Optional[dict] = {}
23
25
 
24
26
  def input_keys(self) -> list[str]:
25
27
  """Return the input keys of the Agent."""
@@ -54,12 +56,16 @@ class ExecutingAgent(Agent):
54
56
  llm_result = []
55
57
  executing_result = []
56
58
  futures = planner_result.get('futures')
59
+ index = 1
60
+ # assemble results of the execution process.
57
61
  for future in futures:
58
62
  task_result = future.result()
59
63
  llm_result.append(task_result)
60
64
  executing_result.append({
61
- 'input': task_result['input'], 'output': task_result['output']
65
+ 'input': f"Question {index}: " + task_result.get('input', ''),
66
+ 'output': f"Answer {index}: " + task_result.get('output', '')
62
67
  })
68
+ index += 1
63
69
 
64
70
  return {'executing_result': executing_result, 'llm_result': llm_result}
65
71
 
@@ -73,6 +79,7 @@ class ExecutingAgent(Agent):
73
79
  Returns:
74
80
  dict: Agent result object.
75
81
  """
82
+ self._context_values: dict = FrameworkContextManager().get_all_contexts()
76
83
  framework = agent_input.get('framework', [])
77
84
  futures = []
78
85
  for task in framework:
@@ -81,11 +88,37 @@ class ExecutingAgent(Agent):
81
88
  agent_input_copy['input'] = task
82
89
  planner: Planner = PlannerManager().get_instance_obj(self.agent_model.plan.get('planner').get('name'))
83
90
  futures.append(
84
- self.executor.submit(planner.invoke, self.agent_model, agent_input_copy,
91
+ self.executor.submit(self.run_in_executor, planner, self.agent_model, agent_input_copy,
85
92
  self.process_intput_object(input_object, task, planner.input_key)))
86
93
  wait(futures, return_when=ALL_COMPLETED)
87
94
  return {'futures': futures}
88
95
 
96
+ def run_in_executor(self, planner: Planner, agent_model: AgentModel, planner_input: dict,
97
+ input_object: InputObject) -> dict:
98
+ """The execution function of the thread pool.
99
+
100
+ Args:
101
+ planner(Planner): The planner object.
102
+ agent_model (AgentModel): The agent model object.
103
+ planner_input (dict): The planner input dict.
104
+ input_object (InputObject): The input parameters passed by the user.
105
+ Returns:
106
+ dict: The planner execution result.
107
+ """
108
+ context_tokens = {}
109
+ try:
110
+ # pass the framework context into the thread.
111
+ for var_name, var_value in self._context_values.items():
112
+ token = FrameworkContextManager().set_context(var_name, var_value)
113
+ context_tokens[var_name] = token
114
+ # invoke planner
115
+ res = planner.invoke(agent_model, planner_input, input_object)
116
+ return res
117
+ finally:
118
+ # clear the framework context.
119
+ for var_name, token in context_tokens.items():
120
+ FrameworkContextManager().reset_context(var_name, token)
121
+
89
122
  def process_intput_object(self, input_object: InputObject, subtask: str, planner_input_key: str) -> InputObject:
90
123
  """Process input object for the executing agent.
91
124
 
@@ -24,7 +24,7 @@ class Message(BaseModel):
24
24
  """
25
25
 
26
26
  type: Optional[str] = None
27
- content: Union[str, List[Union[str, Dict]]] = None
27
+ content: Optional[Union[str, List[Union[str, Dict]]]] = None
28
28
 
29
29
  def as_langchain(self):
30
30
  """Convert the agentUniverse(aU) message class to the langchain message class."""
@@ -6,6 +6,7 @@
6
6
  # @FileName: peer_planner.py
7
7
  """Peer planner module."""
8
8
  from agentuniverse.agent.action.tool.tool_manager import ToolManager
9
+ from agentuniverse.agent.agent import Agent
9
10
  from agentuniverse.agent.agent_manager import AgentManager
10
11
  from agentuniverse.agent.agent_model import AgentModel
11
12
  from agentuniverse.agent.input_object import InputObject
@@ -42,7 +43,7 @@ class PeerPlanner(Planner):
42
43
  """
43
44
  planner_config = agent_model.plan.get('planner')
44
45
  sub_agents = self.generate_sub_agents(planner_config)
45
- return self.agents_run(agent_model, sub_agents, planner_config, planner_input, input_object)
46
+ return self.agents_run(sub_agents, planner_config, planner_input, input_object)
46
47
 
47
48
  @staticmethod
48
49
  def generate_sub_agents(planner_config: dict) -> dict:
@@ -79,7 +80,7 @@ class PeerPlanner(Planner):
79
80
  elif context:
80
81
  input_object.add_data('expert_framework', context)
81
82
 
82
- def agents_run(self, agent_mode: AgentModel, agents: dict, planner_config: dict, agent_input: dict,
83
+ def agents_run(self, agents: dict, planner_config: dict, agent_input: dict,
83
84
  input_object: InputObject) -> dict:
84
85
  """Planner agents run.
85
86
 
@@ -105,10 +106,10 @@ class PeerPlanner(Planner):
105
106
 
106
107
  self.build_expert_framework(planner_config, input_object)
107
108
 
108
- planningAgent = agents.get('planning')
109
- executingAgent = agents.get('executing')
110
- expressingAgent = agents.get('expressing')
111
- reviewingAgent = agents.get('reviewing')
109
+ planningAgent: Agent = agents.get('planning')
110
+ executingAgent: Agent = agents.get('executing')
111
+ expressingAgent: Agent = agents.get('expressing')
112
+ reviewingAgent: Agent = agents.get('reviewing')
112
113
 
113
114
  for _ in range(retry_count):
114
115
  LOGGER.info(f"Starting peer agents, retry_count is {_ + 1}.")
@@ -126,10 +127,13 @@ class PeerPlanner(Planner):
126
127
  for index, one_framework in enumerate(planning_result.get_data('framework')):
127
128
  logger_info += f"[{index + 1}] {one_framework} \n"
128
129
  LOGGER.info(logger_info)
129
- self.stream_output(input_object, {"data": {
130
- 'output': planning_result.to_dict(),
131
- "agent_info": agent_mode.info
132
- }, "type": "planning"})
130
+
131
+ # add planning agent intermediate steps
132
+ if planningAgent:
133
+ self.stream_output(input_object, {"data": {
134
+ 'output': planning_result.get_data('framework'),
135
+ "agent_info": planningAgent.agent_model.info
136
+ }, "type": "planning"})
133
137
 
134
138
  if not executing_result or jump_step in ["planning", "executing"]:
135
139
  if not executingAgent:
@@ -148,10 +152,13 @@ class PeerPlanner(Planner):
148
152
  one_exec_log_info += f"[{index + 1}] output: {one_exec_res['output']}\n"
149
153
  logger_info += one_exec_log_info
150
154
  LOGGER.info(logger_info)
151
- self.stream_output(input_object, {"data": {
152
- 'output': executing_result.to_dict(),
153
- "agent_info": agent_mode.info
154
- }, "type": "executing"})
155
+
156
+ # add executing agent intermediate steps
157
+ if executingAgent:
158
+ self.stream_output(input_object, {"data": {
159
+ 'output': executing_result.get_data('executing_result'),
160
+ "agent_info": executingAgent.agent_model.info
161
+ }, "type": "executing"})
155
162
 
156
163
  if not expressing_result or jump_step in ["planning", "executing", "expressing"]:
157
164
  if not expressingAgent:
@@ -166,10 +173,13 @@ class PeerPlanner(Planner):
166
173
  logger_info = f"\nExpressing agent execution result is :\n"
167
174
  logger_info += f"{expressing_result.get_data('output')}"
168
175
  LOGGER.info(logger_info)
169
- self.stream_output(input_object, {"data": {
170
- 'output': expressing_result.get_data('output'),
171
- "agent_info": agent_mode.info
172
- }, "type": "expressing"})
176
+
177
+ # add expressing agent intermediate steps
178
+ if expressingAgent:
179
+ self.stream_output(input_object, {"data": {
180
+ 'output': expressing_result.get_data('output'),
181
+ "agent_info": expressingAgent.agent_model.info
182
+ }, "type": "expressing"})
173
183
 
174
184
  if not reviewing_result or jump_step in ["planning", "executing", "expressing", "reviewing"]:
175
185
  if not reviewingAgent:
@@ -193,7 +203,14 @@ class PeerPlanner(Planner):
193
203
  reviewing_info_str = f"review suggestion: {reviewing_result.get_data('suggestion')} \n"
194
204
  reviewing_info_str += f"review score: {reviewing_result.get_data('score')} \n"
195
205
  LOGGER.info(logger_info + reviewing_info_str)
196
- self.stream_output(input_object, {"data": reviewing_result.to_dict(), "type": "reviewing"})
206
+
207
+ # add reviewing agent intermediate steps
208
+ self.stream_output(input_object,
209
+ {"data": {
210
+ 'output': reviewing_result.get_data('suggestion'),
211
+ "agent_info": reviewingAgent.agent_model.info
212
+ }, "type": "reviewing"})
213
+
197
214
  if reviewing_result.get_data('score') and reviewing_result.get_data('score') >= eval_threshold:
198
215
  loopResults.append({
199
216
  "planning_result": planning_result,
@@ -76,6 +76,10 @@ class Planner(ComponentBase):
76
76
  """
77
77
  chat_history: list = planner_input.get('chat_history')
78
78
  memory_name = agent_model.memory.get('name')
79
+ memory: ChatMemory = MemoryManager().get_instance_obj(component_instance_name=memory_name)
80
+ if memory is None:
81
+ return None
82
+
79
83
  llm_model = agent_model.memory.get('llm_model') or dict()
80
84
  llm_name = llm_model.get('name') or agent_model.profile.get('llm_model').get('name')
81
85
 
@@ -86,10 +90,6 @@ class Planner(ComponentBase):
86
90
  params['llm'] = llm
87
91
  params['input_key'] = self.input_key
88
92
  params['output_key'] = self.output_key
89
-
90
- memory: ChatMemory = MemoryManager().get_instance_obj(component_instance_name=memory_name)
91
- if memory is None:
92
- return None
93
93
  return memory.set_by_agent_model(**params)
94
94
 
95
95
  def run_all_actions(self, agent_model: AgentModel, planner_input: dict, input_object: InputObject):
@@ -118,8 +118,8 @@ class Planner(ComponentBase):
118
118
  knowledge: Knowledge = KnowledgeManager().get_instance_obj(knowledge_name)
119
119
  if knowledge is None:
120
120
  continue
121
- knowledge_res: List[Document] = knowledge.store.query(
122
- Query(query_str=input_object.get_data(self.input_key), similarity_top_k=2), **input_object.to_dict())
121
+ knowledge_res: List[Document] = knowledge.query_knowledge(query_str=input_object.get_data(self.input_key),
122
+ **input_object.to_dict())
123
123
  for document in knowledge_res:
124
124
  action_result.append(document.text)
125
125
 
@@ -185,8 +185,9 @@ class Planner(ComponentBase):
185
185
  return
186
186
  output_stream.put_nowait(data)
187
187
 
188
- def invoke_chain(self, agent_model: AgentModel, chain: RunnableSerializable[Any, str], planner_input: dict, chat_history,
189
- input_object: InputObject):
188
+ def invoke_chain(self, agent_model: AgentModel, chain: RunnableSerializable[Any, str], planner_input: dict,
189
+ chat_history,
190
+ input_object: InputObject):
190
191
 
191
192
  if not input_object.get_data('output_stream'):
192
193
  res = chain.invoke(input=planner_input, config={"configurable": {"session_id": "unused"}})
@@ -202,4 +203,3 @@ class Planner(ComponentBase):
202
203
  })
203
204
  result.append(token)
204
205
  return "".join(result)
205
-
@@ -7,10 +7,12 @@
7
7
  # @FileName: stream_callback.py
8
8
 
9
9
  import asyncio
10
- from typing import Optional, Dict, Any
10
+ from typing import Optional, Dict, Any, Union
11
+ from uuid import UUID
11
12
 
12
13
  from langchain_core.agents import AgentAction, AgentFinish
13
14
  from langchain_core.callbacks import BaseCallbackHandler
15
+ from langchain_core.outputs import GenerationChunk, ChatGenerationChunk
14
16
 
15
17
 
16
18
  class StreamOutPutCallbackHandler(BaseCallbackHandler):
@@ -43,6 +45,24 @@ class StreamOutPutCallbackHandler(BaseCallbackHandler):
43
45
  }
44
46
  })
45
47
 
48
+ def on_llm_new_token(
49
+ self,
50
+ token: str,
51
+ *,
52
+ chunk: Optional[Union[GenerationChunk, ChatGenerationChunk]] = None,
53
+ run_id: UUID,
54
+ parent_run_id: Optional[UUID] = None,
55
+ **kwargs: Any,
56
+ ) -> Any:
57
+ # add token chunk to the queue.
58
+ self.queueStream.put_nowait({
59
+ "type": "token",
60
+ "data": {
61
+ "chunk": chunk.text,
62
+ "agent_info": self.agent_info
63
+ }
64
+ })
65
+
46
66
  def on_tool_end(
47
67
  self,
48
68
  output: str,
@@ -5,7 +5,6 @@
5
5
  # @Author : fanen.lhy
6
6
  # @Email : fanen.lhy@antgroup.com
7
7
  # @FileName: request_library.py
8
-
9
8
  import datetime
10
9
 
11
10
  from sqlalchemy import JSON, Integer, String, DateTime, Text, Column
@@ -65,7 +64,8 @@ class RequestLibrary:
65
64
  name="__system_db__",
66
65
  db_wrapper_configer=_configer
67
66
  )
68
- SQLDBWrapperManager().register("__system_db__", self.sqldb_wrapper)
67
+ SQLDBWrapperManager().register(self.sqldb_wrapper.get_instance_code(),
68
+ self.sqldb_wrapper)
69
69
 
70
70
  def __init_request_table(self):
71
71
  with self.sqldb_wrapper.sql_database._engine.connect() as conn:
@@ -76,8 +76,7 @@ class RequestLibrary:
76
76
  if not self.session:
77
77
  self.__init_request_table()
78
78
  self.session = self.sqldb_wrapper.get_session()
79
- return self.session
80
-
79
+ return self.session()
81
80
 
82
81
  def query_request_by_request_id(self, request_id: str) -> RequestDO | None:
83
82
  """Get a RequestDO with given request_id.
@@ -5,7 +5,6 @@
5
5
  # @Author : fanen.lhy
6
6
  # @Email : fanen.lhy@antgroup.com
7
7
  # @FileName: request_task.py
8
-
9
8
  import enum
10
9
  from enum import Enum
11
10
  import json
@@ -20,6 +19,7 @@ from .dal.request_library import RequestLibrary
20
19
  from .dal.entity.request_do import RequestDO
21
20
  from .thread_with_result import ThreadWithReturnValue
22
21
  from agentuniverse.base.util.logging.logging_util import LOGGER
22
+ from ...agent.output_object import OutputObject
23
23
 
24
24
  EOF_SIGNAL = '{"type": "EOF"}'
25
25
 
@@ -75,6 +75,8 @@ class RequestTask:
75
75
  return
76
76
  try:
77
77
  result = self.thread.result()
78
+ if isinstance(result, OutputObject):
79
+ result = result.to_dict()
78
80
  yield "data:" + json.dumps({"result": result},
79
81
  ensure_ascii=False) + "\n\n "
80
82
  except Exception as e:
@@ -122,6 +124,7 @@ class RequestTask:
122
124
  def stream_run(self):
123
125
  """Run the service in a separate thread and yield result stream."""
124
126
  self.kwargs['output_stream'] = self.queue
127
+
125
128
  self.thread = ThreadWithReturnValue(target=self.func,
126
129
  kwargs=self.kwargs)
127
130
  self.thread.start()
@@ -8,9 +8,12 @@
8
8
 
9
9
  from threading import Thread
10
10
 
11
+ from agentuniverse.base.context.framework_context_manager import FrameworkContextManager
12
+
11
13
 
12
14
  class ThreadWithReturnValue(Thread):
13
15
  """A thread can save the target func exec result."""
16
+
14
17
  def __init__(self, group=None, target=None, name=None,
15
18
  args=(), kwargs=None):
16
19
  super().__init__(group, target, name, args, kwargs)
@@ -22,10 +25,17 @@ class ThreadWithReturnValue(Thread):
22
25
  self.target = target
23
26
  self._return = None
24
27
  self.error = None
28
+ self._context_values: dict = FrameworkContextManager().get_all_contexts()
25
29
 
26
30
  def run(self):
27
31
  """Run the target func and save result in _return."""
28
32
  if self.target is not None:
33
+ context_tokens = {}
34
+ # set the context values in the thread
35
+ for var_name, var_value in self._context_values.items():
36
+ token = FrameworkContextManager().set_context(var_name, var_value)
37
+ context_tokens[var_name] = token
38
+
29
39
  try:
30
40
  self._return = self.target(*self.args, **self.kwargs)
31
41
  except Exception as e:
@@ -33,6 +43,9 @@ class ThreadWithReturnValue(Thread):
33
43
  finally:
34
44
  if 'output_stream' in self.kwargs:
35
45
  self.kwargs['output_stream'].put('{"type": "EOF"}')
46
+ # finally, clear the context values in the thread
47
+ for var_name, token in context_tokens.items():
48
+ FrameworkContextManager().reset_context(var_name, token)
36
49
 
37
50
  def result(self):
38
51
  """Wait for target func finished, then return the result or raise an
@@ -13,10 +13,13 @@ import json
13
13
  from flask import request, make_response, jsonify
14
14
 
15
15
  from ..service_instance import ServiceInstance
16
+ from ...agent.agent import Agent
17
+ from ...agent.agent_manager import AgentManager
16
18
 
17
19
 
18
20
  def request_param(func):
19
21
  """An annotation used to parse the flask request params."""
22
+
20
23
  def wrapper(*args, **kwargs):
21
24
  if request.method == "GET":
22
25
  req_data = request.args.to_dict()
@@ -65,6 +68,23 @@ def service_run_queue(service_id, **kwargs):
65
68
  stream.put_nowait('{"type": "EOF"}')
66
69
 
67
70
 
71
+ def agent_run_queue(agent_id, **kwargs):
72
+ """
73
+ The func used in a separate thread to run an agent, and the result will be saved in a queue if provided.
74
+ Args:
75
+ agent_id: The agent id
76
+ **kwargs: Arbitrary keyword arguments.
77
+ """
78
+ stream: queue.Queue = kwargs.get('output_stream')
79
+ try:
80
+ agent: Agent = AgentManager().get_instance_obj(agent_id)
81
+ res = agent.run(**kwargs)
82
+ return res
83
+ finally:
84
+ if stream:
85
+ stream.put_nowait('{"type": "EOF"}')
86
+
87
+
68
88
  def make_standard_response(success: bool,
69
89
  result=None,
70
90
  message: str = None,
@@ -142,16 +142,16 @@ class AgentUniverse(object):
142
142
  for component_enum, package_list in component_package_map.items():
143
143
  if not package_list:
144
144
  continue
145
- component_configer_list = self.__scan(package_list, ConfigTypeEnum.YAML, component_enum)
145
+ component_configer_list = self.scan(package_list, ConfigTypeEnum.YAML, component_enum)
146
146
  component_configer_list_map[component_enum] = component_configer_list
147
147
 
148
148
  for component_enum, component_configer_list in component_configer_list_map.items():
149
149
  self.__register(component_enum, component_configer_list)
150
150
 
151
- def __scan(self,
152
- package_list: [str],
153
- config_type_enum: ConfigTypeEnum,
154
- component_enum: ComponentEnum) -> list:
151
+ def scan(self,
152
+ package_list: [str],
153
+ config_type_enum: ConfigTypeEnum,
154
+ component_enum: ComponentEnum) -> list:
155
155
  """Scan the component directory and return certain component configer list.
156
156
 
157
157
  Args:
@@ -191,6 +191,7 @@ class AgentUniverse(object):
191
191
  component_instance: ComponentBase = component_clz().initialize_by_component_configer(configer_instance)
192
192
  if component_instance is None:
193
193
  continue
194
+ component_instance.component_config_path = component_configer.configer.path
194
195
  component_manager_clz().register(component_instance.get_instance_code(), component_instance)
195
196
 
196
197
  def __package_name_to_path(self, package_name: str) -> str: