contextagent 0.1.0__py3-none-any.whl

contextagent-0.1.0.dist-info/METADATA

@@ -0,0 +1,269 @@
+Metadata-Version: 2.4
+Name: contextagent
+Version: 0.1.0
+Summary: A Research-Oriented Multi-Agent System Platform
+Author-email: Zhimeng Guo <gzjz07@outlook.com>
+License: MIT
+Project-URL: Homepage, https://github.com/TimeLovercc/agentz
+Project-URL: Repository, https://github.com/TimeLovercc/agentz
+Project-URL: Documentation, https://github.com/TimeLovercc/agentz#readme
+Project-URL: Bug Tracker, https://github.com/TimeLovercc/agentz/issues
+Keywords: agent,multi-agent,ai,llm,research,machine-learning,autonomous-agents
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: build>=1.3.0
+Requires-Dist: google-generativeai>=0.8.5
+Requires-Dist: ipdb>=0.13.13
+Requires-Dist: litellm==1.76
+Requires-Dist: loguru>=0.7.3
+Requires-Dist: matplotlib>=3.10.6
+Requires-Dist: numpy>=2.3.3
+Requires-Dist: openai-agents>=0.3.2
+Requires-Dist: pandas>=2.3.3
+Requires-Dist: rich>=14.1.0
+Requires-Dist: scikit-learn>=1.7.2
+Requires-Dist: seaborn>=0.13.2
+Requires-Dist: twine>=6.2.0
+Requires-Dist: weave>=0.52.8
+Dynamic: license-file
+
+<div align="center">
+
+# AgentZ: Agent from Zero
+
+**A Research-Oriented Multi-Agent System Platform**
+
+</div>
+
+AgentZ is a minimal, extensible codebase for multi-agent systems research. Build intelligent agent workflows with minimal code while achieving strong baseline performance. The platform enables autonomous reasoning, experience learning, and dynamic tool creation - providing both a comparative baseline and production-ready foundation for multi-agent research.
+
+## Features
+
+- **🎯 Minimal Implementation** - Build new systems with just a few lines of code
+- **🔄 Stateful Workflows** - Persistent memory and object management throughout agent lifecycle
+- **📚 Experience Learning** - Agents improve over time through memory-based reasoning
+- **🛠️ Dynamic Tool Creation** - Agents can generate and use custom tools on-demand
+- **🧠 Autonomous Reasoning** - Built-in cognitive capabilities for complex multi-step tasks
+- **⚙️ Config-Driven** - Easily modify behavior through configuration files
+
+## Installation
+
+This project uses [uv](https://docs.astral.sh/uv/) for fast, reliable package management.
+
+### Install uv
+
+```bash
+# macOS/Linux
+curl -LsSf https://astral.sh/uv/install.sh | sh
+
+# Or via pip
+pip install uv
+```
+
+See the [uv installation guide](https://docs.astral.sh/uv/getting-started/installation/) for more options.
+
+### Setup Environment
+
+```bash
+# Clone the repository
+git clone https://github.com/yourusername/agentz.git
+cd agentz
+
+# Sync dependencies
+uv sync
+```
+
+## Quick Start
+
+```python
+from pipelines.data_scientist import DataScientistPipeline
+
+pipe = DataScientistPipeline("pipelines/configs/data_science.yaml")
+
+pipe.run_sync()
+```
+
+## Building Your Own System
+
+### 1. Create a Custom Pipeline
+
+Inherit from `BasePipeline` to create your own agent workflow:
+
+```python
+from pipelines.base import BasePipeline
+
+
+class MyCustomPipeline(BasePipeline):
+    DEFAULT_CONFIG_PATH = "pipelines/configs/my_pipeline.yaml"
+
+    def __init__(self, config=None):
+        super().__init__(config)
+        # Add your custom initialization
+
+    async def run(self):
+        # Implement your workflow logic
+        pass
+```
+
+### 2. Add Custom Agents
+
+Implement your agents following the standard interface:
+
+```python
+from agents import Agent
+
+def create_my_agent(config):
+    return Agent(
+        name="my_agent",
+        instructions="Your agent instructions here",
+        model=config.main_model
+    )
+```
+
+### 3. Configure & Run
+
+Create a config file and run your pipeline:
+
+```python
+pipe = MyCustomPipeline(
+    data_path="your_data.csv",
+    user_prompt="Your task description",
+    provider="gemini",
+    model="gemini-2.5-flash"
+)
+
+pipe.run_sync()
+```
+
+## Architecture
+
+AgentZ is organised around a **central conversation state** and a set of declarative
+flow specifications that describe how agents collaborate. The main
+components you will interact with are:
+
+- **`pipelines/`** – High level orchestration that wires agents together.
+- **`agentz/agents/`** – Capability definitions for manager agents and tool agents.
+- **`agentz/flow/`** – Flow primitives (`FlowRunner`, `FlowNode`, `IterationFlow`) that
+  execute declarative pipelines.
+- **`agentz/memory/`** – Structured state management (`ConversationState`,
+  `ToolExecutionResult`, global memory helpers).
+- **`examples/`** – Example scripts showing end-to-end usage.
+
+```
+agentz/
+├── pipelines/
+│   ├── base.py               # Base pipeline with config management & helpers
+│   ├── flow_runner.py        # Declarative flow executor utilities
+│   └── data_scientist.py     # Reference research pipeline
+├── agentz/
+│   ├── agents/
+│   │   ├── manager_agents/   # Observe, evaluate, routing, writer agents
+│   │   └── tool_agents/      # Specialised tool executors
+│   ├── flow/                 # Flow node definitions and runtime objects
+│   ├── memory/               # Conversation state & persistence utilities
+│   ├── llm/                  # LLM adapters and setup helpers
+│   └── tools/                # Built-in tools
+└── examples/
+    └── data_science.py       # Example workflows
+```
+
+### Declarative Pipeline Flow
+
+The reference `DataScientistPipeline` models an entire research workflow using
+three building blocks:
+
+1. **Central ConversationState** – A shared store that captures every field any
+   agent might read or write (iteration metadata, gaps, observations, tool
+   results, timing, final report, etc.). Each loop creates a new
+   `IterationRecord`, enabling partial updates and clean tracking of tool
+   outcomes.
+2. **Structured IO Contracts** – Each agent step declares the Pydantic model it
+   expects and produces (for example `KnowledgeGapOutput` or
+   `AgentSelectionPlan`). Input builders map slices of `ConversationState` into
+   those models and output handlers merge the validated results back into the
+   central state.
+3. **Declarative FlowRunner** – The pipeline defines an `IterationFlow` of
+   `FlowNode`s such as observe → evaluate → route → tools. Loop and termination
+   logic are expressed with predicates (`loop_condition`, `condition`), so the
+   executor can stop when evaluation marks `state.complete` or constraints are
+   reached. Finalisation steps (like the writer agent) run after the iteration
+   loop using the same structured IO.
+
+Because the flow is declarative and all state is centralised, extending the
+pipeline is as simple as adding a new node, output field, or tool capability—no
+custom `run()` logic is required beyond sequencing the flow runner.
+
+## Benchmarks
+
+AgentZ has been verified on several benchmarks for multi-agent research:
+
+- **Data Science Tasks**: State-of-the-art performance on automated ML pipelines
+- **Complex Reasoning**: Competitive results on multi-step reasoning benchmarks
+- **Tool Usage**: High accuracy in dynamic tool selection and execution
+
+*Detailed benchmark results and comparisons coming soon.*
+
+## Roadmap
+
+- [x] Persistence Process - Stateful agent workflows
+- [x] Experience Learning - Memory-based reasoning
+- [x] Tool Design - Dynamic tool creation
+- [ ] Workflow RAG - Retrieval-augmented generation for complex workflows
+- [ ] MCPs - Model Context Protocol support for enhanced agent communication
+
+## Key Design Principles
+
+1. **Minimal Core** - Keep the base system simple and extensible
+2. **Intelligent Defaults** - Provide strong baseline implementations
+3. **Research-First** - Design for experimentation and comparison
+4. **Modular Architecture** - Easy to swap components and test variations
+5. **Production-Ready** - Scale from research prototypes to deployed systems
+
+## Use Cases
+
+- **Multi-Agent Research** - Baseline for comparing agent architectures
+- **Automated Data Science** - End-to-end ML pipeline automation
+- **Complex Task Decomposition** - Break down and solve multi-step problems
+- **Tool-Using Agents** - Research on dynamic tool creation and usage
+- **Agent Memory Systems** - Study persistence and experience learning
+
+## Citation
+
+If you use AgentZ in your research, please cite:
+
+```bibtex
+@software{agentz2025,
+  title={AgentZ: A Research-Oriented Multi-Agent System Platform},
+  author={Your Name},
+  year={2025},
+  url={https://github.com/yourusername/agentz}
+}
+```
+
+## Contributing
+
+We welcome contributions! AgentZ is designed to be a community resource for multi-agent research. Please open an issue or submit a pull request.
+
+## License
+
+[Your License Here]
+
+## Acknowledgements
+
+AgentZ is built with inspiration from the multi-agent systems research community. We thank the developers of various LLM frameworks and tools that make this work possible.
+
+---
+
+<div align="center">
+
+**AgentZ**: Building intelligent agents from zero to hero 🚀
+
+</div>

contextagent-0.1.0.dist-info/RECORD

@@ -0,0 +1,66 @@
+agentz/agent/base.py,sha256=YU_SRXJ3KV-rSWADpjJNVONm3uf_B0c4YnAsqm6ICwg,9986
+agentz/artifacts/__init__.py,sha256=rKlnz1a9kNHfssKtLvCxMUH9jN4oezLRcHh2tWKb60Q,107
+agentz/artifacts/artifact_writer.py,sha256=55gABvX8_5-sWHBwtV5FaxcwsHiqvY-G9MoOTY10uBM,17670
+agentz/artifacts/reporter.py,sha256=aKEUy_7W1Jjto4k1UOT6TLhHhk431jGT6Ytw4j0jLJY,7084
+agentz/artifacts/terminal_writer.py,sha256=EfvNL9blWCiuZz8EAGGCYQlCIYFmvng2aOoZAsm4Ojk,3310
+agentz/context/__init__.py,sha256=v2uECoSmcUhdtiOVzfJnyLPoqu94tzlF84KEFEe2mYg,202
+agentz/context/context.py,sha256=-RWvRfTKcXtqIqrwLuad6EePZWUOPQMsQRR4jl4oB_A,3534
+agentz/context/conversation.py,sha256=G_rSBO3mi5rD8PSGmkh7m-Wb87XW8I-WxJIYS2RoP3M,7691
+agentz/context/data_store.py,sha256=8E40xigsAI4k7iiedFIH8V5Ry0oZXMCRAN6HpUIV1rU,6082
+agentz/llm/llm_setup.py,sha256=U7w-6kw058uxPDq8OXiAFu1P_eo8Wops40ghtyMSshk,6001
+agentz/mcp/manager.py,sha256=snrQSvU8rcC1szOjrQWoHVHKiphtupX9fATMTcJVrQM,4849
+agentz/mcp/patches.py,sha256=Vmo5AYDScIWn33QssP6Zxr3LbHnoSbv0IiSYz-_PwWE,2592
+agentz/mcp/servers/chrome_devtools/server.py,sha256=-_q6lXLPVXZctU2qgPeoUZb1CFtDEPPl0CH71_D80IA,361
+agentz/profiles/base.py,sha256=0gMFY3kiSoV3I6N2GCxHJzPXzrFFaw3n-IS4Oysbkf0,4441
+agentz/profiles/data/data_analysis.py,sha256=Ns8E2pyjLGSWxNvjNGWdLhvI84whPi0N3JiprTw7yM4,1376
+agentz/profiles/data/data_loader.py,sha256=dsHkBzEKc7a_pPPGOGBqdeE0Xvq_sVHp0iK3NZ-k-jU,1089
+agentz/profiles/data/evaluation.py,sha256=Q9zHcHKZ0vPkK69bunr40_lSLFMoavi22wTlxtg-p3c,1504
+agentz/profiles/data/model_training.py,sha256=UkCVsfD64iL4Q_Z2AV0FtIejkK9W34RS3L6JjK7ZgZ0,1584
+agentz/profiles/data/preprocessing.py,sha256=tAVZZUdz6_FlEor1Y81YnZH57aS8WyMJMewU9AVcnEo,1626
+agentz/profiles/data/visualization.py,sha256=2mnvyvBrloOOKqo050hn9z_g1O-GAk5CeRaZ74jMWgk,1633
+agentz/profiles/manager/evaluate.py,sha256=l2gv_LYTBOmSkoIVGmlx1SEkCjrGIBh6HEUI0GvGanc,1946
+agentz/profiles/manager/memory.py,sha256=eXYd_hI--KnOrRB1pkg0VWp4OSjD4wtKHPtH0ngiTBI,2501
+agentz/profiles/manager/observe.py,sha256=Y9JlZFkRpzqgkAynkdvE3Hf8YrVAXZDs4LsJIOgLZCc,1615
+agentz/profiles/manager/routing.py,sha256=M3pkR_Ric2uaNwSlOXfQ8OHeIdFhkPJMJLIShn4jJVw,2704
+agentz/profiles/manager/writer.py,sha256=8yzCG_HmKMU42NxkPVnk-r7mKgjkX3zJMiDwfIyyzME,1863
+agentz/profiles/mcp/browser.py,sha256=USFGzm3habnVEIqnGUSeRYhKhWbO0KHPuPUl1A-c0eU,625
+agentz/profiles/mcp/chrome.py,sha256=DfiktKJqT2iGwLLWU1aQvOrOcOmEZCEqBmIpCAz3E1Q,618
+agentz/profiles/mcp/notion.py,sha256=mSCE1ITZgn38th8UkfCeNWyXczM3N4CBve4e_h5MVuU,621
+agentz/runner/__init__.py,sha256=es55WDHKWENMTuhRShWlKop800wr6fHGUEXnrABNgdw,2302
+agentz/runner/base.py,sha256=4jKcMO4c6xu0QJfOKLodAbvRauPS17ygV6eGoHUTd1I,909
+agentz/runner/executor.py,sha256=XAzThcpu_2zj3SG4hdrEdRPnd83wMAdrzXI7qoMxVjs,12302
+agentz/runner/hooks.py,sha256=OglEkfgy1Yj8OA-smAGwEy7JltPUtrigE4jj-mHkeAU,3306
+agentz/runner/iteration.py,sha256=X9gwdE2Ivrp1uG73Ky-QV3_pXam--7To1sBP9neUMOs,4542
+agentz/runner/patterns.py,sha256=sv8IPvfaKQyxvi-iRNGJC-mJV6jdx9Gcl7duHizhHn0,7447
+agentz/runner/tracker.py,sha256=3zIzHTleaI8k8sIbKzI5-60yMoTYRkLBTx7OISjgvMQ,6166
+agentz/runner/utils.py,sha256=xgICIclYodAKTfdYvuY4RB0k-y1ihcGczu1_m4ASyLI,1292
+agentz/runner/workflow.py,sha256=jHmGJSKU_TXjHdkiroKsEqN5RdnHuhMlo7y0EBxcYBM,8330
+agentz/tools/__init__.py,sha256=CxJtpTcPHpA4106VlpiiAYgdlGLoEejOZWlR9NNSZ0k,379
+agentz/tools/data_tools/__init__.py,sha256=nSEvokYw8h61iE6mbusOYWH0pNoOm3DmHDmmdf3eSgE,485
+agentz/tools/data_tools/data_analysis.py,sha256=M-eZkr9WtaX9Eb6siuhA9TnKX-LYAJteICpJjhf2nSU,6592
+agentz/tools/data_tools/data_loading.py,sha256=fGid76Vz3ayS-suNaKjO8eL-eKAzNsnR73qMpw3KE8c,3681
+agentz/tools/data_tools/evaluation.py,sha256=mujpvk_LhnjnlMYcwQ7_rw_z_7pi-7vIB90VD6UozGI,7255
+agentz/tools/data_tools/helpers.py,sha256=l14vbOrwa67zxwIVAnfEKkj20qxpEgYrEmPh6VE07Es,3903
+agentz/tools/data_tools/model_training.py,sha256=G68ZqpdBfX40t1s2aF2dFMxP4hlp8RATUshUg7gyYC0,7934
+agentz/tools/data_tools/preprocessing.py,sha256=FEQ9DdAy49GuwdfAT4o-yyQWSHxs3WESCd4Byouu5ks,9991
+agentz/tools/data_tools/visualization.py,sha256=sVm8dPB6Obri_bjcfYYSatHgZqjLYSPojiBVsQafPU4,10966
+agentz/utils/__init__.py,sha256=YKjsYeJZqxeBBOeMVTAaKMgblRzkE3MKDA22bYJuNtw,1518
+agentz/utils/config.py,sha256=t_JvHhHAyJSduUi9gBogRXD1zXrVOPK4mwjCMUXLuP8,24118
+agentz/utils/helpers.py,sha256=d03wracF2T7jaz1LCJ0Q7aN21X6aFG774EPFjE-B8DI,206
+agentz/utils/parsers.py,sha256=lLAIYbLCLdqN9ER4Jc5ee41KO0EJgI1mTR50m-qsnSU,4911
+agentz/utils/printer.py,sha256=0xG_ObfYFh3wbKH3ysT2dCLLUCl6Eg1Qnd9QJ6nxerw,19960
+contextagent-0.1.0.dist-info/licenses/LICENSE,sha256=EEIv2C8nusNfEwOuJWmmC7X0fLziF7BlXEdDuQRB_NM,1068
+pipelines/base.py,sha256=YgKHOLiFkODdeTdPCBGFu4tkcgCvJ4-YbrNOXW64-YE,33357
+pipelines/data_scientist.py,sha256=D6-9iqMd8UCYN4PNLXZiJ5Zwvn5YPq4NSBVqOZI7D7g,3553
+pipelines/data_scientist_memory.py,sha256=SL1ZfikAxE26NUMXFClvagcWMp85FNRfef33UTCIjog,7021
+pipelines/experience_learner.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pipelines/prompt_generator.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pipelines/simple.py,sha256=JBrFZlZCmFHFzvf9afjeOpqf1O7D4IaLf3cm91IR35U,2561
+pipelines/simple_browser.py,sha256=vaqiU6hGc7r_7NsiLW1EFATabBBcrfzJEYy4q7FCoec,5180
+pipelines/simple_chrome.py,sha256=L4Fua0fUckoN6u6e5o3l21QZipL-FakjswVNR0LM8AM,2415
+pipelines/simple_notion.py,sha256=5HyLsotAAQyKq6YUzNnMyjFz_Ar8ewb8DhVarH-oULQ,3444
+pipelines/tool_builder.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+contextagent-0.1.0.dist-info/METADATA,sha256=15iOz150dcXZ_B8w9roXwlBLAgFiseZsT2hnKLgElIo,9246
+contextagent-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+contextagent-0.1.0.dist-info/top_level.txt,sha256=4tj4T9doPJLNCS3txrtXoGgRmQynh0pN-01pfLX7-2c,17
+contextagent-0.1.0.dist-info/RECORD,,

contextagent-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

contextagent-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Zhimeng Guo
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

contextagent-0.1.0.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+agentz
+pipelines