agentstate-lib 0.1.0__tar.gz

agentstate_lib-0.1.0/.gitignore

@@ -0,0 +1,226 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[codz]
+*$py.class
+
+
+# db
+*.db
+
+.checkpoints/
+examples/benchmark/results/*.json
+.ngrok/
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#   Usually these files are written by a python script from a template
+#   before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py.cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+# Pipfile.lock
+
+# UV
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+# uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+# poetry.lock
+# poetry.toml
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#   pdm recommends including project-wide configuration in pdm.toml, but excluding .pdm-python.
+#   https://pdm-project.org/en/latest/usage/project/#working-with-version-control
+# pdm.lock
+# pdm.toml
+.pdm-python
+.pdm-build/
+
+# pixi
+#   Similar to Pipfile.lock, it is generally recommended to include pixi.lock in version control.
+# pixi.lock
+#   Pixi creates a virtual environment in the .pixi directory, just like venv module creates one
+#   in the .venv directory. It is recommended not to include this directory in version control.
+.pixi
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# Redis
+*.rdb
+*.aof
+*.pid
+
+# RabbitMQ
+mnesia/
+rabbitmq/
+rabbitmq-data/
+
+# ActiveMQ
+activemq-data/
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.envrc
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#   JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#   be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#   and can be added to the global gitignore or merged into this file.  For a more nuclear
+#   option (not recommended) you can uncomment the following to ignore the entire idea folder.
+# .idea/
+
+# Abstra
+#   Abstra is an AI-powered process automation framework.
+#   Ignore directories containing user credentials, local state, and settings.
+#   Learn more at https://abstra.io/docs
+.abstra/
+
+# Visual Studio Code
+#   Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore 
+#   that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
+#   and can be added to the global gitignore or merged into this file. However, if you prefer, 
+#   you could uncomment the following to ignore the entire vscode folder
+.vscode/
+# Temporary file for partial code execution
+tempCodeRunnerFile.py
+
+# Ruff stuff:
+.ruff_cache/
+
+# PyPI configuration file
+.pypirc
+
+# Marimo
+marimo/_static/
+marimo/_lsp/
+__marimo__/
+
+# Streamlit
+.streamlit/secrets.toml

agentstate_lib-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Tanveer kaur
+
+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.

agentstate_lib-0.1.0/PKG-INFO

@@ -0,0 +1,33 @@
+Metadata-Version: 2.4
+Name: agentstate_lib
+Version: 0.1.0
+Summary: Stateful coordination layer for multi-agent AI systems
+Author-email: Tanveer Kaur <tanveerkaur1292@outlook.com>
+License: MIT
+License-File: LICENSE
+Keywords: agents,llm,multi-agent,orchestration,state
+Requires-Python: >=3.11
+Requires-Dist: aiosqlite>=0.19
+Requires-Dist: pydantic>=2.0
+Provides-Extra: api
+Requires-Dist: fastapi>=0.100; extra == 'api'
+Requires-Dist: uvicorn[standard]>=0.20; extra == 'api'
+Provides-Extra: contrib
+Provides-Extra: dashboard
+Requires-Dist: rich>=13.0; extra == 'dashboard'
+Provides-Extra: dev
+Requires-Dist: httpx>=0.24; extra == 'dev'
+Requires-Dist: hypothesis>=6.0; extra == 'dev'
+Requires-Dist: mypy>=1.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.21; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == 'dev'
+Requires-Dist: ruff>=0.1; extra == 'dev'
+Provides-Extra: otel
+Requires-Dist: opentelemetry-api; extra == 'otel'
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc; extra == 'otel'
+Requires-Dist: opentelemetry-sdk; extra == 'otel'
+Description-Content-Type: text/markdown
+
+# agentstate
+Building a stateful coordination layer for multi-agent AI systems.
+

agentstate_lib-0.1.0/README.md

@@ -0,0 +1,3 @@
+# agentstate
+Building a stateful coordination layer for multi-agent AI systems.
+

agentstate_lib-0.1.0/agentstate/__init__.py

@@ -0,0 +1,63 @@
+from __future__ import annotations
+
+from agentstate.core.events import (
+    AgentErrored,
+    BaseStateEvent,
+    CheckpointSaved,
+    ConflictDetected,
+    PatchApplied,
+    StateEvent,
+    WorkflowCompleted,
+    WorkflowStarted,
+    event_adapter,
+)
+from agentstate.core.patch import (
+    StatePatch,
+    apply_patch,
+    get_nested,
+    set_nested,
+)
+from agentstate.core.state import (
+    Artifact,
+    Decision,
+    Goal,
+    SharedState,
+    Task,
+    WorkflowStatus,
+)
+from agentstate.memory.store import InMemoryStore, SQLiteStore, StateStore
+from agentstate.router.context import slice_state
+from agentstate.router.graph import AgentGraph
+from agentstate.router.types import AgentFn, EdgeCondition
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "SharedState",
+    "Task",
+    "Goal",
+    "Artifact",
+    "Decision",
+    "WorkflowStatus",
+    "StatePatch",
+    "apply_patch",
+    "set_nested",
+    "get_nested",
+    "StateEvent",
+    "BaseStateEvent",
+    "WorkflowStarted",
+    "WorkflowCompleted",
+    "PatchApplied",
+    "ConflictDetected",
+    "CheckpointSaved",
+    "AgentErrored",
+    "event_adapter",
+    "AgentGraph",
+    "AgentFn",
+    "EdgeCondition",
+    "slice_state",
+    "StateStore",
+    "InMemoryStore",
+    "SQLiteStore",
+    "__version__",
+]

agentstate_lib-0.1.0/agentstate/core/__init__.py

@@ -0,0 +1,30 @@
+from .events import (
+    AgentErrored,
+    BaseStateEvent,
+    CheckpointSaved,
+    ConflictDetected,
+    PatchApplied,
+    StateEvent,
+    WorkflowCompleted,
+    WorkflowStarted,
+    event_adapter,
+)
+from .state import Artifact, Decision, Goal, SharedState, Task, WorkflowStatus
+
+__all__ = [
+    "SharedState",
+    "Task",
+    "Goal",
+    "Artifact",
+    "Decision",
+    "WorkflowStatus",
+    "StateEvent",
+    "BaseStateEvent",
+    "WorkflowStarted",
+    "WorkflowCompleted",
+    "PatchApplied",
+    "ConflictDetected",
+    "CheckpointSaved",
+    "AgentErrored",
+    "event_adapter",
+]

agentstate_lib-0.1.0/agentstate/core/events.py

@@ -0,0 +1,74 @@
+import time
+import uuid
+from typing import Annotated, Any, Literal
+
+from pydantic import BaseModel, Field, TypeAdapter
+
+
+# This is the parent all events share
+class BaseStateEvent(BaseModel):
+    event_id : str = Field(default_factory=lambda: str(uuid.uuid4()))
+    workflow_id : str
+    agent_id : str
+    timestamp : float = Field(default_factory=time.time)
+
+
+# six subclasses inherting from BaseStateEvent, 
+# with each one, having a 'type' field typed as a literal with 
+# a single string value, and set its default to that string.
+
+class WorkflowStarted(BaseStateEvent):
+    type : Literal["workflow_started"]
+    workflow_type : str
+    goal : str
+
+
+class WorkflowCompleted(BaseStateEvent):
+    type : Literal["workflow_completed"]
+    final_status : Any
+
+
+class PatchApplied(BaseStateEvent):
+    type : Literal["patch_applied"]
+    patch_id : str
+    target : str
+    old_value : Any
+    new_value : Any
+    reason : str
+
+
+class ConflictDetected(BaseStateEvent):
+    type : Literal["conflict_detected"]
+    conflict_id : str
+    path : str
+    winner_agent_id : str
+    loser_agent_id : str
+    resolution_strategy : str
+
+
+class CheckpointSaved(BaseStateEvent):
+    type : Literal["checkpoint_saved"]
+    checkpoint_id : str
+    event_count : int
+
+class AgentErrored(BaseStateEvent):
+    type : Literal["agent_errored"]
+    error_type : str
+    error_message : str
+    retry_count : int
+
+# Pydantic will use the type field to decide which model to instantiate 
+# when deserializing
+StateEvent = Annotated[
+        WorkflowStarted
+    |    WorkflowCompleted
+    |    PatchApplied
+    |   ConflictDetected
+    |    CheckpointSaved
+    |   AgentErrored,
+    Field(discriminator="type")
+]
+
+# at module level, writing a typeadapter for statevent
+# this is how you deserialize a JSON string into the correct event subtype
+event_adapter: TypeAdapter[StateEvent] = TypeAdapter(StateEvent)

agentstate_lib-0.1.0/agentstate/core/patch.py

@@ -0,0 +1,59 @@
+from __future__ import annotations
+
+import time
+import uuid
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+from agentstate.core.state import SharedState
+
+
+# represents "one change an agent wants to make" to the shared state
+class StatePatch(BaseModel):
+    patch_id:str = Field(default_factory=lambda: str(uuid.uuid4()))
+    agent_id: str
+    target : str # dotted path, eg: tasks.task_1.status
+    value : Any
+    reason : str
+    timestamp : float = Field(default_factory=time.time)
+    priority : int = 0
+
+def set_nested(obj: dict[str, Any], path: str, value : Any) -> dict[str, Any]:
+    """
+    Set value at a dotted path inside a nested dict, creating dicts as needed.
+    """
+    parts = path.split(".") # turns "a.b.c" into ["a","b","c"]
+    current : dict[str, Any] = obj
+
+    for key in parts[:-1]:
+        if key not in current or not isinstance(current[key], dict):
+            current[key] = {}
+        current = current[key]
+    
+    current[parts[-1]] = value
+    return obj
+
+def get_nested(obj: dict[str, Any], path: str) -> Any:
+    """
+    Get value at a dotted path from a nested dict, or None if missing.
+    """
+
+    parts = path.split(".")
+    current : Any = obj
+
+    for key in parts:
+        if not isinstance(current, dict):
+            return None
+        if key not in current:
+            return None
+        current = current[key]
+    
+    return current
+
+# creates a dict view of SharedState with model_dump()
+def apply_patch(state: SharedState, patch: StatePatch) -> SharedState:
+    """Return a new SharedState with the patch applied at patch.target"""
+    state_dict = state.model_dump()
+    set_nested(state_dict, patch.target, patch.value)
+    return SharedState.model_validate(state_dict)

agentstate_lib-0.1.0/agentstate/core/state.py

@@ -0,0 +1,49 @@
+import time
+import uuid
+from typing import Any, Literal
+
+from pydantic import BaseModel, Field
+
+WorkflowStatus = Literal["running", "complete", "failed", "paused"]
+
+class Task(BaseModel):
+    id : str = Field(default_factory=lambda: str(uuid.uuid4()))
+    description : str
+    status : Literal["pending", "running", "done", "failed"] = 'pending'
+    result : Any | None = None
+    created_at : float = Field(default_factory=time.time)
+    updated_as: float = Field(default_factory=time.time)
+
+
+class Goal(BaseModel):
+    id: str = Field(default_factory=lambda: str(uuid.uuid4()))
+    description : str
+    status : Literal["pending", "active", "complete", "failed"] = 'pending'
+    created_at : float = Field(default_factory=time.time)
+
+class Artifact(BaseModel):
+    id : str = Field(default_factory=lambda : str(uuid.uuid4()))
+    produced_by : str
+    artifact_type : str
+    content : Any
+    created_at : float = Field(default_factory=time.time)
+
+class Decision(BaseModel):
+    id : str = Field(default_factory=lambda : str(uuid.uuid4()))
+    made_by : str
+    description : str
+    rationale : str
+    timestamp : float = Field(default_factory=time.time)
+
+class SharedState(BaseModel):
+    workflow_id : str = Field(default_factory=lambda: str(uuid.uuid4()))
+    workflow_type : str = "general"
+    goal : str
+    goals : dict[str, Goal] = Field(default_factory=dict)
+    tasks : dict[str, Task] = Field(default_factory=dict)
+    artifacts : dict[str, Artifact] = Field(default_factory=dict)
+    decisions : list[Decision] = Field(default_factory=list)
+    facts : dict[str, Any] = Field(default_factory=dict)
+    status : WorkflowStatus = 'running' 
+    created_at : float = Field(default_factory=time.time)
+    updated_at : float = Field(default_factory=time.time)

agentstate_lib-0.1.0/agentstate/memory/__init__.py

@@ -0,0 +1,11 @@
+from .store import (
+    InMemoryStore,
+    SQLiteStore,
+    StateStore,
+)
+
+__all__ = [
+    "StateStore",
+    "InMemoryStore",
+    "SQLiteStore",
+]