xrtm-forecast 0.1.0__tar.gz

xrtm_forecast-0.1.0/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that you changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tour (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright 2026 XRTM Team
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

xrtm_forecast-0.1.0/PKG-INFO

@@ -0,0 +1,114 @@
+Metadata-Version: 2.4
+Name: xrtm-forecast
+Version: 0.1.0
+Summary: Institutional-grade parallelized agentic reasoning engine.
+Author-email: XRTM Team <moy@xrtm.org>
+License: Apache-2.0
+Project-URL: Repository, https://github.com/xrtm-org/forecast
+Project-URL: Issues, https://github.com/xrtm-org/forecast/issues
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: google-genai>=0.1.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: aiohttp>=3.9.0
+Requires-Dist: rich>=13.0.0
+Provides-Extra: memory
+Requires-Dist: chromadb>=0.4.0; extra == "memory"
+Provides-Extra: redis
+Requires-Dist: redis>=5.0.0; extra == "redis"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Requires-Dist: mkdocs>=1.5.0; extra == "dev"
+Requires-Dist: mkdocs-material>=9.4.0; extra == "dev"
+Requires-Dist: mkdocstrings[python]>=0.23.0; extra == "dev"
+Dynamic: license-file
+
+# xrtm-forecast
+
+Institutional-grade parallelized agentic reasoning engine.
+
+## Overview
+`xrtm-forecast` is the core intelligence engine originally developed for the CAFE (Computer-Aided Financial Engineering) platform. It provides a domain-agnostic framework for:
+- **Inference Layer**: Standardized provider interfaces for Gemini and OpenAI.
+- **Reasoning Graph**: A pluggable state-machine orchestrator for multi-agent workflows.
+- **Agent Core**: Standardized `BaseAgent` for structured reasoning and parsing.
+- **Utilities**: Distributed rate limiting (Redis) and real-time interaction telemetry.
+
+## Architectural Design: "Engine vs. Specialists"
+
+`xrtm-forecast` is designed for modularity using a "Lego" philosophy:
+- **The Engine (`agents/`)**: Core structural bricks like `LLMAgent` and `ToolAgent`.
+- **The Specialists (`agents/specialists/`)**: Pre-built expert roles like the `ForecastingAnalyst`.
+- **The Registry**: A central exchange for discovering and plugging in agents and tools.
+
+For a deep dive, see [Architecture & Design Principles](docs/architecture.md).
+
+## Installation
+
+### From PyPI (Stable)
+```bash
+pip install xrtm-forecast
+
+# With extras (redis, memory)
+pip install "xrtm-forecast[redis,memory]"
+```
+
+### From Source (Latest)
+```bash
+pip install git+https://github.com/xrtm-org/forecast.git
+```
+
+## Configuration
+
+`xrtm-forecast` relies on environment variables for API keys and service connections. Create a `.env` file in your project root:
+
+```bash
+GEMINI_API_KEY=your_key_here
+REDIS_URL=redis://localhost:6379/0  # Optional (fallback to in-memory)
+```
+
+## Quick Start: Inference
+
+```python
+import asyncio
+from forecast import ModelFactory
+from forecast.inference.config import GeminiConfig
+from pydantic import SecretStr
+
+async def main():
+    config = GeminiConfig(
+        api_key=SecretStr("your-key"), 
+        model_id="gemini-2.0-flash-lite"
+    )
+    provider = ModelFactory.get_provider(config)
+    response = await provider.generate_content_async("What is the causality of inflation?")
+    print(response.text)
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+## Documentation & Examples
+
+- **Architecture**: [The "Lego" Design](docs/architecture.md)
+- **Agent Registry**: [Pre-built & Core Agents](docs/agents_registry.md)
+- **Examples**: Check the `examples/` directory for full pipelines (Analyst, Custom Graphs, etc.)
+
+## Contributing
+
+We welcome institutional-grade contributions! Please see [CONTRIBUTING.md](CONTRIBUTING.md) for setup instructions and our development workflow.
+
+
+---
+
+## License
+
+`xrtm-forecast` is open-source software licensed under the **Apache-2.0** license. See the [LICENSE](LICENSE) file for more details.
+
+Copyright © 2026 XRTM Team.

xrtm_forecast-0.1.0/README.md

@@ -0,0 +1,83 @@
+# xrtm-forecast
+
+Institutional-grade parallelized agentic reasoning engine.
+
+## Overview
+`xrtm-forecast` is the core intelligence engine originally developed for the CAFE (Computer-Aided Financial Engineering) platform. It provides a domain-agnostic framework for:
+- **Inference Layer**: Standardized provider interfaces for Gemini and OpenAI.
+- **Reasoning Graph**: A pluggable state-machine orchestrator for multi-agent workflows.
+- **Agent Core**: Standardized `BaseAgent` for structured reasoning and parsing.
+- **Utilities**: Distributed rate limiting (Redis) and real-time interaction telemetry.
+
+## Architectural Design: "Engine vs. Specialists"
+
+`xrtm-forecast` is designed for modularity using a "Lego" philosophy:
+- **The Engine (`agents/`)**: Core structural bricks like `LLMAgent` and `ToolAgent`.
+- **The Specialists (`agents/specialists/`)**: Pre-built expert roles like the `ForecastingAnalyst`.
+- **The Registry**: A central exchange for discovering and plugging in agents and tools.
+
+For a deep dive, see [Architecture & Design Principles](docs/architecture.md).
+
+## Installation
+
+### From PyPI (Stable)
+```bash
+pip install xrtm-forecast
+
+# With extras (redis, memory)
+pip install "xrtm-forecast[redis,memory]"
+```
+
+### From Source (Latest)
+```bash
+pip install git+https://github.com/xrtm-org/forecast.git
+```
+
+## Configuration
+
+`xrtm-forecast` relies on environment variables for API keys and service connections. Create a `.env` file in your project root:
+
+```bash
+GEMINI_API_KEY=your_key_here
+REDIS_URL=redis://localhost:6379/0  # Optional (fallback to in-memory)
+```
+
+## Quick Start: Inference
+
+```python
+import asyncio
+from forecast import ModelFactory
+from forecast.inference.config import GeminiConfig
+from pydantic import SecretStr
+
+async def main():
+    config = GeminiConfig(
+        api_key=SecretStr("your-key"), 
+        model_id="gemini-2.0-flash-lite"
+    )
+    provider = ModelFactory.get_provider(config)
+    response = await provider.generate_content_async("What is the causality of inflation?")
+    print(response.text)
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+## Documentation & Examples
+
+- **Architecture**: [The "Lego" Design](docs/architecture.md)
+- **Agent Registry**: [Pre-built & Core Agents](docs/agents_registry.md)
+- **Examples**: Check the `examples/` directory for full pipelines (Analyst, Custom Graphs, etc.)
+
+## Contributing
+
+We welcome institutional-grade contributions! Please see [CONTRIBUTING.md](CONTRIBUTING.md) for setup instructions and our development workflow.
+
+
+---
+
+## License
+
+`xrtm-forecast` is open-source software licensed under the **Apache-2.0** license. See the [LICENSE](LICENSE) file for more details.
+
+Copyright © 2026 XRTM Team.

xrtm_forecast-0.1.0/pyproject.toml

@@ -0,0 +1,66 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "xrtm-forecast"
+version = "0.1.0"
+description = "Institutional-grade parallelized agentic reasoning engine."
+readme = "README.md"
+requires-python = ">=3.11"
+license = {text = "Apache-2.0"}
+authors = [
+    {name = "XRTM Team", email = "moy@xrtm.org"}
+]
+dependencies = [
+    "pydantic>=2.0.0",
+    "google-genai>=0.1.0",
+    "openai>=1.0.0",
+    "aiohttp>=3.9.0",
+    "rich>=13.0.0"
+]
+
+[project.urls]
+Repository = "https://github.com/xrtm-org/forecast"
+Issues = "https://github.com/xrtm-org/forecast/issues"
+
+[project.optional-dependencies]
+memory = [
+    "chromadb>=0.4.0"
+]
+redis = [
+    "redis>=5.0.0"
+]
+dev = [
+    "pytest>=7.0.0",
+    "pytest-asyncio>=0.21.0",
+    "pytest-cov>=4.1.0",
+    "ruff>=0.1.0",
+    "mypy>=1.0.0",
+    "mkdocs>=1.5.0",
+    "mkdocs-material>=9.4.0",
+    "mkdocstrings[python]>=0.23.0"
+]
+
+[tool.pytest.ini_options]
+pythonpath = ["."]
+testpaths = ["tests"]
+asyncio_mode = "strict"
+
+[tool.setuptools]
+package-dir = {"" = "src"}
+packages = {find = {where = ["src"]}}
+
+[tool.ruff]
+line-length = 120
+target-version = "py311"
+exclude = [".git", ".venv", "build", "dist"]
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "W"]
+ignore = ["E501"]
+
+[tool.mypy]
+python_version = "3.11"
+ignore_missing_imports = true
+strict = false

