aiecs 1.0.1__py3-none-any.whl → 1.7.17__py3-none-any.whl

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.

Potentially problematic release.


This version of aiecs might be problematic. Click here for more details.

Files changed (340) hide show
  1. aiecs/__init__.py +13 -16
  2. aiecs/__main__.py +7 -7
  3. aiecs/aiecs_client.py +269 -75
  4. aiecs/application/executors/operation_executor.py +79 -54
  5. aiecs/application/knowledge_graph/__init__.py +7 -0
  6. aiecs/application/knowledge_graph/builder/__init__.py +37 -0
  7. aiecs/application/knowledge_graph/builder/data_quality.py +302 -0
  8. aiecs/application/knowledge_graph/builder/data_reshaping.py +293 -0
  9. aiecs/application/knowledge_graph/builder/document_builder.py +369 -0
  10. aiecs/application/knowledge_graph/builder/graph_builder.py +490 -0
  11. aiecs/application/knowledge_graph/builder/import_optimizer.py +396 -0
  12. aiecs/application/knowledge_graph/builder/schema_inference.py +462 -0
  13. aiecs/application/knowledge_graph/builder/schema_mapping.py +563 -0
  14. aiecs/application/knowledge_graph/builder/structured_pipeline.py +1384 -0
  15. aiecs/application/knowledge_graph/builder/text_chunker.py +317 -0
  16. aiecs/application/knowledge_graph/extractors/__init__.py +27 -0
  17. aiecs/application/knowledge_graph/extractors/base.py +98 -0
  18. aiecs/application/knowledge_graph/extractors/llm_entity_extractor.py +422 -0
  19. aiecs/application/knowledge_graph/extractors/llm_relation_extractor.py +347 -0
  20. aiecs/application/knowledge_graph/extractors/ner_entity_extractor.py +241 -0
  21. aiecs/application/knowledge_graph/fusion/__init__.py +78 -0
  22. aiecs/application/knowledge_graph/fusion/ab_testing.py +395 -0
  23. aiecs/application/knowledge_graph/fusion/abbreviation_expander.py +327 -0
  24. aiecs/application/knowledge_graph/fusion/alias_index.py +597 -0
  25. aiecs/application/knowledge_graph/fusion/alias_matcher.py +384 -0
  26. aiecs/application/knowledge_graph/fusion/cache_coordinator.py +343 -0
  27. aiecs/application/knowledge_graph/fusion/entity_deduplicator.py +433 -0
  28. aiecs/application/knowledge_graph/fusion/entity_linker.py +511 -0
  29. aiecs/application/knowledge_graph/fusion/evaluation_dataset.py +240 -0
  30. aiecs/application/knowledge_graph/fusion/knowledge_fusion.py +632 -0
  31. aiecs/application/knowledge_graph/fusion/matching_config.py +489 -0
  32. aiecs/application/knowledge_graph/fusion/name_normalizer.py +352 -0
  33. aiecs/application/knowledge_graph/fusion/relation_deduplicator.py +183 -0
  34. aiecs/application/knowledge_graph/fusion/semantic_name_matcher.py +464 -0
  35. aiecs/application/knowledge_graph/fusion/similarity_pipeline.py +534 -0
  36. aiecs/application/knowledge_graph/pattern_matching/__init__.py +21 -0
  37. aiecs/application/knowledge_graph/pattern_matching/pattern_matcher.py +342 -0
  38. aiecs/application/knowledge_graph/pattern_matching/query_executor.py +366 -0
  39. aiecs/application/knowledge_graph/profiling/__init__.py +12 -0
  40. aiecs/application/knowledge_graph/profiling/query_plan_visualizer.py +195 -0
  41. aiecs/application/knowledge_graph/profiling/query_profiler.py +223 -0
  42. aiecs/application/knowledge_graph/reasoning/__init__.py +27 -0
  43. aiecs/application/knowledge_graph/reasoning/evidence_synthesis.py +341 -0
  44. aiecs/application/knowledge_graph/reasoning/inference_engine.py +500 -0
  45. aiecs/application/knowledge_graph/reasoning/logic_form_parser.py +163 -0
  46. aiecs/application/knowledge_graph/reasoning/logic_parser/__init__.py +79 -0
  47. aiecs/application/knowledge_graph/reasoning/logic_parser/ast_builder.py +513 -0
  48. aiecs/application/knowledge_graph/reasoning/logic_parser/ast_nodes.py +913 -0
  49. aiecs/application/knowledge_graph/reasoning/logic_parser/ast_validator.py +866 -0
  50. aiecs/application/knowledge_graph/reasoning/logic_parser/error_handler.py +475 -0
  51. aiecs/application/knowledge_graph/reasoning/logic_parser/parser.py +396 -0
  52. aiecs/application/knowledge_graph/reasoning/logic_parser/query_context.py +208 -0
  53. aiecs/application/knowledge_graph/reasoning/logic_query_integration.py +170 -0
  54. aiecs/application/knowledge_graph/reasoning/query_planner.py +855 -0
  55. aiecs/application/knowledge_graph/reasoning/reasoning_engine.py +518 -0
  56. aiecs/application/knowledge_graph/retrieval/__init__.py +27 -0
  57. aiecs/application/knowledge_graph/retrieval/query_intent_classifier.py +211 -0
  58. aiecs/application/knowledge_graph/retrieval/retrieval_strategies.py +592 -0
  59. aiecs/application/knowledge_graph/retrieval/strategy_types.py +23 -0
  60. aiecs/application/knowledge_graph/search/__init__.py +59 -0
  61. aiecs/application/knowledge_graph/search/hybrid_search.py +457 -0
  62. aiecs/application/knowledge_graph/search/reranker.py +293 -0
  63. aiecs/application/knowledge_graph/search/reranker_strategies.py +535 -0
  64. aiecs/application/knowledge_graph/search/text_similarity.py +392 -0
  65. aiecs/application/knowledge_graph/traversal/__init__.py +15 -0
  66. aiecs/application/knowledge_graph/traversal/enhanced_traversal.py +305 -0
  67. aiecs/application/knowledge_graph/traversal/path_scorer.py +271 -0
  68. aiecs/application/knowledge_graph/validators/__init__.py +13 -0
  69. aiecs/application/knowledge_graph/validators/relation_validator.py +239 -0
  70. aiecs/application/knowledge_graph/visualization/__init__.py +11 -0
  71. aiecs/application/knowledge_graph/visualization/graph_visualizer.py +313 -0
  72. aiecs/common/__init__.py +9 -0
  73. aiecs/common/knowledge_graph/__init__.py +17 -0
  74. aiecs/common/knowledge_graph/runnable.py +471 -0
  75. aiecs/config/__init__.py +20 -5
  76. aiecs/config/config.py +762 -31
  77. aiecs/config/graph_config.py +131 -0
  78. aiecs/config/tool_config.py +435 -0
  79. aiecs/core/__init__.py +29 -13
  80. aiecs/core/interface/__init__.py +2 -2
  81. aiecs/core/interface/execution_interface.py +22 -22
  82. aiecs/core/interface/storage_interface.py +37 -88
  83. aiecs/core/registry/__init__.py +31 -0
  84. aiecs/core/registry/service_registry.py +92 -0
  85. aiecs/domain/__init__.py +270 -1
  86. aiecs/domain/agent/__init__.py +191 -0
  87. aiecs/domain/agent/base_agent.py +3949 -0
  88. aiecs/domain/agent/exceptions.py +99 -0
  89. aiecs/domain/agent/graph_aware_mixin.py +569 -0
  90. aiecs/domain/agent/hybrid_agent.py +1731 -0
  91. aiecs/domain/agent/integration/__init__.py +29 -0
  92. aiecs/domain/agent/integration/context_compressor.py +216 -0
  93. aiecs/domain/agent/integration/context_engine_adapter.py +587 -0
  94. aiecs/domain/agent/integration/protocols.py +281 -0
  95. aiecs/domain/agent/integration/retry_policy.py +218 -0
  96. aiecs/domain/agent/integration/role_config.py +213 -0
  97. aiecs/domain/agent/knowledge_aware_agent.py +1892 -0
  98. aiecs/domain/agent/lifecycle.py +291 -0
  99. aiecs/domain/agent/llm_agent.py +692 -0
  100. aiecs/domain/agent/memory/__init__.py +12 -0
  101. aiecs/domain/agent/memory/conversation.py +1124 -0
  102. aiecs/domain/agent/migration/__init__.py +14 -0
  103. aiecs/domain/agent/migration/conversion.py +163 -0
  104. aiecs/domain/agent/migration/legacy_wrapper.py +86 -0
  105. aiecs/domain/agent/models.py +894 -0
  106. aiecs/domain/agent/observability.py +479 -0
  107. aiecs/domain/agent/persistence.py +449 -0
  108. aiecs/domain/agent/prompts/__init__.py +29 -0
  109. aiecs/domain/agent/prompts/builder.py +159 -0
  110. aiecs/domain/agent/prompts/formatters.py +187 -0
  111. aiecs/domain/agent/prompts/template.py +255 -0
  112. aiecs/domain/agent/registry.py +253 -0
  113. aiecs/domain/agent/tool_agent.py +444 -0
  114. aiecs/domain/agent/tools/__init__.py +15 -0
  115. aiecs/domain/agent/tools/schema_generator.py +377 -0
  116. aiecs/domain/community/__init__.py +155 -0
  117. aiecs/domain/community/agent_adapter.py +469 -0
  118. aiecs/domain/community/analytics.py +432 -0
  119. aiecs/domain/community/collaborative_workflow.py +648 -0
  120. aiecs/domain/community/communication_hub.py +634 -0
  121. aiecs/domain/community/community_builder.py +320 -0
  122. aiecs/domain/community/community_integration.py +796 -0
  123. aiecs/domain/community/community_manager.py +803 -0
  124. aiecs/domain/community/decision_engine.py +849 -0
  125. aiecs/domain/community/exceptions.py +231 -0
  126. aiecs/domain/community/models/__init__.py +33 -0
  127. aiecs/domain/community/models/community_models.py +234 -0
  128. aiecs/domain/community/resource_manager.py +461 -0
  129. aiecs/domain/community/shared_context_manager.py +589 -0
  130. aiecs/domain/context/__init__.py +40 -10
  131. aiecs/domain/context/context_engine.py +1910 -0
  132. aiecs/domain/context/conversation_models.py +87 -53
  133. aiecs/domain/context/graph_memory.py +582 -0
  134. aiecs/domain/execution/model.py +12 -4
  135. aiecs/domain/knowledge_graph/__init__.py +19 -0
  136. aiecs/domain/knowledge_graph/models/__init__.py +52 -0
  137. aiecs/domain/knowledge_graph/models/entity.py +148 -0
  138. aiecs/domain/knowledge_graph/models/evidence.py +178 -0
  139. aiecs/domain/knowledge_graph/models/inference_rule.py +184 -0
  140. aiecs/domain/knowledge_graph/models/path.py +171 -0
  141. aiecs/domain/knowledge_graph/models/path_pattern.py +171 -0
  142. aiecs/domain/knowledge_graph/models/query.py +261 -0
  143. aiecs/domain/knowledge_graph/models/query_plan.py +181 -0
  144. aiecs/domain/knowledge_graph/models/relation.py +202 -0
  145. aiecs/domain/knowledge_graph/schema/__init__.py +23 -0
  146. aiecs/domain/knowledge_graph/schema/entity_type.py +131 -0
  147. aiecs/domain/knowledge_graph/schema/graph_schema.py +253 -0
  148. aiecs/domain/knowledge_graph/schema/property_schema.py +143 -0
  149. aiecs/domain/knowledge_graph/schema/relation_type.py +163 -0
  150. aiecs/domain/knowledge_graph/schema/schema_manager.py +691 -0
  151. aiecs/domain/knowledge_graph/schema/type_enums.py +209 -0
  152. aiecs/domain/task/dsl_processor.py +172 -56
  153. aiecs/domain/task/model.py +20 -8
  154. aiecs/domain/task/task_context.py +27 -24
  155. aiecs/infrastructure/__init__.py +0 -2
  156. aiecs/infrastructure/graph_storage/__init__.py +11 -0
  157. aiecs/infrastructure/graph_storage/base.py +837 -0
  158. aiecs/infrastructure/graph_storage/batch_operations.py +458 -0
  159. aiecs/infrastructure/graph_storage/cache.py +424 -0
  160. aiecs/infrastructure/graph_storage/distributed.py +223 -0
  161. aiecs/infrastructure/graph_storage/error_handling.py +380 -0
  162. aiecs/infrastructure/graph_storage/graceful_degradation.py +294 -0
  163. aiecs/infrastructure/graph_storage/health_checks.py +378 -0
  164. aiecs/infrastructure/graph_storage/in_memory.py +1197 -0
  165. aiecs/infrastructure/graph_storage/index_optimization.py +446 -0
  166. aiecs/infrastructure/graph_storage/lazy_loading.py +431 -0
  167. aiecs/infrastructure/graph_storage/metrics.py +344 -0
  168. aiecs/infrastructure/graph_storage/migration.py +400 -0
  169. aiecs/infrastructure/graph_storage/pagination.py +483 -0
  170. aiecs/infrastructure/graph_storage/performance_monitoring.py +456 -0
  171. aiecs/infrastructure/graph_storage/postgres.py +1563 -0
  172. aiecs/infrastructure/graph_storage/property_storage.py +353 -0
  173. aiecs/infrastructure/graph_storage/protocols.py +76 -0
  174. aiecs/infrastructure/graph_storage/query_optimizer.py +642 -0
  175. aiecs/infrastructure/graph_storage/schema_cache.py +290 -0
  176. aiecs/infrastructure/graph_storage/sqlite.py +1373 -0
  177. aiecs/infrastructure/graph_storage/streaming.py +487 -0
  178. aiecs/infrastructure/graph_storage/tenant.py +412 -0
  179. aiecs/infrastructure/messaging/celery_task_manager.py +92 -54
  180. aiecs/infrastructure/messaging/websocket_manager.py +51 -35
  181. aiecs/infrastructure/monitoring/__init__.py +22 -0
  182. aiecs/infrastructure/monitoring/executor_metrics.py +45 -11
  183. aiecs/infrastructure/monitoring/global_metrics_manager.py +212 -0
  184. aiecs/infrastructure/monitoring/structured_logger.py +3 -7
  185. aiecs/infrastructure/monitoring/tracing_manager.py +63 -35
  186. aiecs/infrastructure/persistence/__init__.py +14 -1
  187. aiecs/infrastructure/persistence/context_engine_client.py +184 -0
  188. aiecs/infrastructure/persistence/database_manager.py +67 -43
  189. aiecs/infrastructure/persistence/file_storage.py +180 -103
  190. aiecs/infrastructure/persistence/redis_client.py +74 -21
  191. aiecs/llm/__init__.py +73 -25
  192. aiecs/llm/callbacks/__init__.py +11 -0
  193. aiecs/llm/{custom_callbacks.py → callbacks/custom_callbacks.py} +26 -19
  194. aiecs/llm/client_factory.py +230 -37
  195. aiecs/llm/client_resolver.py +155 -0
  196. aiecs/llm/clients/__init__.py +38 -0
  197. aiecs/llm/clients/base_client.py +328 -0
  198. aiecs/llm/clients/google_function_calling_mixin.py +415 -0
  199. aiecs/llm/clients/googleai_client.py +314 -0
  200. aiecs/llm/clients/openai_client.py +158 -0
  201. aiecs/llm/clients/openai_compatible_mixin.py +367 -0
  202. aiecs/llm/clients/vertex_client.py +1186 -0
  203. aiecs/llm/clients/xai_client.py +201 -0
  204. aiecs/llm/config/__init__.py +51 -0
  205. aiecs/llm/config/config_loader.py +272 -0
  206. aiecs/llm/config/config_validator.py +206 -0
  207. aiecs/llm/config/model_config.py +143 -0
  208. aiecs/llm/protocols.py +149 -0
  209. aiecs/llm/utils/__init__.py +10 -0
  210. aiecs/llm/utils/validate_config.py +89 -0
  211. aiecs/main.py +140 -121
  212. aiecs/scripts/aid/VERSION_MANAGEMENT.md +138 -0
  213. aiecs/scripts/aid/__init__.py +19 -0
  214. aiecs/scripts/aid/module_checker.py +499 -0
  215. aiecs/scripts/aid/version_manager.py +235 -0
  216. aiecs/scripts/{DEPENDENCY_SYSTEM_SUMMARY.md → dependance_check/DEPENDENCY_SYSTEM_SUMMARY.md} +1 -0
  217. aiecs/scripts/{README_DEPENDENCY_CHECKER.md → dependance_check/README_DEPENDENCY_CHECKER.md} +1 -0
  218. aiecs/scripts/dependance_check/__init__.py +15 -0
  219. aiecs/scripts/dependance_check/dependency_checker.py +1835 -0
  220. aiecs/scripts/{dependency_fixer.py → dependance_check/dependency_fixer.py} +192 -90
  221. aiecs/scripts/{download_nlp_data.py → dependance_check/download_nlp_data.py} +203 -71
  222. aiecs/scripts/dependance_patch/__init__.py +7 -0
  223. aiecs/scripts/dependance_patch/fix_weasel/__init__.py +11 -0
  224. aiecs/scripts/{fix_weasel_validator.py → dependance_patch/fix_weasel/fix_weasel_validator.py} +21 -14
  225. aiecs/scripts/{patch_weasel_library.sh → dependance_patch/fix_weasel/patch_weasel_library.sh} +1 -1
  226. aiecs/scripts/knowledge_graph/__init__.py +3 -0
  227. aiecs/scripts/knowledge_graph/run_threshold_experiments.py +212 -0
  228. aiecs/scripts/migrations/multi_tenancy/README.md +142 -0
  229. aiecs/scripts/tools_develop/README.md +671 -0
  230. aiecs/scripts/tools_develop/README_CONFIG_CHECKER.md +273 -0
  231. aiecs/scripts/tools_develop/TOOLS_CONFIG_GUIDE.md +1287 -0
  232. aiecs/scripts/tools_develop/TOOL_AUTO_DISCOVERY.md +234 -0
  233. aiecs/scripts/tools_develop/__init__.py +21 -0
  234. aiecs/scripts/tools_develop/check_all_tools_config.py +548 -0
  235. aiecs/scripts/tools_develop/check_type_annotations.py +257 -0
  236. aiecs/scripts/tools_develop/pre-commit-schema-coverage.sh +66 -0
  237. aiecs/scripts/tools_develop/schema_coverage.py +511 -0
  238. aiecs/scripts/tools_develop/validate_tool_schemas.py +475 -0
  239. aiecs/scripts/tools_develop/verify_executor_config_fix.py +98 -0
  240. aiecs/scripts/tools_develop/verify_tools.py +352 -0
  241. aiecs/tasks/__init__.py +0 -1
  242. aiecs/tasks/worker.py +115 -47
  243. aiecs/tools/__init__.py +194 -72
  244. aiecs/tools/apisource/__init__.py +99 -0
  245. aiecs/tools/apisource/intelligence/__init__.py +19 -0
  246. aiecs/tools/apisource/intelligence/data_fusion.py +632 -0
  247. aiecs/tools/apisource/intelligence/query_analyzer.py +417 -0
  248. aiecs/tools/apisource/intelligence/search_enhancer.py +385 -0
  249. aiecs/tools/apisource/monitoring/__init__.py +9 -0
  250. aiecs/tools/apisource/monitoring/metrics.py +330 -0
  251. aiecs/tools/apisource/providers/__init__.py +112 -0
  252. aiecs/tools/apisource/providers/base.py +671 -0
  253. aiecs/tools/apisource/providers/census.py +397 -0
  254. aiecs/tools/apisource/providers/fred.py +535 -0
  255. aiecs/tools/apisource/providers/newsapi.py +409 -0
  256. aiecs/tools/apisource/providers/worldbank.py +352 -0
  257. aiecs/tools/apisource/reliability/__init__.py +12 -0
  258. aiecs/tools/apisource/reliability/error_handler.py +363 -0
  259. aiecs/tools/apisource/reliability/fallback_strategy.py +376 -0
  260. aiecs/tools/apisource/tool.py +832 -0
  261. aiecs/tools/apisource/utils/__init__.py +9 -0
  262. aiecs/tools/apisource/utils/validators.py +334 -0
  263. aiecs/tools/base_tool.py +415 -21
  264. aiecs/tools/docs/__init__.py +121 -0
  265. aiecs/tools/docs/ai_document_orchestrator.py +607 -0
  266. aiecs/tools/docs/ai_document_writer_orchestrator.py +2350 -0
  267. aiecs/tools/docs/content_insertion_tool.py +1320 -0
  268. aiecs/tools/docs/document_creator_tool.py +1464 -0
  269. aiecs/tools/docs/document_layout_tool.py +1160 -0
  270. aiecs/tools/docs/document_parser_tool.py +1016 -0
  271. aiecs/tools/docs/document_writer_tool.py +2008 -0
  272. aiecs/tools/knowledge_graph/__init__.py +17 -0
  273. aiecs/tools/knowledge_graph/graph_reasoning_tool.py +807 -0
  274. aiecs/tools/knowledge_graph/graph_search_tool.py +944 -0
  275. aiecs/tools/knowledge_graph/kg_builder_tool.py +524 -0
  276. aiecs/tools/langchain_adapter.py +300 -138
  277. aiecs/tools/schema_generator.py +455 -0
  278. aiecs/tools/search_tool/__init__.py +100 -0
  279. aiecs/tools/search_tool/analyzers.py +581 -0
  280. aiecs/tools/search_tool/cache.py +264 -0
  281. aiecs/tools/search_tool/constants.py +128 -0
  282. aiecs/tools/search_tool/context.py +224 -0
  283. aiecs/tools/search_tool/core.py +778 -0
  284. aiecs/tools/search_tool/deduplicator.py +119 -0
  285. aiecs/tools/search_tool/error_handler.py +242 -0
  286. aiecs/tools/search_tool/metrics.py +343 -0
  287. aiecs/tools/search_tool/rate_limiter.py +172 -0
  288. aiecs/tools/search_tool/schemas.py +275 -0
  289. aiecs/tools/statistics/__init__.py +80 -0
  290. aiecs/tools/statistics/ai_data_analysis_orchestrator.py +646 -0
  291. aiecs/tools/statistics/ai_insight_generator_tool.py +508 -0
  292. aiecs/tools/statistics/ai_report_orchestrator_tool.py +684 -0
  293. aiecs/tools/statistics/data_loader_tool.py +555 -0
  294. aiecs/tools/statistics/data_profiler_tool.py +638 -0
  295. aiecs/tools/statistics/data_transformer_tool.py +580 -0
  296. aiecs/tools/statistics/data_visualizer_tool.py +498 -0
  297. aiecs/tools/statistics/model_trainer_tool.py +507 -0
  298. aiecs/tools/statistics/statistical_analyzer_tool.py +472 -0
  299. aiecs/tools/task_tools/__init__.py +49 -36
  300. aiecs/tools/task_tools/chart_tool.py +200 -184
  301. aiecs/tools/task_tools/classfire_tool.py +268 -267
  302. aiecs/tools/task_tools/image_tool.py +220 -141
  303. aiecs/tools/task_tools/office_tool.py +226 -146
  304. aiecs/tools/task_tools/pandas_tool.py +477 -121
  305. aiecs/tools/task_tools/report_tool.py +390 -142
  306. aiecs/tools/task_tools/research_tool.py +149 -79
  307. aiecs/tools/task_tools/scraper_tool.py +339 -145
  308. aiecs/tools/task_tools/stats_tool.py +448 -209
  309. aiecs/tools/temp_file_manager.py +26 -24
  310. aiecs/tools/tool_executor/__init__.py +18 -16
  311. aiecs/tools/tool_executor/tool_executor.py +364 -52
  312. aiecs/utils/LLM_output_structor.py +74 -48
  313. aiecs/utils/__init__.py +14 -3
  314. aiecs/utils/base_callback.py +0 -3
  315. aiecs/utils/cache_provider.py +696 -0
  316. aiecs/utils/execution_utils.py +50 -31
  317. aiecs/utils/prompt_loader.py +1 -0
  318. aiecs/utils/token_usage_repository.py +37 -11
  319. aiecs/ws/socket_server.py +14 -4
  320. {aiecs-1.0.1.dist-info → aiecs-1.7.17.dist-info}/METADATA +52 -15
  321. aiecs-1.7.17.dist-info/RECORD +337 -0
  322. aiecs-1.7.17.dist-info/entry_points.txt +13 -0
  323. aiecs/config/registry.py +0 -19
  324. aiecs/domain/context/content_engine.py +0 -982
  325. aiecs/llm/base_client.py +0 -99
  326. aiecs/llm/openai_client.py +0 -125
  327. aiecs/llm/vertex_client.py +0 -186
  328. aiecs/llm/xai_client.py +0 -184
  329. aiecs/scripts/dependency_checker.py +0 -857
  330. aiecs/scripts/quick_dependency_check.py +0 -269
  331. aiecs/tools/task_tools/search_api.py +0 -7
  332. aiecs-1.0.1.dist-info/RECORD +0 -90
  333. aiecs-1.0.1.dist-info/entry_points.txt +0 -7
  334. /aiecs/scripts/{setup_nlp_data.sh → dependance_check/setup_nlp_data.sh} +0 -0
  335. /aiecs/scripts/{README_WEASEL_PATCH.md → dependance_patch/fix_weasel/README_WEASEL_PATCH.md} +0 -0
  336. /aiecs/scripts/{fix_weasel_validator.sh → dependance_patch/fix_weasel/fix_weasel_validator.sh} +0 -0
  337. /aiecs/scripts/{run_weasel_patch.sh → dependance_patch/fix_weasel/run_weasel_patch.sh} +0 -0
  338. {aiecs-1.0.1.dist-info → aiecs-1.7.17.dist-info}/WHEEL +0 -0
  339. {aiecs-1.0.1.dist-info → aiecs-1.7.17.dist-info}/licenses/LICENSE +0 -0
  340. {aiecs-1.0.1.dist-info → aiecs-1.7.17.dist-info}/top_level.txt +0 -0
