helix-framework 0.3.2__tar.gz → 0.3.4__tar.gz

{helix_framework-0.3.2/src/helix_framework.egg-info → helix_framework-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: helix-framework
-Version: 0.3.2
+Version: 0.3.4
 Summary: Production-grade AI agent framework — cost governance, memory, caching, multi-agent teams, and built-in eval
 Author: Dhruv Choudhary
 Maintainer: Dhruv Choudhary
@@ -219,7 +219,7 @@ License-File: LICENSE
 License-File: NOTICE
 Requires-Dist: pydantic>=2.0
 Provides-Extra: gemini
-Requires-Dist: google-generativeai>=0.8; extra == "gemini"
+Requires-Dist: google-genai>=1.0; extra == "gemini"
 Provides-Extra: openai
 Requires-Dist: openai>=1.0; extra == "openai"
 Provides-Extra: anthropic
@@ -239,7 +239,7 @@ Requires-Dist: openai>=1.0; extra == "openrouter"
 Provides-Extra: deepseek
 Requires-Dist: openai>=1.0; extra == "deepseek"
 Provides-Extra: providers
-Requires-Dist: google-generativeai>=0.8; extra == "providers"
+Requires-Dist: google-genai>=1.0; extra == "providers"
 Requires-Dist: openai>=1.0; extra == "providers"
 Requires-Dist: anthropic>=0.20; extra == "providers"
 Requires-Dist: groq>=0.9; extra == "providers"
@@ -251,7 +251,7 @@ Requires-Dist: tiktoken>=0.5; extra == "tiktoken"
 Provides-Extra: http
 Requires-Dist: httpx>=0.25; extra == "http"
 Provides-Extra: search
-Requires-Dist: duckduckgo-search>=3.9; extra == "search"
+Requires-Dist: ddgs>=9.0; extra == "search"
 Provides-Extra: redis
 Requires-Dist: redis>=5.0; extra == "redis"
 Provides-Extra: schedule
@@ -263,7 +263,7 @@ Requires-Dist: pytest-cov>=4.0; extra == "dev"
 Requires-Dist: ruff>=0.4; extra == "dev"
 Requires-Dist: mypy>=1.0; extra == "dev"
 Provides-Extra: all
-Requires-Dist: google-generativeai>=0.8; extra == "all"
+Requires-Dist: google-genai>=1.0; extra == "all"
 Requires-Dist: openai>=1.0; extra == "all"
 Requires-Dist: anthropic>=0.20; extra == "all"
 Requires-Dist: groq>=0.9; extra == "all"
@@ -272,7 +272,7 @@ Requires-Dist: cohere>=5.0; extra == "all"
 Requires-Dist: together>=1.2; extra == "all"
 Requires-Dist: tiktoken>=0.5; extra == "all"
 Requires-Dist: httpx>=0.25; extra == "all"
-Requires-Dist: duckduckgo-search>=3.9; extra == "all"
+Requires-Dist: ddgs>=9.0; extra == "all"
 Requires-Dist: redis>=5.0; extra == "all"
 Requires-Dist: croniter>=2.0; extra == "all"
 Dynamic: license-file

{helix_framework-0.3.2 → helix_framework-0.3.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "helix-framework"
-version = "0.3.2"
+version = "0.3.4"
 description = "Production-grade AI agent framework — cost governance, memory, caching, multi-agent teams, and built-in eval"
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -33,7 +33,7 @@ dependencies = ["pydantic>=2.0"]
 
 [project.optional-dependencies]
 # ── Individual providers ──────────────────────────────────────────────────────
-gemini      = ["google-generativeai>=0.8"]
+gemini      = ["google-genai>=1.0"]
 openai      = ["openai>=1.0"]
 anthropic   = ["anthropic>=0.20"]
 groq        = ["groq>=0.9"]
@@ -46,7 +46,7 @@ deepseek    = ["openai>=1.0"]        # DeepSeek uses the openai package
 
 # ── All cloud API providers bundled ──────────────────────────────────────────
 providers = [
-    "google-generativeai>=0.8",
+    "google-genai>=1.0",
     "openai>=1.0",
     "anthropic>=0.20",
     "groq>=0.9",
@@ -58,7 +58,7 @@ providers = [
 # ── Other optional features ───────────────────────────────────────────────────
 tiktoken = ["tiktoken>=0.5"]
 http     = ["httpx>=0.25"]
-search   = ["duckduckgo-search>=3.9"]
+search   = ["ddgs>=9.0"]
 redis    = ["redis>=5.0"]
 schedule = ["croniter>=2.0"]
 
@@ -73,7 +73,7 @@ dev = [
 
 # ── Full install — everything ─────────────────────────────────────────────────
 all = [
-    "google-generativeai>=0.8",
+    "google-genai>=1.0",
     "openai>=1.0",
     "anthropic>=0.20",
     "groq>=0.9",
@@ -82,7 +82,7 @@ all = [
     "together>=1.2",
     "tiktoken>=0.5",
     "httpx>=0.25",
-    "duckduckgo-search>=3.9",
+    "ddgs>=9.0",
     "redis>=5.0",
     "croniter>=2.0",
 ]

{helix_framework-0.3.2 → helix_framework-0.3.4}/src/helix/__init__.py

@@ -32,6 +32,7 @@ Quickstart
 from __future__ import annotations
 
 import asyncio
+import contextlib
 from typing import Any
 
 from helix.config import (
@@ -60,6 +61,8 @@ from helix.config import (
 from helix.config_store import apply_saved_config as _apply_saved_config
 from helix.context import ExecutionContext
 from helix.core.agent import Agent, AgentResult
+from helix.core.agent_decorator import agent
+from helix.core.graph import END, START, CompiledGraph, StateGraph
 from helix.core.group_chat import (
     ChatMessage,
     ConversableAgent,
@@ -67,10 +70,12 @@ from helix.core.group_chat import (
     GroupChatResult,
     HumanAgent,
 )
+from helix.core.hooks import HookEvent
+from helix.core.pipeline import AgentPipeline
 from helix.core.session import Session
 from helix.core.task import Pipeline, PipelineResult, Task, TaskOutput
 from helix.core.team import Team
-from helix.core.tool import ToolRegistry, registry, tool
+from helix.core.tool import ToolRegistry, discover_tools, registry, tool
 from helix.core.workflow import Workflow, step
 from helix.errors import (
     BudgetExceededError,
@@ -83,6 +88,83 @@ from helix.errors import (
 
 _apply_saved_config()
 
+# ── Register built-in tools in sys.modules so 'from helix.tools.builtin import …' works ──
+with contextlib.suppress(Exception):
+    import helix.tools.builtin as _tools_builtin  # noqa: F401 — side-effect import
+
+# ── Preset agents: helix.presets.web_researcher(), helix.presets.coder()…
+with contextlib.suppress(Exception):
+    from helix import presets  # noqa: F401
+
+
+def quick(
+    system_prompt: str,
+    *,
+    name: str = "Agent",
+    model: str | None = None,
+    tools: list[Any] | None = None,
+    budget_usd: float = 0.10,
+    on_event: Any | None = None,
+) -> Agent:
+    """
+    Create a minimal agent from a single system prompt string.
+
+    This is the fastest way to get started — no role, goal, or config
+    objects required.  Perfect for experimentation and notebooks.
+
+    Args:
+        system_prompt: The agent's purpose, written as plain instructions.
+        name:          Agent name (shown in traces).  Default ``"Agent"``.
+        model:         LLM model string, e.g. ``"gpt-4o"`` or ``"claude-sonnet-4-6"``.
+                       Omit to auto-detect from available API keys.
+        tools:         List of ``@helix.tool``-decorated functions.
+        budget_usd:    Hard spend cap per run.  Default ``0.10``.
+        on_event:      Optional async or sync callback for live events.
+
+    Returns:
+        A :class:`helix.Agent` ready to run.
+
+    Example::
+
+        import helix
+
+        agent = helix.quick("You are a concise Python tutor.")
+        result = helix.run(agent, "Explain list comprehensions.")
+        print(result.output)
+    """
+    return Agent(
+        name=name,
+        role=name,
+        goal=system_prompt,
+        system_prompt=system_prompt,
+        tools=tools or [],
+        model=ModelConfig(primary=model) if model else ModelConfig(),
+        budget=BudgetConfig(budget_usd=budget_usd),
+        on_event=on_event,
+    )
+
+
+def chain(*agents_or_pipelines: Any) -> AgentPipeline:
+    """
+    Build an :class:`AgentPipeline` from two or more agents.
+
+    Equivalent to ``agent_a | agent_b | agent_c`` but more readable
+    when composing many agents.
+
+    Example::
+
+        from helix.presets import web_researcher, summariser, writer
+
+        pipeline = helix.chain(web_researcher(), summariser(), writer())
+        result = pipeline.run_sync("Quantum computing advances 2026")
+    """
+    if not agents_or_pipelines:
+        raise ValueError("chain() requires at least one agent")
+    result: Any = agents_or_pipelines[0]
+    for nxt in agents_or_pipelines[1:]:
+        result = result | nxt
+    return result
+
 
 def run(
     agent: Agent,
@@ -222,11 +304,13 @@ try:
 
     __version__: str = _pkg_version("helix-framework")
 except _PNFE:  # editable / source install without metadata
-    __version__ = "0.3.2"
+    __version__ = "0.3.3"
 
 __all__ = [
     "run",
     "run_async",
+    "quick",
+    "chain",
     "create_agent",
     "wrap_llm",
     "from_crewai",
@@ -234,9 +318,15 @@ __all__ = [
     "from_autogen",
     "from_yaml",
     "eval_suite",
+    "discover_tools",
+    # Decorators
+    "agent",
+    "tool",
+    "step",
     # Core classes
     "Agent",
     "AgentResult",
+    "AgentPipeline",
     "ConversableAgent",
     "HumanAgent",
     "GroupChat",
@@ -249,11 +339,10 @@ __all__ = [
     "Workflow",
     "Team",
     "Session",
-    "step",
-    "tool",
     "ToolRegistry",
     "registry",
     "ExecutionContext",
+    "HookEvent",
     # Config
     "AgentConfig",
     "AgentMode",
@@ -274,6 +363,12 @@ __all__ = [
     "WorkflowMode",
     "EpisodeOutcome",
     "FailureClass",
+    # Graph
+    "StateGraph",
+    "CompiledGraph",
+    "END",
+    "START",
+    # Errors
     "HelixError",
     "BudgetExceededError",
     "LoopDetectedError",