QuLab 2.12.3__tar.gz → 2.13.0__tar.gz

{qulab-2.12.3 → qulab-2.13.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: QuLab
-Version: 2.12.3
+Version: 2.13.0
 Summary: contral instruments and manage data
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>
@@ -35,8 +35,10 @@ Requires-Dist: matplotlib>=3.7.2
 Requires-Dist: msgpack>=1.0.5
 Requires-Dist: nevergrad>=1.0.2
 Requires-Dist: numpy>=1.13.3
+Requires-Dist: openai>=1.0.0
 Requires-Dist: ply>=3.11
 Requires-Dist: pyperclip>=1.8.2
+Requires-Dist: pyyaml>=6.0
 Requires-Dist: pyzmq>=25.1.0
 Requires-Dist: qlisp>=1.1.4
 Requires-Dist: qlispc>=1.1.8
@@ -50,6 +52,9 @@ Requires-Dist: waveforms>=1.9.4
 Requires-Dist: rich>=14.0.0
 Requires-Dist: pydantic>=2.12.5
 Requires-Dist: openpyxl>=3.1.5
+Requires-Dist: anthropic>=0.84.0
+Requires-Dist: fastapi>=0.115.0
+Requires-Dist: uvicorn>=0.30.0
 Provides-Extra: full
 Requires-Dist: paramiko>=3.5.1; extra == "full"
 Requires-Dist: uvloop>=0.19.0; extra == "full"

{qulab-2.12.3 → qulab-2.13.0}/QuLab.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: QuLab
-Version: 2.12.3
+Version: 2.13.0
 Summary: contral instruments and manage data
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>
@@ -35,8 +35,10 @@ Requires-Dist: matplotlib>=3.7.2
 Requires-Dist: msgpack>=1.0.5
 Requires-Dist: nevergrad>=1.0.2
 Requires-Dist: numpy>=1.13.3
+Requires-Dist: openai>=1.0.0
 Requires-Dist: ply>=3.11
 Requires-Dist: pyperclip>=1.8.2
+Requires-Dist: pyyaml>=6.0
 Requires-Dist: pyzmq>=25.1.0
 Requires-Dist: qlisp>=1.1.4
 Requires-Dist: qlispc>=1.1.8
@@ -50,6 +52,9 @@ Requires-Dist: waveforms>=1.9.4
 Requires-Dist: rich>=14.0.0
 Requires-Dist: pydantic>=2.12.5
 Requires-Dist: openpyxl>=3.1.5
+Requires-Dist: anthropic>=0.84.0
+Requires-Dist: fastapi>=0.115.0
+Requires-Dist: uvicorn>=0.30.0
 Provides-Extra: full
 Requires-Dist: paramiko>=3.5.1; extra == "full"
 Requires-Dist: uvloop>=0.19.0; extra == "full"

{qulab-2.12.3 → qulab-2.13.0}/QuLab.egg-info/SOURCES.txt

@@ -14,6 +14,50 @@ qulab/__main__.py
 qulab/typing.py
 qulab/utils.py
 qulab/version.py
+qulab/auto/__init__.py
+qulab/auto/cli.py
+qulab/auto/config.py
+qulab/auto/exceptions.py
+qulab/auto/main.py
+qulab/auto/agent/__init__.py
+qulab/auto/agent/analyzer.py
+qulab/auto/agent/executor.py
+qulab/auto/agent/loop.py
+qulab/auto/agent/memory.py
+qulab/auto/agent/planner.py
+qulab/auto/bus/__init__.py
+qulab/auto/bus/queue.py
+qulab/auto/executor/__init__.py
+qulab/auto/executor/base.py
+qulab/auto/executor/context.py
+qulab/auto/executor/local.py
+qulab/auto/llm/__init__.py
+qulab/auto/llm/anthropic.py
+qulab/auto/llm/base.py
+qulab/auto/llm/openai.py
+qulab/auto/llm/registry.py
+qulab/auto/models/__init__.py
+qulab/auto/models/session.py
+qulab/auto/skills/__init__.py
+qulab/auto/skills/base.py
+qulab/auto/skills/cache.py
+qulab/auto/skills/generator.py
+qulab/auto/skills/loader.py
+qulab/auto/skills/builtin/__init__.py
+qulab/auto/tools/__init__.py
+qulab/auto/tools/analysis.py
+qulab/auto/tools/base.py
+qulab/auto/tools/config.py
+qulab/auto/tools/human.py
+qulab/auto/tools/lesson.py
+qulab/auto/tools/measurement.py
+qulab/auto/tools/query.py
+qulab/auto/tools/registry.py
+qulab/auto/world_model/__init__.py
+qulab/auto/world_model/base.py
+qulab/auto/world_model/history.py
+qulab/auto/world_model/parameter.py
+qulab/auto/world_model/state.py
 qulab/cli/__init__.py
 qulab/cli/commands.py
 qulab/cli/config.py
@@ -49,21 +93,25 @@ qulab/scan/space.py
 qulab/scan/utils.py
 qulab/storage/__init__.py
 qulab/storage/__main__.py
-qulab/storage/base_dataset.py
+qulab/storage/array.py
+qulab/storage/array_utils.py
+qulab/storage/attachment.py
+qulab/storage/base.py
 qulab/storage/chunk.py
+qulab/storage/cli.py
 qulab/storage/dataset.py
-qulab/storage/file.py
-qulab/storage/storage.py
-qulab/storage/backend/__init__.py
-qulab/storage/backend/redis.py
+qulab/storage/document.py
+qulab/storage/local.py
+qulab/storage/remote.py
+qulab/storage/render.py
+qulab/storage/server.py
 qulab/storage/models/__init__.py
+qulab/storage/models/attachment.py
 qulab/storage/models/base.py
 qulab/storage/models/config.py
-qulab/storage/models/file.py
-qulab/storage/models/ipy.py
-qulab/storage/models/models.py
-qulab/storage/models/record.py
-qulab/storage/models/report.py
+qulab/storage/models/dataset.py
+qulab/storage/models/document.py
+qulab/storage/models/script.py
 qulab/storage/models/tag.py
 qulab/sys/__init__.py
 qulab/sys/chat.py
@@ -99,6 +147,15 @@ qulab/sys/rpc/zmq_socket.py
 qulab/tools/__init__.py
 qulab/tools/connection_helper.py
 qulab/tools/xlsx.py
+qulab/trace/__init__.py
+qulab/trace/capture.py
+qulab/trace/cli.py
+qulab/trace/client.py
+qulab/trace/hooks.py
+qulab/trace/models.py
+qulab/trace/server.py
+qulab/trace/storage.py
+qulab/trace/watcher.py
 qulab/visualization/__init__.py
 qulab/visualization/__main__.py
 qulab/visualization/_autoplot.py
@@ -109,4 +166,5 @@ qulab/visualization/qdat.py
 qulab/visualization/rot3d.py
 qulab/visualization/widgets.py
 src/qulab.h
+tests/test_auto_framework.py
 tests/test_scan.py

{qulab-2.12.3 → qulab-2.13.0}/QuLab.egg-info/requires.txt

@@ -9,8 +9,10 @@ matplotlib>=3.7.2
 msgpack>=1.0.5
 nevergrad>=1.0.2
 numpy>=1.13.3
+openai>=1.0.0
 ply>=3.11
 pyperclip>=1.8.2
+pyyaml>=6.0
 pyzmq>=25.1.0
 qlisp>=1.1.4
 qlispc>=1.1.8
@@ -24,6 +26,9 @@ waveforms>=1.9.4
 rich>=14.0.0
 pydantic>=2.12.5
 openpyxl>=3.1.5
+anthropic>=0.84.0
+fastapi>=0.115.0
+uvicorn>=0.30.0
 
 [full]
 paramiko>=3.5.1

{qulab-2.12.3 → qulab-2.13.0}/pyproject.toml

@@ -39,8 +39,10 @@ dependencies = [
     "msgpack>=1.0.5",
     "nevergrad>=1.0.2",
     "numpy>=1.13.3",
+    "openai>=1.0.0",
     "ply>=3.11",
     "pyperclip>=1.8.2",
+    "pyyaml>=6.0",
     "pyzmq>=25.1.0",
     "qlisp>=1.1.4",
     "qlispc>=1.1.8",
@@ -54,6 +56,9 @@ dependencies = [
     "rich>=14.0.0",
     "pydantic>=2.12.5",
     "openpyxl>=3.1.5",
+    "anthropic>=0.84.0",
+    "fastapi>=0.115.0",
+    "uvicorn>=0.30.0",
 ]
 dynamic = ["version"]
 
@@ -81,3 +86,10 @@ include-package-data = true
 
 [tool.setuptools.dynamic]
 version = {attr = "qulab.version.__version__"}
+
+[dependency-groups]
+dev = [
+    "pylint[spelling]>=4.0.5",
+    "pytest>=9.0.2",
+    "pytest-asyncio>=1.3.0",
+]

qulab-2.13.0/qulab/auto/__init__.py

@@ -0,0 +1,109 @@
+"""QuLab Automatic Experiment Framework
+
+This module provides an automated experiment framework similar to nanobot,
+using LLM as the decision center for coordinating measurement and analysis tasks.
+
+Example:
+    >>> from qulab.auto import AutoLab
+    >>> from qulab.storage import LocalStorage
+    >>>
+    >>> # Initialize storage
+    >>> storage = LocalStorage("./data")
+    >>>
+    >>> # Initialize AutoLab with Kimi
+    >>> lab = AutoLab(storage, llm_config={
+    ...     "provider": "openai",
+    ...     "base_url": "https://api.moonshot.cn/v1",
+    ...     "api_key": "sk-...",
+    ...     "model": "kimi-k2.5"
+    ... })
+    >>>
+    >>> # Start a session
+    >>> async for event in lab.start("Calibrate qubit 1 frequency"):
+    ...     print(f"[{event.type}] {event.content}")
+
+The framework consists of:
+- **Agent**: ReAct decision loop that coordinates tasks
+- **LLM**: Pluggable LLM providers (Anthropic, OpenAI-compatible like Kimi)
+- **Skills**: YAML + Markdown files describing experiments and analyses
+- **Tools**: Query, measurement, analysis, config update, and human query
+- **Storage**: Persistent storage via qulab.storage
+"""
+
+from .config import (
+    AutoLabConfig,
+    BusConfig,
+    ExecutorConfig,
+    LLMConfig,
+    MemoryConfig,
+    SkillConfig,
+    WorldModelConfig,
+)
+from .exceptions import (
+    AutoLabError,
+    ConfigError,
+    LLMError,
+    SessionError,
+    SkillError,
+    SkillNotFoundError,
+    ToolError,
+    ToolNotFoundError,
+)
+from .main import AutoLab
+
+__all__ = [
+    # Main class
+    "AutoLab",
+    # Configuration
+    "AutoLabConfig",
+    "LLMConfig",
+    "SkillConfig",
+    "ExecutorConfig",
+    "MemoryConfig",
+    "WorldModelConfig",
+    "BusConfig",
+    # Exceptions
+    "AutoLabError",
+    "SkillError",
+    "SkillNotFoundError",
+    "ToolError",
+    "ToolNotFoundError",
+    "LLMError",
+    "ConfigError",
+    "SessionError",
+]
+
+
+def __getattr__(name):
+    """Lazy import for agent components."""
+    if name == "AgentLoop":
+        from .agent.loop import AgentLoop
+        return AgentLoop
+    if name == "AgentConfig":
+        from .agent.loop import AgentConfig
+        return AgentConfig
+    if name == "AgentEvent":
+        from .agent.loop import AgentEvent
+        return AgentEvent
+    if name == "PlannerAgent":
+        from .agent.planner import PlannerAgent
+        return PlannerAgent
+    if name == "ExecutorAgent":
+        from .agent.executor import ExecutorAgent
+        return ExecutorAgent
+    if name == "AnalysisAgent":
+        from .agent.analyzer import AnalysisAgent
+        return AnalysisAgent
+    if name == "Skill":
+        from .skills.base import Skill
+        return Skill
+    if name == "SkillLoader":
+        from .skills.loader import SkillLoader
+        return SkillLoader
+    if name == "MessageBus":
+        from .bus import MessageBus
+        return MessageBus
+    if name == "WorldModel":
+        from .world_model import WorldModel
+        return WorldModel
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

qulab-2.13.0/qulab/auto/agent/__init__.py

@@ -0,0 +1,34 @@
+"""Agent system for auto experiment framework.
+
+This module provides a three-layer agent architecture:
+- PlannerAgent: High-level strategy and planning
+- ExecutorAgent: Experiment skill execution
+- AnalysisAgent: Data analysis execution
+"""
+
+from .analyzer import AnalysisAgent, AnalysisContext, AnalysisResult
+from .executor import ExecutorAgent, ExecutionContext, ExecutionResult
+from .loop import AgentConfig, AgentEvent, AgentLoop
+from .memory import SessionMemory
+from .planner import ExecutionPlan, PlanStatus, PlanStep, PlannerAgent
+
+__all__ = [
+    # Original classes
+    "AgentLoop",
+    "AgentConfig",
+    "AgentEvent",
+    "SessionMemory",
+    # Planner Agent
+    "PlannerAgent",
+    "ExecutionPlan",
+    "PlanStep",
+    "PlanStatus",
+    # Executor Agent
+    "ExecutorAgent",
+    "ExecutionContext",
+    "ExecutionResult",
+    # Analysis Agent
+    "AnalysisAgent",
+    "AnalysisContext",
+    "AnalysisResult",
+]