aiecs/domain/__init__.py CHANGED
@@ -6,15 +6,284 @@ Contains business logic and domain models.
6
6
  from .execution.model import TaskStepResult, TaskStatus, ErrorCode
7
7
  from .task.model import TaskContext, DSLStep
8
8
  from .task.dsl_processor import DSLProcessor
9
+ from .context import (
10
+ ContextEngine,
11
+ SessionMetrics,
12
+ ConversationMessage,
13
+ ConversationParticipant,
14
+ ConversationSession,
15
+ AgentCommunicationMessage,
16
+ create_session_key,
17
+ validate_conversation_isolation_pattern,
18
+ )
19
+ from .community import (
20
+ # Core managers
21
+ CommunityManager,
22
+ CommunityIntegration,
23
+ DecisionEngine,
24
+ ResourceManager,
25
+ CollaborativeWorkflowEngine,
26
+ CommunityAnalytics,
27
+ MemberLifecycleHooks,
28
+ # Communication and context
29
+ CommunicationHub,
30
+ Message,
31
+ Event,
32
+ MessageType,
33
+ EventType,
34
+ SharedContextManager,
35
+ SharedContext,
36
+ ContextScope,
37
+ ContextConflictStrategy,
38
+ # Agent adapters
39
+ AgentAdapter,
40
+ StandardLLMAdapter,
41
+ CustomAgentAdapter,
42
+ AgentAdapterRegistry,
43
+ AgentCapability,
44
+ # Builder
45
+ CommunityBuilder,
46
+ builder,
47
+ # Enums
48
+ CommunityRole,
49
+ GovernanceType,
50
+ DecisionStatus,
51
+ ResourceType,
52
+ ConsensusAlgorithm,
53
+ ConflictResolutionStrategy,
54
+ # Models
55
+ CommunityMember,
56
+ CommunityResource,
57
+ CommunityDecision,
58
+ AgentCommunity,
59
+ CollaborationSession,
60
+ # Exceptions
61
+ CommunityException,
62
+ CommunityNotFoundError,
63
+ MemberNotFoundError,
64
+ ResourceNotFoundError,
65
+ DecisionNotFoundError,
66
+ AccessDeniedError,
67
+ MembershipError,
68
+ VotingError,
69
+ GovernanceError,
70
+ CollaborationError,
71
+ CommunityInitializationError,
72
+ CommunityValidationError,
73
+ QuorumNotMetError,
74
+ ConflictResolutionError,
75
+ CommunityCapacityError,
76
+ AgentAdapterError,
77
+ CommunicationError,
78
+ ContextError,
79
+ )
80
+
81
+ from .agent import (
82
+ # Exceptions
83
+ AgentException,
84
+ AgentNotFoundError,
85
+ AgentAlreadyRegisteredError,
86
+ InvalidStateTransitionError,
87
+ ConfigurationError,
88
+ TaskExecutionError,
89
+ ToolAccessDeniedError,
90
+ SerializationError,
91
+ AgentInitializationError,
92
+ # Enums
93
+ AgentState,
94
+ AgentType,
95
+ GoalStatus,
96
+ GoalPriority,
97
+ CapabilityLevel,
98
+ MemoryType,
99
+ # Models
100
+ RetryPolicy,
101
+ AgentConfiguration,
102
+ AgentGoal,
103
+ AgentCapabilityDeclaration,
104
+ AgentMetrics,
105
+ AgentInteraction,
106
+ AgentMemory,
107
+ # Base and Concrete Agents
108
+ BaseAIAgent,
109
+ LLMAgent,
110
+ ToolAgent,
111
+ HybridAgent,
112
+ # Lifecycle Management
113
+ AgentRegistry,
114
+ AgentLifecycleManager,
115
+ get_global_registry,
116
+ get_global_lifecycle_manager,
117
+ # Persistence
118
+ InMemoryPersistence,
119
+ FilePersistence,
120
+ get_global_persistence,
121
+ set_global_persistence,
122
+ # Observability
123
+ AgentController,
124
+ LoggingObserver,
125
+ MetricsObserver,
126
+ # Prompts
127
+ PromptTemplate,
128
+ ChatPromptTemplate,
129
+ MessageBuilder,
130
+ # Tools
131
+ ToolSchemaGenerator,
132
+ generate_tool_schema,
133
+ # Memory
134
+ ConversationMemory,
135
+ Session,
136
+ # Integration
137
+ ContextEngineAdapter,
138
+ EnhancedRetryPolicy,
139
+ RoleConfiguration,
140
+ ContextCompressor,
141
+ compress_messages,
142
+ # Migration
143
+ LegacyAgentWrapper,
144
+ convert_langchain_prompt,
145
+ convert_legacy_config,
146
+ )
9
147
 
