aiecs 1.0.1__py3-none-any.whl → 1.7.6__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.
- aiecs/__init__.py +13 -16
- aiecs/__main__.py +7 -7
- aiecs/aiecs_client.py +269 -75
- aiecs/application/executors/operation_executor.py +79 -54
- aiecs/application/knowledge_graph/__init__.py +7 -0
- aiecs/application/knowledge_graph/builder/__init__.py +37 -0
- aiecs/application/knowledge_graph/builder/data_quality.py +302 -0
- aiecs/application/knowledge_graph/builder/data_reshaping.py +293 -0
- aiecs/application/knowledge_graph/builder/document_builder.py +369 -0
- aiecs/application/knowledge_graph/builder/graph_builder.py +490 -0
- aiecs/application/knowledge_graph/builder/import_optimizer.py +396 -0
- aiecs/application/knowledge_graph/builder/schema_inference.py +462 -0
- aiecs/application/knowledge_graph/builder/schema_mapping.py +563 -0
- aiecs/application/knowledge_graph/builder/structured_pipeline.py +1384 -0
- aiecs/application/knowledge_graph/builder/text_chunker.py +317 -0
- aiecs/application/knowledge_graph/extractors/__init__.py +27 -0
- aiecs/application/knowledge_graph/extractors/base.py +98 -0
- aiecs/application/knowledge_graph/extractors/llm_entity_extractor.py +422 -0
- aiecs/application/knowledge_graph/extractors/llm_relation_extractor.py +347 -0
- aiecs/application/knowledge_graph/extractors/ner_entity_extractor.py +241 -0
- aiecs/application/knowledge_graph/fusion/__init__.py +78 -0
- aiecs/application/knowledge_graph/fusion/ab_testing.py +395 -0
- aiecs/application/knowledge_graph/fusion/abbreviation_expander.py +327 -0
- aiecs/application/knowledge_graph/fusion/alias_index.py +597 -0
- aiecs/application/knowledge_graph/fusion/alias_matcher.py +384 -0
- aiecs/application/knowledge_graph/fusion/cache_coordinator.py +343 -0
- aiecs/application/knowledge_graph/fusion/entity_deduplicator.py +433 -0
- aiecs/application/knowledge_graph/fusion/entity_linker.py +511 -0
- aiecs/application/knowledge_graph/fusion/evaluation_dataset.py +240 -0
- aiecs/application/knowledge_graph/fusion/knowledge_fusion.py +632 -0
- aiecs/application/knowledge_graph/fusion/matching_config.py +489 -0
- aiecs/application/knowledge_graph/fusion/name_normalizer.py +352 -0
- aiecs/application/knowledge_graph/fusion/relation_deduplicator.py +183 -0
- aiecs/application/knowledge_graph/fusion/semantic_name_matcher.py +464 -0
- aiecs/application/knowledge_graph/fusion/similarity_pipeline.py +534 -0
- aiecs/application/knowledge_graph/pattern_matching/__init__.py +21 -0
- aiecs/application/knowledge_graph/pattern_matching/pattern_matcher.py +342 -0
- aiecs/application/knowledge_graph/pattern_matching/query_executor.py +366 -0
- aiecs/application/knowledge_graph/profiling/__init__.py +12 -0
- aiecs/application/knowledge_graph/profiling/query_plan_visualizer.py +195 -0
- aiecs/application/knowledge_graph/profiling/query_profiler.py +223 -0
- aiecs/application/knowledge_graph/reasoning/__init__.py +27 -0
- aiecs/application/knowledge_graph/reasoning/evidence_synthesis.py +341 -0
- aiecs/application/knowledge_graph/reasoning/inference_engine.py +500 -0
- aiecs/application/knowledge_graph/reasoning/logic_form_parser.py +163 -0
- aiecs/application/knowledge_graph/reasoning/logic_parser/__init__.py +79 -0
- aiecs/application/knowledge_graph/reasoning/logic_parser/ast_builder.py +513 -0
- aiecs/application/knowledge_graph/reasoning/logic_parser/ast_nodes.py +913 -0
- aiecs/application/knowledge_graph/reasoning/logic_parser/ast_validator.py +866 -0
- aiecs/application/knowledge_graph/reasoning/logic_parser/error_handler.py +475 -0
- aiecs/application/knowledge_graph/reasoning/logic_parser/parser.py +396 -0
- aiecs/application/knowledge_graph/reasoning/logic_parser/query_context.py +208 -0
- aiecs/application/knowledge_graph/reasoning/logic_query_integration.py +170 -0
- aiecs/application/knowledge_graph/reasoning/query_planner.py +855 -0
- aiecs/application/knowledge_graph/reasoning/reasoning_engine.py +518 -0
- aiecs/application/knowledge_graph/retrieval/__init__.py +27 -0
- aiecs/application/knowledge_graph/retrieval/query_intent_classifier.py +211 -0
- aiecs/application/knowledge_graph/retrieval/retrieval_strategies.py +592 -0
- aiecs/application/knowledge_graph/retrieval/strategy_types.py +23 -0
- aiecs/application/knowledge_graph/search/__init__.py +59 -0
- aiecs/application/knowledge_graph/search/hybrid_search.py +457 -0
- aiecs/application/knowledge_graph/search/reranker.py +293 -0
- aiecs/application/knowledge_graph/search/reranker_strategies.py +535 -0
- aiecs/application/knowledge_graph/search/text_similarity.py +392 -0
- aiecs/application/knowledge_graph/traversal/__init__.py +15 -0
- aiecs/application/knowledge_graph/traversal/enhanced_traversal.py +305 -0
- aiecs/application/knowledge_graph/traversal/path_scorer.py +271 -0
- aiecs/application/knowledge_graph/validators/__init__.py +13 -0
- aiecs/application/knowledge_graph/validators/relation_validator.py +239 -0
- aiecs/application/knowledge_graph/visualization/__init__.py +11 -0
- aiecs/application/knowledge_graph/visualization/graph_visualizer.py +313 -0
- aiecs/common/__init__.py +9 -0
- aiecs/common/knowledge_graph/__init__.py +17 -0
- aiecs/common/knowledge_graph/runnable.py +471 -0
- aiecs/config/__init__.py +20 -5
- aiecs/config/config.py +762 -31
- aiecs/config/graph_config.py +131 -0
- aiecs/config/tool_config.py +399 -0
- aiecs/core/__init__.py +29 -13
- aiecs/core/interface/__init__.py +2 -2
- aiecs/core/interface/execution_interface.py +22 -22
- aiecs/core/interface/storage_interface.py +37 -88
- aiecs/core/registry/__init__.py +31 -0
- aiecs/core/registry/service_registry.py +92 -0
- aiecs/domain/__init__.py +270 -1
- aiecs/domain/agent/__init__.py +191 -0
- aiecs/domain/agent/base_agent.py +3870 -0
- aiecs/domain/agent/exceptions.py +99 -0
- aiecs/domain/agent/graph_aware_mixin.py +569 -0
- aiecs/domain/agent/hybrid_agent.py +1435 -0
- aiecs/domain/agent/integration/__init__.py +29 -0
- aiecs/domain/agent/integration/context_compressor.py +216 -0
- aiecs/domain/agent/integration/context_engine_adapter.py +587 -0
- aiecs/domain/agent/integration/protocols.py +281 -0
- aiecs/domain/agent/integration/retry_policy.py +218 -0
- aiecs/domain/agent/integration/role_config.py +213 -0
- aiecs/domain/agent/knowledge_aware_agent.py +1892 -0
- aiecs/domain/agent/lifecycle.py +291 -0
- aiecs/domain/agent/llm_agent.py +692 -0
- aiecs/domain/agent/memory/__init__.py +12 -0
- aiecs/domain/agent/memory/conversation.py +1124 -0
- aiecs/domain/agent/migration/__init__.py +14 -0
- aiecs/domain/agent/migration/conversion.py +163 -0
- aiecs/domain/agent/migration/legacy_wrapper.py +86 -0
- aiecs/domain/agent/models.py +884 -0
- aiecs/domain/agent/observability.py +479 -0
- aiecs/domain/agent/persistence.py +449 -0
- aiecs/domain/agent/prompts/__init__.py +29 -0
- aiecs/domain/agent/prompts/builder.py +159 -0
- aiecs/domain/agent/prompts/formatters.py +187 -0
- aiecs/domain/agent/prompts/template.py +255 -0
- aiecs/domain/agent/registry.py +253 -0
- aiecs/domain/agent/tool_agent.py +444 -0
- aiecs/domain/agent/tools/__init__.py +15 -0
- aiecs/domain/agent/tools/schema_generator.py +364 -0
- aiecs/domain/community/__init__.py +155 -0
- aiecs/domain/community/agent_adapter.py +469 -0
- aiecs/domain/community/analytics.py +432 -0
- aiecs/domain/community/collaborative_workflow.py +648 -0
- aiecs/domain/community/communication_hub.py +634 -0
- aiecs/domain/community/community_builder.py +320 -0
- aiecs/domain/community/community_integration.py +796 -0
- aiecs/domain/community/community_manager.py +803 -0
- aiecs/domain/community/decision_engine.py +849 -0
- aiecs/domain/community/exceptions.py +231 -0
- aiecs/domain/community/models/__init__.py +33 -0
- aiecs/domain/community/models/community_models.py +234 -0
- aiecs/domain/community/resource_manager.py +461 -0
- aiecs/domain/community/shared_context_manager.py +589 -0
- aiecs/domain/context/__init__.py +40 -10
- aiecs/domain/context/context_engine.py +1910 -0
- aiecs/domain/context/conversation_models.py +87 -53
- aiecs/domain/context/graph_memory.py +582 -0
- aiecs/domain/execution/model.py +12 -4
- aiecs/domain/knowledge_graph/__init__.py +19 -0
- aiecs/domain/knowledge_graph/models/__init__.py +52 -0
- aiecs/domain/knowledge_graph/models/entity.py +148 -0
- aiecs/domain/knowledge_graph/models/evidence.py +178 -0
- aiecs/domain/knowledge_graph/models/inference_rule.py +184 -0
- aiecs/domain/knowledge_graph/models/path.py +171 -0
- aiecs/domain/knowledge_graph/models/path_pattern.py +171 -0
- aiecs/domain/knowledge_graph/models/query.py +261 -0
- aiecs/domain/knowledge_graph/models/query_plan.py +181 -0
- aiecs/domain/knowledge_graph/models/relation.py +202 -0
- aiecs/domain/knowledge_graph/schema/__init__.py +23 -0
- aiecs/domain/knowledge_graph/schema/entity_type.py +131 -0
- aiecs/domain/knowledge_graph/schema/graph_schema.py +253 -0
- aiecs/domain/knowledge_graph/schema/property_schema.py +143 -0
- aiecs/domain/knowledge_graph/schema/relation_type.py +163 -0
- aiecs/domain/knowledge_graph/schema/schema_manager.py +691 -0
- aiecs/domain/knowledge_graph/schema/type_enums.py +209 -0
- aiecs/domain/task/dsl_processor.py +172 -56
- aiecs/domain/task/model.py +20 -8
- aiecs/domain/task/task_context.py +27 -24
- aiecs/infrastructure/__init__.py +0 -2
- aiecs/infrastructure/graph_storage/__init__.py +11 -0
- aiecs/infrastructure/graph_storage/base.py +837 -0
- aiecs/infrastructure/graph_storage/batch_operations.py +458 -0
- aiecs/infrastructure/graph_storage/cache.py +424 -0
- aiecs/infrastructure/graph_storage/distributed.py +223 -0
- aiecs/infrastructure/graph_storage/error_handling.py +380 -0
- aiecs/infrastructure/graph_storage/graceful_degradation.py +294 -0
- aiecs/infrastructure/graph_storage/health_checks.py +378 -0
- aiecs/infrastructure/graph_storage/in_memory.py +1197 -0
- aiecs/infrastructure/graph_storage/index_optimization.py +446 -0
- aiecs/infrastructure/graph_storage/lazy_loading.py +431 -0
- aiecs/infrastructure/graph_storage/metrics.py +344 -0
- aiecs/infrastructure/graph_storage/migration.py +400 -0
- aiecs/infrastructure/graph_storage/pagination.py +483 -0
- aiecs/infrastructure/graph_storage/performance_monitoring.py +456 -0
- aiecs/infrastructure/graph_storage/postgres.py +1563 -0
- aiecs/infrastructure/graph_storage/property_storage.py +353 -0
- aiecs/infrastructure/graph_storage/protocols.py +76 -0
- aiecs/infrastructure/graph_storage/query_optimizer.py +642 -0
- aiecs/infrastructure/graph_storage/schema_cache.py +290 -0
- aiecs/infrastructure/graph_storage/sqlite.py +1373 -0
- aiecs/infrastructure/graph_storage/streaming.py +487 -0
- aiecs/infrastructure/graph_storage/tenant.py +412 -0
- aiecs/infrastructure/messaging/celery_task_manager.py +92 -54
- aiecs/infrastructure/messaging/websocket_manager.py +51 -35
- aiecs/infrastructure/monitoring/__init__.py +22 -0
- aiecs/infrastructure/monitoring/executor_metrics.py +45 -11
- aiecs/infrastructure/monitoring/global_metrics_manager.py +212 -0
- aiecs/infrastructure/monitoring/structured_logger.py +3 -7
- aiecs/infrastructure/monitoring/tracing_manager.py +63 -35
- aiecs/infrastructure/persistence/__init__.py +14 -1
- aiecs/infrastructure/persistence/context_engine_client.py +184 -0
- aiecs/infrastructure/persistence/database_manager.py +67 -43
- aiecs/infrastructure/persistence/file_storage.py +180 -103
- aiecs/infrastructure/persistence/redis_client.py +74 -21
- aiecs/llm/__init__.py +73 -25
- aiecs/llm/callbacks/__init__.py +11 -0
- aiecs/llm/{custom_callbacks.py → callbacks/custom_callbacks.py} +26 -19
- aiecs/llm/client_factory.py +224 -36
- aiecs/llm/client_resolver.py +155 -0
- aiecs/llm/clients/__init__.py +38 -0
- aiecs/llm/clients/base_client.py +324 -0
- aiecs/llm/clients/google_function_calling_mixin.py +457 -0
- aiecs/llm/clients/googleai_client.py +241 -0
- aiecs/llm/clients/openai_client.py +158 -0
- aiecs/llm/clients/openai_compatible_mixin.py +367 -0
- aiecs/llm/clients/vertex_client.py +897 -0
- aiecs/llm/clients/xai_client.py +201 -0
- aiecs/llm/config/__init__.py +51 -0
- aiecs/llm/config/config_loader.py +272 -0
- aiecs/llm/config/config_validator.py +206 -0
- aiecs/llm/config/model_config.py +143 -0
- aiecs/llm/protocols.py +149 -0
- aiecs/llm/utils/__init__.py +10 -0
- aiecs/llm/utils/validate_config.py +89 -0
- aiecs/main.py +140 -121
- aiecs/scripts/aid/VERSION_MANAGEMENT.md +138 -0
- aiecs/scripts/aid/__init__.py +19 -0
- aiecs/scripts/aid/module_checker.py +499 -0
- aiecs/scripts/aid/version_manager.py +235 -0
- aiecs/scripts/{DEPENDENCY_SYSTEM_SUMMARY.md → dependance_check/DEPENDENCY_SYSTEM_SUMMARY.md} +1 -0
- aiecs/scripts/{README_DEPENDENCY_CHECKER.md → dependance_check/README_DEPENDENCY_CHECKER.md} +1 -0
- aiecs/scripts/dependance_check/__init__.py +15 -0
- aiecs/scripts/dependance_check/dependency_checker.py +1835 -0
- aiecs/scripts/{dependency_fixer.py → dependance_check/dependency_fixer.py} +192 -90
- aiecs/scripts/{download_nlp_data.py → dependance_check/download_nlp_data.py} +203 -71
- aiecs/scripts/dependance_patch/__init__.py +7 -0
- aiecs/scripts/dependance_patch/fix_weasel/__init__.py +11 -0
- aiecs/scripts/{fix_weasel_validator.py → dependance_patch/fix_weasel/fix_weasel_validator.py} +21 -14
- aiecs/scripts/{patch_weasel_library.sh → dependance_patch/fix_weasel/patch_weasel_library.sh} +1 -1
- aiecs/scripts/knowledge_graph/__init__.py +3 -0
- aiecs/scripts/knowledge_graph/run_threshold_experiments.py +212 -0
- aiecs/scripts/migrations/multi_tenancy/README.md +142 -0
- aiecs/scripts/tools_develop/README.md +671 -0
- aiecs/scripts/tools_develop/README_CONFIG_CHECKER.md +273 -0
- aiecs/scripts/tools_develop/TOOLS_CONFIG_GUIDE.md +1287 -0
- aiecs/scripts/tools_develop/TOOL_AUTO_DISCOVERY.md +234 -0
- aiecs/scripts/tools_develop/__init__.py +21 -0
- aiecs/scripts/tools_develop/check_all_tools_config.py +548 -0
- aiecs/scripts/tools_develop/check_type_annotations.py +257 -0
- aiecs/scripts/tools_develop/pre-commit-schema-coverage.sh +66 -0
- aiecs/scripts/tools_develop/schema_coverage.py +511 -0
- aiecs/scripts/tools_develop/validate_tool_schemas.py +475 -0
- aiecs/scripts/tools_develop/verify_executor_config_fix.py +98 -0
- aiecs/scripts/tools_develop/verify_tools.py +352 -0
- aiecs/tasks/__init__.py +0 -1
- aiecs/tasks/worker.py +115 -47
- aiecs/tools/__init__.py +194 -72
- aiecs/tools/apisource/__init__.py +99 -0
- aiecs/tools/apisource/intelligence/__init__.py +19 -0
- aiecs/tools/apisource/intelligence/data_fusion.py +632 -0
- aiecs/tools/apisource/intelligence/query_analyzer.py +417 -0
- aiecs/tools/apisource/intelligence/search_enhancer.py +385 -0
- aiecs/tools/apisource/monitoring/__init__.py +9 -0
- aiecs/tools/apisource/monitoring/metrics.py +330 -0
- aiecs/tools/apisource/providers/__init__.py +112 -0
- aiecs/tools/apisource/providers/base.py +671 -0
- aiecs/tools/apisource/providers/census.py +397 -0
- aiecs/tools/apisource/providers/fred.py +535 -0
- aiecs/tools/apisource/providers/newsapi.py +409 -0
- aiecs/tools/apisource/providers/worldbank.py +352 -0
- aiecs/tools/apisource/reliability/__init__.py +12 -0
- aiecs/tools/apisource/reliability/error_handler.py +363 -0
- aiecs/tools/apisource/reliability/fallback_strategy.py +376 -0
- aiecs/tools/apisource/tool.py +832 -0
- aiecs/tools/apisource/utils/__init__.py +9 -0
- aiecs/tools/apisource/utils/validators.py +334 -0
- aiecs/tools/base_tool.py +415 -21
- aiecs/tools/docs/__init__.py +121 -0
- aiecs/tools/docs/ai_document_orchestrator.py +607 -0
- aiecs/tools/docs/ai_document_writer_orchestrator.py +2350 -0
- aiecs/tools/docs/content_insertion_tool.py +1320 -0
- aiecs/tools/docs/document_creator_tool.py +1323 -0
- aiecs/tools/docs/document_layout_tool.py +1160 -0
- aiecs/tools/docs/document_parser_tool.py +1011 -0
- aiecs/tools/docs/document_writer_tool.py +1829 -0
- aiecs/tools/knowledge_graph/__init__.py +17 -0
- aiecs/tools/knowledge_graph/graph_reasoning_tool.py +807 -0
- aiecs/tools/knowledge_graph/graph_search_tool.py +944 -0
- aiecs/tools/knowledge_graph/kg_builder_tool.py +524 -0
- aiecs/tools/langchain_adapter.py +300 -138
- aiecs/tools/schema_generator.py +455 -0
- aiecs/tools/search_tool/__init__.py +100 -0
- aiecs/tools/search_tool/analyzers.py +581 -0
- aiecs/tools/search_tool/cache.py +264 -0
- aiecs/tools/search_tool/constants.py +128 -0
- aiecs/tools/search_tool/context.py +224 -0
- aiecs/tools/search_tool/core.py +778 -0
- aiecs/tools/search_tool/deduplicator.py +119 -0
- aiecs/tools/search_tool/error_handler.py +242 -0
- aiecs/tools/search_tool/metrics.py +343 -0
- aiecs/tools/search_tool/rate_limiter.py +172 -0
- aiecs/tools/search_tool/schemas.py +275 -0
- aiecs/tools/statistics/__init__.py +80 -0
- aiecs/tools/statistics/ai_data_analysis_orchestrator.py +646 -0
- aiecs/tools/statistics/ai_insight_generator_tool.py +508 -0
- aiecs/tools/statistics/ai_report_orchestrator_tool.py +684 -0
- aiecs/tools/statistics/data_loader_tool.py +555 -0
- aiecs/tools/statistics/data_profiler_tool.py +638 -0
- aiecs/tools/statistics/data_transformer_tool.py +580 -0
- aiecs/tools/statistics/data_visualizer_tool.py +498 -0
- aiecs/tools/statistics/model_trainer_tool.py +507 -0
- aiecs/tools/statistics/statistical_analyzer_tool.py +472 -0
- aiecs/tools/task_tools/__init__.py +49 -36
- aiecs/tools/task_tools/chart_tool.py +200 -184
- aiecs/tools/task_tools/classfire_tool.py +268 -267
- aiecs/tools/task_tools/image_tool.py +175 -131
- aiecs/tools/task_tools/office_tool.py +226 -146
- aiecs/tools/task_tools/pandas_tool.py +477 -121
- aiecs/tools/task_tools/report_tool.py +390 -142
- aiecs/tools/task_tools/research_tool.py +149 -79
- aiecs/tools/task_tools/scraper_tool.py +339 -145
- aiecs/tools/task_tools/stats_tool.py +448 -209
- aiecs/tools/temp_file_manager.py +26 -24
- aiecs/tools/tool_executor/__init__.py +18 -16
- aiecs/tools/tool_executor/tool_executor.py +364 -52
- aiecs/utils/LLM_output_structor.py +74 -48
- aiecs/utils/__init__.py +14 -3
- aiecs/utils/base_callback.py +0 -3
- aiecs/utils/cache_provider.py +696 -0
- aiecs/utils/execution_utils.py +50 -31
- aiecs/utils/prompt_loader.py +1 -0
- aiecs/utils/token_usage_repository.py +37 -11
- aiecs/ws/socket_server.py +14 -4
- {aiecs-1.0.1.dist-info → aiecs-1.7.6.dist-info}/METADATA +52 -15
- aiecs-1.7.6.dist-info/RECORD +337 -0
- aiecs-1.7.6.dist-info/entry_points.txt +13 -0
- aiecs/config/registry.py +0 -19
- aiecs/domain/context/content_engine.py +0 -982
- aiecs/llm/base_client.py +0 -99
- aiecs/llm/openai_client.py +0 -125
- aiecs/llm/vertex_client.py +0 -186
- aiecs/llm/xai_client.py +0 -184
- aiecs/scripts/dependency_checker.py +0 -857
- aiecs/scripts/quick_dependency_check.py +0 -269
- aiecs/tools/task_tools/search_api.py +0 -7
- aiecs-1.0.1.dist-info/RECORD +0 -90
- aiecs-1.0.1.dist-info/entry_points.txt +0 -7
- /aiecs/scripts/{setup_nlp_data.sh → dependance_check/setup_nlp_data.sh} +0 -0
- /aiecs/scripts/{README_WEASEL_PATCH.md → dependance_patch/fix_weasel/README_WEASEL_PATCH.md} +0 -0
- /aiecs/scripts/{fix_weasel_validator.sh → dependance_patch/fix_weasel/fix_weasel_validator.sh} +0 -0
- /aiecs/scripts/{run_weasel_patch.sh → dependance_patch/fix_weasel/run_weasel_patch.sh} +0 -0
- {aiecs-1.0.1.dist-info → aiecs-1.7.6.dist-info}/WHEEL +0 -0
- {aiecs-1.0.1.dist-info → aiecs-1.7.6.dist-info}/licenses/LICENSE +0 -0
- {aiecs-1.0.1.dist-info → aiecs-1.7.6.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
|
+
]
|