xrtm_forecast-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

xrtm_forecast-0.1.0/src/forecast/__init__.py

@@ -0,0 +1,33 @@
+from forecast.agents.base import Agent
+from forecast.agents.graph import GraphAgent
+from forecast.agents.llm import LLMAgent
+from forecast.agents.registry import registry
+from forecast.agents.specialists.analyst import ForecastingAnalyst
+from forecast.agents.tool import ToolAgent
+from forecast.graph.orchestrator import Orchestrator
+from forecast.inference.factory import ModelFactory
+from forecast.memory.unified import Memory
+from forecast.schemas.graph import BaseGraphState
+from forecast.telemetry.audit import auditor
+from forecast.tools.registry import tool_registry
+
+__all__ = [
+    "Agent",
+    "ForecastingAnalyst",
+    "registry",
+    "LLMAgent",
+    "ToolAgent",
+    "GraphAgent",
+    "Orchestrator",
+    "BaseGraphState",
+    "ModelFactory",
+    "Memory",
+    "auditor",
+    "tool_registry",
+]
+
+__version__ = "0.1.0"
+__author__ = "XRTM Team"
+__contact__ = "moy@xrtm.org"
+__license__ = "Apache-2.0"
+__copyright__ = "Copyright 2026 XRTM Team"

xrtm_forecast-0.1.0/src/forecast/agents/base.py