10
148
  __all__ = [
11
149
  # Execution domain
12
150
  "TaskStepResult",
13
151
  "TaskStatus",
14
152
  "ErrorCode",
15
-
16
153
  # Task domain
17
154
  "TaskContext",
18
155
  "DSLStep",
19
156
  "DSLProcessor",
157
+ # Context domain
158
+ "ContextEngine",
159
+ "SessionMetrics",
160
+ "ConversationMessage",
161
+ "ConversationParticipant",
162
+ "ConversationSession",
163
+ "AgentCommunicationMessage",
164
+ "create_session_key",
165
+ "validate_conversation_isolation_pattern",
166
+ # Community domain - Core managers
167
+ "CommunityManager",
168
+ "CommunityIntegration",
169
+ "DecisionEngine",
170
+ "ResourceManager",
171
+ "CollaborativeWorkflowEngine",
172
+ "CommunityAnalytics",
173
+ "MemberLifecycleHooks",
174
+ # Community domain - Communication and context
175
+ "CommunicationHub",
176
+ "Message",
177
+ "Event",
178
+ "MessageType",
179
+ "EventType",
180
+ "SharedContextManager",
181
+ "SharedContext",
182
+ "ContextScope",
183
+ "ContextConflictStrategy",
184
+ # Community domain - Agent adapters
185
+ "AgentAdapter",
186
+ "StandardLLMAdapter",
187
+ "CustomAgentAdapter",
188
+ "AgentAdapterRegistry",
189
+ "AgentCapability",
190
+ # Community domain - Builder
191
+ "CommunityBuilder",
192
+ "builder",
193
+ # Community domain - Enums
194
+ "CommunityRole",
195
+ "GovernanceType",
196
+ "DecisionStatus",
197
+ "ResourceType",
198
+ "ConsensusAlgorithm",
199
+ "ConflictResolutionStrategy",
200
+ # Community domain - Models
201
+ "CommunityMember",
202
+ "CommunityResource",
203
+ "CommunityDecision",
204
+ "AgentCommunity",
205
+ "CollaborationSession",
206
+ # Community domain - Exceptions
207
+ "CommunityException",
208
+ "CommunityNotFoundError",
209
+ "MemberNotFoundError",
210
+ "ResourceNotFoundError",
211
+ "DecisionNotFoundError",
212
+ "AccessDeniedError",
213
+ "MembershipError",
214
+ "VotingError",
215
+ "GovernanceError",
216
+ "CollaborationError",
217
+ "CommunityInitializationError",
218
+ "CommunityValidationError",
219
+ "QuorumNotMetError",
220
+ "ConflictResolutionError",
221
+ "CommunityCapacityError",
222
+ "AgentAdapterError",
223
+ "CommunicationError",
224
+ "ContextError",
225
+ # Agent domain - Exceptions
226
+ "AgentException",
227
+ "AgentNotFoundError",
228
+ "AgentAlreadyRegisteredError",
229
+ "InvalidStateTransitionError",
230
+ "ConfigurationError",
231
+ "TaskExecutionError",
232
+ "ToolAccessDeniedError",
233
+ "SerializationError",
234
+ "AgentInitializationError",
235
+ # Agent domain - Enums
236
+ "AgentState",
237
+ "AgentType",
238
+ "GoalStatus",
239
+ "GoalPriority",
240
+ "CapabilityLevel",
241
+ "MemoryType",
242
+ # Agent domain - Models
243
+ "RetryPolicy",
244
+ "AgentConfiguration",
245
+ "AgentGoal",
246
+ "AgentCapabilityDeclaration",
247
+ "AgentMetrics",
248
+ "AgentInteraction",
249
+ "AgentMemory",
250
+ # Agent domain - Base and Concrete Agents
251
+ "BaseAIAgent",
252
+ "LLMAgent",
253
+ "ToolAgent",
254
+ "HybridAgent",
255
+ # Agent domain - Lifecycle Management
256
+ "AgentRegistry",
257
+ "AgentLifecycleManager",
258
+ "get_global_registry",
259
+ "get_global_lifecycle_manager",
260
+ # Agent domain - Persistence
261
+ "InMemoryPersistence",
262
+ "FilePersistence",
263
+ "get_global_persistence",
264
+ "set_global_persistence",
265
+ # Agent domain - Observability
266
+ "AgentController",
267
+ "LoggingObserver",
268
+ "MetricsObserver",
269
+ # Agent domain - Prompts
270
+ "PromptTemplate",
271
+ "ChatPromptTemplate",
272
+ "MessageBuilder",
273
+ # Agent domain - Tools
274
+ "ToolSchemaGenerator",
275
+ "generate_tool_schema",
276
+ # Agent domain - Memory
277
+ "ConversationMemory",
278
+ "Session",
279
+ # Agent domain - Integration
280
+ "ContextEngineAdapter",
281
+ "EnhancedRetryPolicy",
282
+ "RoleConfiguration",
283
+ "ContextCompressor",
284
+ "compress_messages",
285
+ # Agent domain - Migration
286
+ "LegacyAgentWrapper",
287
+ "convert_langchain_prompt",
288
+ "convert_legacy_config",
20
289
  ]
