omnicoreagent 0.3.5__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.
Files changed (93) hide show
  1. omnicoreagent/__init__.py +80 -0
  2. omnicoreagent/core/__init__.py +34 -0
  3. omnicoreagent/core/agents/__init__.py +22 -0
  4. omnicoreagent/core/agents/base.py +2003 -0
  5. omnicoreagent/core/agents/react_agent.py +48 -0
  6. omnicoreagent/core/constants.py +22 -0
  7. omnicoreagent/core/context_manager.py +264 -0
  8. omnicoreagent/core/events/__init__.py +15 -0
  9. omnicoreagent/core/events/base.py +184 -0
  10. omnicoreagent/core/events/event_router.py +104 -0
  11. omnicoreagent/core/events/in_memory.py +23 -0
  12. omnicoreagent/core/events/redis_stream.py +31 -0
  13. omnicoreagent/core/guardrails.py +1061 -0
  14. omnicoreagent/core/llm.py +414 -0
  15. omnicoreagent/core/memory_store/__init__.py +26 -0
  16. omnicoreagent/core/memory_store/base.py +46 -0
  17. omnicoreagent/core/memory_store/in_memory.py +217 -0
  18. omnicoreagent/core/memory_store/memory_router.py +110 -0
  19. omnicoreagent/core/memory_store/mongodb.py +236 -0
  20. omnicoreagent/core/memory_store/redis_memory.py +506 -0
  21. omnicoreagent/core/memory_store/sql_db_memory.py +518 -0
  22. omnicoreagent/core/skills/__init__.py +15 -0
  23. omnicoreagent/core/skills/manager.py +257 -0
  24. omnicoreagent/core/skills/models.py +77 -0
  25. omnicoreagent/core/skills/tools.py +229 -0
  26. omnicoreagent/core/summarizer/__init__.py +22 -0
  27. omnicoreagent/core/summarizer/summarizer_engine.py +298 -0
  28. omnicoreagent/core/summarizer/summarizer_types.py +82 -0
  29. omnicoreagent/core/summarizer/tokenizer.py +86 -0
  30. omnicoreagent/core/system_prompts.py +972 -0
  31. omnicoreagent/core/token_usage.py +229 -0
  32. omnicoreagent/core/tool_response_offloader.py +400 -0
  33. omnicoreagent/core/tools/__init__.py +12 -0
  34. omnicoreagent/core/tools/advance_tools/__init__.py +3 -0
  35. omnicoreagent/core/tools/advance_tools/advanced_tools_use.py +303 -0
  36. omnicoreagent/core/tools/advance_tools_use.py +65 -0
  37. omnicoreagent/core/tools/artifact_tool.py +150 -0
  38. omnicoreagent/core/tools/local_tools_registry.py +183 -0
  39. omnicoreagent/core/tools/memory_tool/__init__.py +35 -0
  40. omnicoreagent/core/tools/memory_tool/base.py +47 -0
  41. omnicoreagent/core/tools/memory_tool/factory.py +133 -0
  42. omnicoreagent/core/tools/memory_tool/local_storage.py +232 -0
  43. omnicoreagent/core/tools/memory_tool/memory_tool.py +270 -0
  44. omnicoreagent/core/tools/memory_tool/r2_storage.py +102 -0
  45. omnicoreagent/core/tools/memory_tool/s3_storage.py +772 -0
  46. omnicoreagent/core/tools/tools_handler.py +284 -0
  47. omnicoreagent/core/types.py +284 -0
  48. omnicoreagent/core/utils.py +873 -0
  49. omnicoreagent/mcp_clients_connection/__init__.py +34 -0
  50. omnicoreagent/mcp_clients_connection/client.py +545 -0
  51. omnicoreagent/mcp_clients_connection/notifications.py +122 -0
  52. omnicoreagent/mcp_clients_connection/prompts.py +208 -0
  53. omnicoreagent/mcp_clients_connection/refresh_server_capabilities.py +65 -0
  54. omnicoreagent/mcp_clients_connection/resources.py +167 -0
  55. omnicoreagent/mcp_clients_connection/sampling.py +210 -0
  56. omnicoreagent/mcp_clients_connection/tools.py +20 -0
  57. omnicoreagent/omni_agent/__init__.py +27 -0
  58. omnicoreagent/omni_agent/agent.py +490 -0
  59. omnicoreagent/omni_agent/background_agent/__init__.py +20 -0
  60. omnicoreagent/omni_agent/background_agent/background_agent_manager.py +527 -0
  61. omnicoreagent/omni_agent/background_agent/background_agents.py +477 -0
  62. omnicoreagent/omni_agent/background_agent/base.py +30 -0
  63. omnicoreagent/omni_agent/background_agent/scheduler_backend.py +153 -0
  64. omnicoreagent/omni_agent/background_agent/task_registry.py +87 -0
  65. omnicoreagent/omni_agent/config/__init__.py +17 -0
  66. omnicoreagent/omni_agent/config/transformer.py +327 -0
  67. omnicoreagent/omni_agent/deep_agent/__init__.py +36 -0
  68. omnicoreagent/omni_agent/deep_agent/deep_agent.py +357 -0
  69. omnicoreagent/omni_agent/deep_agent/prompts.py +956 -0
  70. omnicoreagent/omni_agent/deep_agent/subagent_factory.py +492 -0
  71. omnicoreagent/omni_agent/omni_serve/__init__.py +78 -0
  72. omnicoreagent/omni_agent/omni_serve/cli.py +523 -0
  73. omnicoreagent/omni_agent/omni_serve/config.py +174 -0
  74. omnicoreagent/omni_agent/omni_serve/lifespan.py +66 -0
  75. omnicoreagent/omni_agent/omni_serve/middleware.py +196 -0
  76. omnicoreagent/omni_agent/omni_serve/models.py +124 -0
  77. omnicoreagent/omni_agent/omni_serve/observability.py +258 -0
  78. omnicoreagent/omni_agent/omni_serve/rate_limit.py +180 -0
  79. omnicoreagent/omni_agent/omni_serve/resilience.py +387 -0
  80. omnicoreagent/omni_agent/omni_serve/routes.py +348 -0
  81. omnicoreagent/omni_agent/omni_serve/server.py +218 -0
  82. omnicoreagent/omni_agent/omni_serve/sse.py +196 -0
  83. omnicoreagent/omni_agent/prompts/prompt_builder.py +14 -0
  84. omnicoreagent/omni_agent/prompts/react_suffix.py +237 -0
  85. omnicoreagent/omni_agent/workflow/__init__.py +6 -0
  86. omnicoreagent/omni_agent/workflow/parallel_agent.py +113 -0
  87. omnicoreagent/omni_agent/workflow/router_agent.py +302 -0
  88. omnicoreagent/omni_agent/workflow/sequential_agent.py +104 -0
  89. omnicoreagent-0.3.5.dist-info/METADATA +1819 -0
  90. omnicoreagent-0.3.5.dist-info/RECORD +93 -0
  91. omnicoreagent-0.3.5.dist-info/WHEEL +4 -0
  92. omnicoreagent-0.3.5.dist-info/entry_points.txt +3 -0
  93. omnicoreagent-0.3.5.dist-info/licenses/LICENSE +21 -0