@@ -0,0 +1,34 @@
+import abc
+import logging
+from typing import Any, Dict, Optional, TypeVar
+
+from pydantic import BaseModel
+
+logger = logging.getLogger(__name__)
+
+T = TypeVar("T", bound=BaseModel)
+
+
+class Agent(abc.ABC):
+    """
+    Fundamental Lego piece of the xrtm-forecast engine.
+    Everything (LLMs, Tools, Graphs) is an Agent.
+    """
+    def __init__(self, name: Optional[str] = None):
+        self.name = name or self.__class__.__name__
+
+    @abc.abstractmethod
+    async def run(self, input_data: Any, **kwargs) -> Any:
+        """
+        Asynchronous execution of the agent's logic.
+        Ideally uses Pydantic models for input/output.
+        """
+        pass
+
+    def get_info(self) -> Dict[str, Any]:
+        """Returns metadata about the agent for structural tracing."""
+        return {
+            "name": self.name,
+            "type": self.__class__.__name__,
+            "version": "0.1.0" # Placeholder for versioning logic
+        }

xrtm_forecast-0.1.0/src/forecast/agents/graph.py

@@ -0,0 +1,30 @@
+from typing import Any, Optional
+
+from forecast.agents.base import Agent
+from forecast.graph.orchestrator import Orchestrator
+from forecast.schemas.graph import BaseGraphState
+
+
+class GraphAgent(Agent):
+    """
+    Agent that encapsulates a sub-graph (Orchestrator).
+    Enables 'Composite Nodes' where an agent internally runs a whole pipeline.
+    """
+    def __init__(self, orchestrator: Orchestrator, entry_node: str, name: Optional[str] = None):
+        super().__init__(name)
+        self.orchestrator = orchestrator
+        self.entry_node = entry_node
+
+    async def run(self, input_data: Any, **kwargs) -> Any:
+        """
+        Runs the internal orchestrator.
+        Converts input_data into a BaseGraphState context.
+        """
+        # Create a fresh state for the sub-graph
+        state = BaseGraphState(subject_id=f"subgraph_{self.name}")
+        state.context = {"input": input_data, **kwargs}
+
+        await self.orchestrator.run(state, entry_node=self.entry_node)
+
+        # Return the final context or a specific output if defined in state
+        return state.context.get("output", state.context)