@@ -0,0 +1,191 @@
1
+ """
2
+ Agent Domain Module
3
+
4
+ Provides the base AI agent model and related components.
5
+ """
6
+
7
+ # Exceptions
8
+ from .exceptions import (
9
+ AgentException,
10
+ AgentNotFoundError,
11
+ AgentAlreadyRegisteredError,
12
+ InvalidStateTransitionError,
13
+ ConfigurationError,
14
+ TaskExecutionError,
15
+ ToolAccessDeniedError,
16
+ SerializationError,
17
+ AgentInitializationError,
18
+ )
19
+
20
+ # Models and Enums
21
+ from .models import (
22
+ AgentState,
23
+ AgentType,
24
+ GoalStatus,
25
+ GoalPriority,
26
+ CapabilityLevel,
27
+ MemoryType,
28
+ RetryPolicy,
29
+ AgentConfiguration,
30
+ AgentGoal,
31
+ AgentCapabilityDeclaration,
32
+ AgentMetrics,
33
+ GraphMetrics,
34
+ AgentInteraction,
35
+ AgentMemory,
36
+ )
37
+
38
+ # Base Agent
39
+ from .base_agent import BaseAIAgent
40
+
41
+ # Concrete Agents
42
+ from .llm_agent import LLMAgent
43
+ from .tool_agent import ToolAgent
44
+ from .hybrid_agent import HybridAgent
45
+ from .knowledge_aware_agent import KnowledgeAwareAgent
46
+
47
+ # Graph-Aware Mixin
48
+ from .graph_aware_mixin import GraphAwareAgentMixin
49
+
50
+ # Lifecycle Management
51
+ from .registry import AgentRegistry, get_global_registry, reset_global_registry
52
+ from .lifecycle import (
53
+ AgentLifecycleManager,
54
+ get_global_lifecycle_manager,
55
+ reset_global_lifecycle_manager,
56
+ )
57
+
58
+ # Persistence
59
+ from .persistence import (
60
+ AgentPersistence,
61
+ InMemoryPersistence,
62
+ FilePersistence,
63
+ AgentStateSerializer,
64
+ get_global_persistence,
65
+ set_global_persistence,
66
+ reset_global_persistence,
67
+ )
68
+
69
+ # Observability
70
+ from .observability import (
71
+ AgentObserver,
72
+ LoggingObserver,
73
+ MetricsObserver,
74
+ AgentController,
75
+ )
76
+
77
+ # Prompts
78
+ from .prompts import (
79
+ PromptTemplate,
80
+ ChatPromptTemplate,
81
+ MessageBuilder,
82
+ )
83
+
84
+ # Tools
85
+ from .tools import (
86
+ ToolSchemaGenerator,
87
+ generate_tool_schema,
88
+ )
89
+
90
+ # Memory
91
+ from .memory import (
92
+ ConversationMemory,
93
+ Session,
94
+ )
95
+
96
+ # Integration
97
+ from .integration import (
98
+ ContextEngineAdapter,
99
+ EnhancedRetryPolicy,
100
+ ErrorClassifier,
101
+ RoleConfiguration,
102
+ load_role_config,
103
+ ContextCompressor,
104
+ compress_messages,
105
+ )
106
+
107
+ # Migration
108
+ from .migration import (
109
+ LegacyAgentWrapper,
110
+ convert_langchain_prompt,
111
+ convert_legacy_config,
112
+ )
113
+
114
+ __all__ = [
115
+ # Exceptions
116
+ "AgentException",
117
+ "AgentNotFoundError",
118
+ "AgentAlreadyRegisteredError",
119
+ "InvalidStateTransitionError",
120
+ "ConfigurationError",
121
+ "TaskExecutionError",
122
+ "ToolAccessDeniedError",
123
+ "SerializationError",
124
+ "AgentInitializationError",
125
+ # Enums
126
+ "AgentState",
127
+ "AgentType",
128
+ "GoalStatus",
129
+ "GoalPriority",
130
+ "CapabilityLevel",
131
+ "MemoryType",
132
+ # Models
133
+ "RetryPolicy",
134
+ "AgentConfiguration",
135
+ "AgentGoal",
136
+ "AgentCapabilityDeclaration",
137
+ "AgentMetrics",
138
+ "GraphMetrics",
139
+ "AgentInteraction",
140
+ "AgentMemory",
141
+ # Base Agent
142
+ "BaseAIAgent",
143
+ # Concrete Agents
144
+ "LLMAgent",
145
+ "ToolAgent",
146
+ "HybridAgent", # Original hybrid agent (backward compatible)
147
+ "KnowledgeAwareAgent", # Enhanced agent with knowledge graph integration
148
+ "GraphAwareAgentMixin", # Reusable mixin for graph-aware agents
149
+ # Lifecycle Management
150
+ "AgentRegistry",
151
+ "get_global_registry",
152
+ "reset_global_registry",
153
+ "AgentLifecycleManager",
154
+ "get_global_lifecycle_manager",
155
+ "reset_global_lifecycle_manager",
156
+ # Persistence
157
+ "AgentPersistence",
158
+ "InMemoryPersistence",
159
+ "FilePersistence",
160
+ "AgentStateSerializer",
161
+ "get_global_persistence",
162
+ "set_global_persistence",
163
+ "reset_global_persistence",
164
+ # Observability
165
+ "AgentObserver",
166
+ "LoggingObserver",
167
+ "MetricsObserver",
168
+ "AgentController",
169
+ # Prompts
170
+ "PromptTemplate",
171
+ "ChatPromptTemplate",
172
+ "MessageBuilder",
173
+ # Tools
174
+ "ToolSchemaGenerator",
175
+ "generate_tool_schema",
176
+ # Memory
177
+ "ConversationMemory",
178
+ "Session",
179
+ # Integration
180
+ "ContextEngineAdapter",
181
+ "EnhancedRetryPolicy",
182
+ "ErrorClassifier",
183
+ "RoleConfiguration",
184
+ "load_role_config",
185
+ "ContextCompressor",
186
+ "compress_messages",
187
+ # Migration
188
+ "LegacyAgentWrapper",
189
+ "convert_langchain_prompt",
190
+ "convert_legacy_config",
191
+ ]