omnicoreagent/__init__.py ADDED
@@ -0,0 +1,80 @@
1
+ """
2
+ OmniCoreAgent AI Framework
3
+
4
+ A comprehensive AI agent framework with MCP client capabilities.
5
+ """
6
+
7
+ from .core.agents import ReactAgent
8
+ from .core.memory_store import MemoryRouter
9
+ from .core.llm import LLMConnection
10
+ from .core.events import EventRouter
11
+ from .core.memory_store import DatabaseMessageStore
12
+ from .core.tools import ToolRegistry, Tool
13
+ from .core.utils import logger
14
+
15
+ from .omni_agent.agent import OmniCoreAgent, OmniAgent
16
+ from .omni_agent.background_agent import (
17
+ BackgroundOmniCoreAgent,
18
+ BackgroundAgentManager,
19
+ TaskRegistry,
20
+ APSchedulerBackend,
21
+ BackgroundTaskScheduler,
22
+ )
23
+ from .omni_agent.omni_serve import OmniServe, OmniServeConfig
24
+
25
+ # OmniServe resilience and observability utilities
26
+ from .omni_agent.omni_serve.resilience import (
27
+ RetryConfig,
28
+ RetryStrategy,
29
+ CircuitBreaker,
30
+ CircuitBreakerConfig,
31
+ with_retry,
32
+ retry_async,
33
+ )
34
+ from .omni_agent.omni_serve.observability import get_metrics
35
+
36
+ from .mcp_clients_connection import MCPClient, Configuration
37
+
38
+ from .omni_agent.workflow.parallel_agent import ParallelAgent
39
+ from .omni_agent.workflow.sequential_agent import SequentialAgent
40
+ from .omni_agent.workflow.router_agent import RouterAgent
41
+
42
+ from .omni_agent.deep_agent import DeepAgent
43
+
44
+ __all__ = [
45
+ # Core
46
+ "ReactAgent",
47
+ "MemoryRouter",
48
+ "LLMConnection",
49
+ "EventRouter",
50
+ "DatabaseMessageStore",
51
+ "ToolRegistry",
52
+ "Tool",
53
+ "logger",
54
+ # Agents
55
+ "OmniCoreAgent",
56
+ "OmniAgent",
57
+ "BackgroundOmniCoreAgent",
58
+ "BackgroundAgentManager",
59
+ "TaskRegistry",
60
+ "APSchedulerBackend",
61
+ "BackgroundTaskScheduler",
62
+ "ParallelAgent",
63
+ "SequentialAgent",
64
+ "RouterAgent",
65
+ "DeepAgent",
66
+ # MCP
67
+ "MCPClient",
68
+ "Configuration",
69
+ # OmniServe
70
+ "OmniServe",
71
+ "OmniServeConfig",
72
+ # Resilience utilities
73
+ "RetryConfig",
74
+ "RetryStrategy",
75
+ "CircuitBreaker",
76
+ "CircuitBreakerConfig",
77
+ "with_retry",
78
+ "retry_async",
79
+ "get_metrics",
80
+ ]
omnicoreagent/core/__init__.py ADDED
@@ -0,0 +1,34 @@
1
+ """
2
+ Core AI Agent Framework Components
3
+
4
+ This package contains the core AI agent functionality including:
5
+ - Agents (React, Sequential)
6
+ - Memory Management (In-Memory, Redis, Database, MongoDB)
7
+ - LLM Connections and Support
8
+ - Event System
9
+ - Database Layer
10
+ - Tools Management
11
+ - Utilities and Constants
12
+ """
13
+
14
+ from .agents import ReactAgent
15
+ from .memory_store import MemoryRouter, DatabaseMessageStore
16
+ from .tools import ToolRegistry, Tool
17
+ from .types import AgentConfig, ParsedResponse, ToolCall
18
+ from .token_usage import UsageLimits, Usage, UsageLimitExceeded
19
+
20
+ __all__ = [
21
+ "ReactAgent",
22
+ "MemoryRouter",
23
+ "LLMConnection",
24
+ "EventRouter",
25
+ "DatabaseMessageStore",
26
+ "ToolRegistry",
27
+ "Tool",
28
+ "AgentConfig",
29
+ "ParsedResponse",
30
+ "ToolCall",
31
+ "UsageLimits",
32
+ "Usage",
33
+ "UsageLimitExceeded",
34
+ ]
omnicoreagent/core/agents/__init__.py ADDED
@@ -0,0 +1,22 @@
1
+ """
2
+ AI Agent Types Package
3
+
4
+ This package contains all the different types of AI agents:
5
+ - BaseReactAgent: Base class for React-style agents
6
+ - ReactAgent: Simple React agent implementation
7
+ - TokenUsage: Usage tracking and limits
8
+ """
9
+
10
+ from .base import BaseReactAgent
11
+ from .react_agent import ReactAgent
12
+
13
+ __all__ = [
14
+ "BaseReactAgent",
15
+ "ReactAgent",
16
+ "AgentConfig",
17
+ "ParsedResponse",
18
+ "ToolCall",
19
+ "UsageLimits",
20
+ "Usage",
21
+ "UsageLimitExceeded",
22
+ ]