xrtm_forecast-0.1.0/src/forecast/agents/llm.py

@@ -0,0 +1,59 @@
+import logging
+from typing import Any, Dict, List, Optional, Type, TypeVar, Union
+
+from pydantic import BaseModel
+
+from forecast.agents.base import Agent
+from forecast.inference.base import InferenceProvider
+from forecast.tools.registry import tool_registry
+from forecast.utils.parser import parse_json_markdown
+
+logger = logging.getLogger(__name__)
+T = TypeVar("T", bound=BaseModel)
+
+class LLMAgent(Agent):
+    """
+    Agent that utilizes an LLM (InferenceProvider) for reasoning.
+    """
+    def __init__(self, model: InferenceProvider, name: Optional[str] = None):
+        super().__init__(name)
+        self.model = model
+
+    def get_tools(self, tool_names: Optional[List[str]] = None) -> List[Any]:
+        """
+        Retrieves tool instances from the global registry.
+        If tool_names is None, returns all registered tools.
+        """
+        if tool_names is None:
+            # Return all wrapped tool objects
+            return [tool_registry.get_tool(name) for name in tool_registry.list_tools()]
+
+        tools = []
+        for name in tool_names:
+            tool = tool_registry.get_tool(name)
+            if tool:
+                tools.append(tool)
+        return tools
+
+    def parse_output(
+        self, text: str, schema: Optional[Type[T]] = None, default: Any = None
+    ) -> Union[T, Dict[str, Any]]:
+        """Parses LLM text into a Pydantic model or dict."""
+        parsed = parse_json_markdown(text)
+        if parsed is None:
+            return default if default is not None else {}
+
+        if schema and isinstance(parsed, dict):
+            try:
+                # Filter out keys not in schema to prevent validation errors
+                valid_fields = schema.model_fields.keys()
+                filtered = {k: v for k, v in parsed.items() if k in valid_fields}
+                return schema(**filtered)
+            except Exception as e:
+                logger.warning(f"Schema validation failed: {e}")
+                return default if default is not None else parsed
+        return parsed
+
+    async def run(self, input_data: Any, **kwargs) -> Any:
+        # LLMAgents will typically override this with specific prompting logic
+        raise NotImplementedError("LLMAgent subclasses must implement run")

xrtm_forecast-0.1.0/src/forecast/agents/registry.py

@@ -0,0 +1,45 @@
+import logging
+from typing import Dict, List, Optional, Union
+
+from forecast.agents.base import Agent
+
+logger = logging.getLogger(__name__)
+
+
+class AgentRegistry:
+    """
+    Handles explicit registration and discovery of specialist agents.
+    Enables modular, multi-agent reasoning swarms.
+    """
+    def __init__(self):
+        self._agents: Dict[str, Agent] = {}
+
+    def register(self, agent: Union[Agent, type], name: Optional[str] = None):
+        """Registers an agent (instance or class) in the registry."""
+        if isinstance(agent, type):
+            # If it's a class, we'll instantiate it if we can
+            # but usually we want instances registered with specific configs
+            agent = agent()
+
+        # Mypy safe cast
+        from typing import cast
+        agent_instance = cast(Agent, agent)
+
+        name = name or agent_instance.name
+        self._agents[name.lower()] = agent_instance
+        logger.info(f"[REGISTRY] Registered agent: {name.upper()}")
+
+    def get_agent(self, name: str) -> Optional[Agent]:
+        """Retrieves an agent by name."""
+        return self._agents.get(name.lower())
+
+    def list_agents(self) -> List[str]:
+        """Returns a list of all registered agent names."""
+        return list(self._agents.keys())
+
+    # Compatibility Aliases for legacy specialist support
+    def get_specialist(self, domain: str) -> Optional[Agent]:
+        return self.get_agent(domain)
+
+# Global Singleton instance for the library
+registry = AgentRegistry()