lionagi 0.14.6__tar.gz → 0.14.8__tar.gz

{lionagi-0.14.6 → lionagi-0.14.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.14.6
+Version: 0.14.8
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>, Liangbingyan Luo <llby_luo@outlook.com>
 License:            Apache License
@@ -234,9 +234,9 @@ Requires-Dist: tiktoken>=0.8.0
 Requires-Dist: toml>=0.9.0
 Provides-Extra: all
 Requires-Dist: aiosqlite>=0.21.0; extra == 'all'
-Requires-Dist: claude-code-sdk>=0.0.14; extra == 'all'
+Requires-Dist: claude-code-sdk>=0.0.15; extra == 'all'
 Requires-Dist: datamodel-code-generator>=0.31.2; extra == 'all'
-Requires-Dist: docling>=2.15.1; extra == 'all'
+Requires-Dist: docling>=2.15.0; extra == 'all'
 Requires-Dist: fastmcp>=2.10.5; extra == 'all'
 Requires-Dist: matplotlib>=3.7.0; extra == 'all'
 Requires-Dist: networkx>=3.0.0; extra == 'all'
@@ -244,36 +244,26 @@ Requires-Dist: ollama>=0.4.0; extra == 'all'
 Requires-Dist: pydapter[postgres]; extra == 'all'
 Requires-Dist: rich>=13.0.0; extra == 'all'
 Provides-Extra: claude-code
-Requires-Dist: claude-code-sdk>=0.0.14; extra == 'claude-code'
-Provides-Extra: docs
-Requires-Dist: furo>=2024.8.6; extra == 'docs'
-Requires-Dist: sphinx-autobuild>=2024.10.3; extra == 'docs'
-Requires-Dist: sphinx>=8.1.3; extra == 'docs'
+Requires-Dist: claude-code-sdk>=0.0.15; extra == 'claude-code'
 Provides-Extra: graph
 Requires-Dist: matplotlib>=3.7.0; extra == 'graph'
 Requires-Dist: networkx>=3.0.0; extra == 'graph'
-Provides-Extra: lint
-Requires-Dist: black[jupyter]>=24.10.0; extra == 'lint'
-Requires-Dist: isort>=5.13.2; extra == 'lint'
-Requires-Dist: pre-commit>=4.0.1; extra == 'lint'
 Provides-Extra: mcp
 Requires-Dist: fastmcp>=2.10.5; extra == 'mcp'
 Provides-Extra: ollama
 Requires-Dist: ollama>=0.4.0; extra == 'ollama'
 Provides-Extra: postgres
-Requires-Dist: aiosqlite>=0.21.0; extra == 'postgres'
 Requires-Dist: pydapter[postgres]; extra == 'postgres'
 Provides-Extra: reader
-Requires-Dist: docling>=2.15.1; extra == 'reader'
+Requires-Dist: docling>=2.15.0; extra == 'reader'
 Provides-Extra: rich
 Requires-Dist: rich>=13.0.0; extra == 'rich'
 Provides-Extra: schema
 Requires-Dist: datamodel-code-generator>=0.31.2; extra == 'schema'
-Provides-Extra: test
-Requires-Dist: pytest-asyncio>=1.0.0; extra == 'test'
-Requires-Dist: pytest>=8.3.4; extra == 'test'
+Provides-Extra: sqlite
+Requires-Dist: aiosqlite>=0.21.0; extra == 'sqlite'
 Provides-Extra: tools
-Requires-Dist: docling>=2.15.1; extra == 'tools'
+Requires-Dist: docling>=2.15.0; extra == 'tools'
 Description-Content-Type: text/markdown
 
 ![PyPI - Version](https://img.shields.io/pypi/v/lionagi?labelColor=233476aa&color=231fc935)
@@ -305,7 +295,9 @@ integrations, and custom validations in a single coherent pipeline.
 ## Installation
 
 ```
-pip install lionagi
+uv add lionagi  # recommended to use pyproject and uv for dependency management
+
+pip install lionagi # or install directly
 ```
 
 ## Quick Start
@@ -314,12 +306,12 @@ pip install lionagi
 from lionagi import Branch, iModel
 
 # Pick a model
-gpt4o = iModel(provider="openai", model="gpt-4o")
+gpt41 = iModel(provider="openai", model="gpt-4.1-mini")
 
 # Create a Branch (conversation context)
 hunter = Branch(
   system="you are a hilarious dragon hunter who responds in 10 words rhymes.",
-  chat_model=gpt4o,
+  chat_model=gpt41,
 )
 
 # Communicate asynchronously
@@ -400,41 +392,75 @@ print(df.tail())
 ```python
 from lionagi import Branch, iModel
 
-gpt4o = iModel(provider="openai", model="gpt-4o")
 sonnet = iModel(
   provider="anthropic",
   model="claude-3-5-sonnet-20241022",
   max_tokens=1000,                    # max_tokens is required for anthropic models
 )
 
-branch = Branch(chat_model=gpt4o)
-# Switch mid-flow
-analysis = await branch.communicate("Analyze these stats", imodel=sonnet)
+branch = Branch(chat_model=gpt41)
+analysis = await branch.communicate("Analyze these stats", chat_model=sonnet) # Switch mid-flow
 ```
 
 Seamlessly route to different models in the same workflow.
 
 ### Claude Code Integration
 
-LionAGI now supports Anthropic's [Claude Code SDK](https://github.com/anthropics/claude-code-sdk), enabling autonomous coding capabilities with persistent session management:
+LionAGI now supports Anthropic's Claude Code [Python SDK](https://github.com/anthropics/claude-code-sdk-python), and [CLI SDK](https://docs.anthropic.com/en/docs/claude-code/sdk) enabling autonomous coding capabilities with persistent session management. The CLI endpoint
+directly connects to claude code, and is recommended, you can either use it via a [proxy server](https://github.com/khive-ai/lionagi/tree/main/cookbooks/claude_proxy) or directly with `query_cli` endpoint, provided you have already logged onto claude code cli in your terminal.
 
 ```python
 from lionagi import iModel, Branch
 
-# Create a Claude Code model
-model = iModel(
-    provider="claude_code",
-    endpoint="query_cli",
-    model="sonnet",
-    allowed_tools=["Write", "Read", "Edit"],  # Control which tools Claude can use
-    permission_mode = "bypassPermissions", # Bypass tool permission checks (use with caution!),
-    verbose_output=True,  # Enable detailed output for debugging
-)
+def create_cc_model():
+  return iModel(
+      provider="claude_code",
+      endpoint="query_cli",
+      model="sonnet",
+      verbose_output=True,  # Enable detailed output for debugging
+  )
 
 # Start a coding session
-branch = Branch(chat_model=model)
-response = await branch.communicate("Explain the architecture of protocols, operations, and branch")
-response2 = await branch.communicate("how do these parts form lionagi system")
+orchestrator = Branch(chat_model=create_cc_model())
+response = await orchestrator.communicate("Explain the architecture of protocols, operations, and branch")
+
+# continue the session with more queries
+response2 = await orchestrator.communicate("how do these parts form lionagi system")
+```
+
+### Fan out fan in pattern orchestration with claude code
+
+```python
+# use structured outputs with claude code
+from lionagi.fields import LIST_INSTRUCT_FIELD_MODEL, Instruct
+
+response3 = await orchestrator.operate(
+  instruct=Instruct(
+    instruction="create 4 research questions for parallel discovery",
+    guidance="put into `instruct_models` field as part of your structured result message",
+    context="I'd like to create an orchestration system for AI agents using lionagi"
+  ),
+  field_models=[LIST_INSTRUCT_FIELD_MODEL],
+)
+
+len(response3.instruct_models)  # should be 4
+
+async def handle_instruct(instruct):
+  sub_branch = Branch(
+    system="You are an diligent research expert.",
+    chat_model=create_cc_model(),
+  )
+  return await sub_branch.operate(instruct=instruct)
+
+# run in parallel across all instruct models
+from lionagi.utils import alcall
+responses = await alcall(response3.instruct_models, handle_instruct)
+
+# now hand these reports back to the orchestrator
+final_response = await orchestrator.communicate(
+  "please synthesize these research findings into a final report",
+  context=responses,
+)
 ```
 
 Key features:
@@ -446,9 +472,14 @@ Key features:
 ### optional dependencies
 
 ```
-pip install "lionagi[reader]"
-pip install "lionagi[ollama]"
-pip install "lionagi[claude-code]"
+"lionagi[reader]" - Reader tool for any unstructured data and web pages
+"lionagi[ollama]" - Ollama model support for local inference
+"lionagi[claude-code]" - Claude code python SDK integration (cli endpoint does not require this)
+"lionagi[rich]" - Rich output formatting for better console display
+"lionagi[schema]" - Convert pydantic schema to make the Model class persistent
+"lionagi[postgres]" - Postgres database support for storing and retrieving structured data
+"lionagi[graph]" - Graph display for visualizing complex workflows
+"lionagi[sqlite]" - SQLite database support for lightweight data storage (also need `postgres` option)
 ```
 
 ## Community & Contributing

{lionagi-0.14.6 → lionagi-0.14.8}/README.md

@@ -27,7 +27,9 @@ integrations, and custom validations in a single coherent pipeline.
 ## Installation
 
 ```
-pip install lionagi
+uv add lionagi  # recommended to use pyproject and uv for dependency management
+
+pip install lionagi # or install directly
 ```
 
 ## Quick Start
@@ -36,12 +38,12 @@ pip install lionagi
 from lionagi import Branch, iModel
 
 # Pick a model
-gpt4o = iModel(provider="openai", model="gpt-4o")
+gpt41 = iModel(provider="openai", model="gpt-4.1-mini")
 
 # Create a Branch (conversation context)
 hunter = Branch(
   system="you are a hilarious dragon hunter who responds in 10 words rhymes.",
-  chat_model=gpt4o,
+  chat_model=gpt41,
 )
 
 # Communicate asynchronously
@@ -122,41 +124,75 @@ print(df.tail())
 ```python
 from lionagi import Branch, iModel
 
-gpt4o = iModel(provider="openai", model="gpt-4o")
 sonnet = iModel(
   provider="anthropic",
   model="claude-3-5-sonnet-20241022",
   max_tokens=1000,                    # max_tokens is required for anthropic models
 )
 
-branch = Branch(chat_model=gpt4o)
-# Switch mid-flow
-analysis = await branch.communicate("Analyze these stats", imodel=sonnet)
+branch = Branch(chat_model=gpt41)
+analysis = await branch.communicate("Analyze these stats", chat_model=sonnet) # Switch mid-flow
 ```
 
 Seamlessly route to different models in the same workflow.
 
 ### Claude Code Integration
 
-LionAGI now supports Anthropic's [Claude Code SDK](https://github.com/anthropics/claude-code-sdk), enabling autonomous coding capabilities with persistent session management:
+LionAGI now supports Anthropic's Claude Code [Python SDK](https://github.com/anthropics/claude-code-sdk-python), and [CLI SDK](https://docs.anthropic.com/en/docs/claude-code/sdk) enabling autonomous coding capabilities with persistent session management. The CLI endpoint
+directly connects to claude code, and is recommended, you can either use it via a [proxy server](https://github.com/khive-ai/lionagi/tree/main/cookbooks/claude_proxy) or directly with `query_cli` endpoint, provided you have already logged onto claude code cli in your terminal.
 
 ```python
 from lionagi import iModel, Branch
 
-# Create a Claude Code model
-model = iModel(
-    provider="claude_code",
-    endpoint="query_cli",
-    model="sonnet",
-    allowed_tools=["Write", "Read", "Edit"],  # Control which tools Claude can use
-    permission_mode = "bypassPermissions", # Bypass tool permission checks (use with caution!),
-    verbose_output=True,  # Enable detailed output for debugging
-)
+def create_cc_model():
+  return iModel(
+      provider="claude_code",
+      endpoint="query_cli",
+      model="sonnet",
+      verbose_output=True,  # Enable detailed output for debugging
+  )
 
 # Start a coding session
-branch = Branch(chat_model=model)
-response = await branch.communicate("Explain the architecture of protocols, operations, and branch")
-response2 = await branch.communicate("how do these parts form lionagi system")
+orchestrator = Branch(chat_model=create_cc_model())
+response = await orchestrator.communicate("Explain the architecture of protocols, operations, and branch")
+
+# continue the session with more queries
+response2 = await orchestrator.communicate("how do these parts form lionagi system")
+```
+
+### Fan out fan in pattern orchestration with claude code
+
+```python
+# use structured outputs with claude code
+from lionagi.fields import LIST_INSTRUCT_FIELD_MODEL, Instruct
+
+response3 = await orchestrator.operate(
+  instruct=Instruct(
+    instruction="create 4 research questions for parallel discovery",
+    guidance="put into `instruct_models` field as part of your structured result message",
+    context="I'd like to create an orchestration system for AI agents using lionagi"
+  ),
+  field_models=[LIST_INSTRUCT_FIELD_MODEL],
+)
+
+len(response3.instruct_models)  # should be 4
+
+async def handle_instruct(instruct):
+  sub_branch = Branch(
+    system="You are an diligent research expert.",
+    chat_model=create_cc_model(),
+  )
+  return await sub_branch.operate(instruct=instruct)
+
+# run in parallel across all instruct models
+from lionagi.utils import alcall
+responses = await alcall(response3.instruct_models, handle_instruct)
+
+# now hand these reports back to the orchestrator
+final_response = await orchestrator.communicate(
+  "please synthesize these research findings into a final report",
+  context=responses,
+)
 ```
 
 Key features:
@@ -168,9 +204,14 @@ Key features:
 ### optional dependencies
 
 ```
-pip install "lionagi[reader]"
-pip install "lionagi[ollama]"
-pip install "lionagi[claude-code]"
+"lionagi[reader]" - Reader tool for any unstructured data and web pages
+"lionagi[ollama]" - Ollama model support for local inference
+"lionagi[claude-code]" - Claude code python SDK integration (cli endpoint does not require this)
+"lionagi[rich]" - Rich output formatting for better console display
+"lionagi[schema]" - Convert pydantic schema to make the Model class persistent
+"lionagi[postgres]" - Postgres database support for storing and retrieving structured data
+"lionagi[graph]" - Graph display for visualizing complex workflows
+"lionagi[sqlite]" - SQLite database support for lightweight data storage (also need `postgres` option)
 ```
 
 ## Community & Contributing

lionagi-0.14.8/cookbooks/007_fan_out_in.py

@@ -0,0 +1,110 @@
+from lionagi import Branch, Builder, Session, iModel
+from lionagi.fields import LIST_INSTRUCT_FIELD_MODEL, Instruct
+
+CC_WORKSPACE = ".khive/workspace"
+
+
+def create_cc(
+    subdir: str,
+    model: str = "sonnet",
+    verbose_output: bool = True,
+    permission_mode="default",
+    auto_finish: bool = False,
+):
+    return iModel(
+        provider="claude_code",
+        endpoint="query_cli",
+        model=model,
+        api_key="dummy_api_key",
+        ws=f"{CC_WORKSPACE}/{subdir}",
+        verbose_output=verbose_output,
+        add_dir="../../../",
+        permission_mode=permission_mode,
+        cli_display_theme="dark",
+        auto_finish=auto_finish,
+    )
+
+
+prompt = """
+Task: Investigate the codebase in the specified directory and provide a comprehensive overview.
+
+---START
+read into the specified dir, glance over the key components and pay attention to architecture, 
+design patterns, and any notable features. Think deeply about the codebase and give three parallel
+instructions, as part of the structured output (`instruct_models`) in the final response message.
+
+---Then
+The instruct models will be run in parallel by each researcher branch, and I will provide you with 
+the researchers' findings for you to continue your investigation.
+
+---Finally
+Once the researchers have completed their tasks, synthesize the information they provided into a cohesive
+overview of the codebase, including:
+1. Key components and their roles
+2. Architectural patterns used
+3. Design patterns and notable features
+---END
+"""
+
+
+async def main():
+    try:
+        orc_cc = create_cc("orchestrator")
+        orc_branch = Branch(
+            chat_model=orc_cc,
+            parse_model=orc_cc,
+            use_lion_system_message=True,
+            system_datetime=True,
+            name="orchestrator",
+        )
+        session = Session(default_branch=orc_branch)
+
+        builder = Builder("CodeInvestigator")
+        root = builder.add_operation(
+            "operate",
+            instruct=Instruct(
+                instruction=prompt,
+                context="lionagi",
+            ),
+            reason=True,
+            field_models=[LIST_INSTRUCT_FIELD_MODEL],
+        )
+
+        result = await session.flow(builder.get_graph())
+
+        instruct_models: list[Instruct] = result["operation_results"][
+            root
+        ].instruct_models
+        research_nodes = []
+
+        for i in instruct_models:
+            node = builder.add_operation(
+                "communicate",
+                depends_on=[root],
+                chat_model=create_cc("researcher"),
+                **i.to_dict(),
+            )
+            research_nodes.append(node)
+
+        synthesis = builder.add_aggregation(
+            "communicate",
+            source_node_ids=research_nodes,
+            branch=orc_branch,
+            instruction="Synthesize the information from the researcher branches.",
+        )
+
+        result2 = await session.flow(builder.get_graph())
+        result_synthesis = result2["operation_results"][synthesis]
+        print(result_synthesis)
+
+    except Exception as e:
+        print(f"Error: {e}")
+        import traceback
+
+        traceback.print_exc()
+
+
+if __name__ == "__main__":
+    import anyio
+
+    anyio.run(main)

lionagi-0.14.8/cookbooks/claude_proxy/README.md

@@ -0,0 +1,31 @@
+# LionAGI Claude Code Proxy
+
+to use please first install `lionagi` and `fastapi`
+
+```bash
+uv add lionagi fastapi loguru
+```
+
+then run the proxy with:
+
+```bash
+uv run cookbooks/claude_proxy/claude_code_proxy.py --port 8000
+```
+
+Next, configure a cutsom endpoint 
+
+```python
+from lionagi.service import Endpoint, EndpointConfig, iModel
+claude_code_config = EndpointConfig(
+    name="claude_code_cli_proxy",
+    provider="claude_code_proxy",
+    base_url="http://localhost:8000/v1",
+    endpoint="query",
+    api_key="dummy_api_key",
+    kwargs={"model": "sonnet"},
+)
+cc_proxy_endpoint = Endpoint(claude_code_config)
+claude_code = iModel(endpoint=cc_proxy_endpoint)
+```
+
+Done, now you can run `claude code` in jupyter or any other environment that supports `lionagi`.

lionagi-0.14.8/cookbooks/claude_proxy/claude_code_proxy.py

@@ -0,0 +1,132 @@
+#!/usr/bin/env python3
+"""
+Claude Code Proxy API Server - OpenAI Compatible
+A FastAPI server that proxies requests to Claude Code CLI using OpenAI API format.
+Compatible with Roo Code, Cline, and other OpenAI API clients.
+"""
+
+from __future__ import annotations
+
+import json
+import subprocess
+import sys
+from contextlib import asynccontextmanager
+
+import uvicorn
+from fastapi import FastAPI, HTTPException, Request, Response
+from loguru import logger
+
+from lionagi.service.connections.providers.claude_code_cli import (
+    ClaudeCodeCLIEndpoint,
+    ClaudeCodeRequest,
+)
+
+# Configure logging
+logger.remove()
+logger.add(sys.stderr, level="INFO")
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    check_claude_code()
+    logger.info("Claude Code CLI Proxy API started successfully")
+    yield
+    pass
+
+
+# FastAPI app
+app = FastAPI(
+    title="Claude Code Proxy API - OpenAI Compatible",
+    description="Proxy server for Claude Code CLI - Personal Development Use",
+    version="1.0.0",
+    lifespan=lifespan,
+)
+
+
+# Check if Claude Code is available
+def check_claude_code():
+    """Verify Claude Code CLI is installed and accessible"""
+    try:
+        from lionagi.service.connections.providers._claude_code.stream_cli import (
+            CLAUDE,
+        )
+
+        result = subprocess.run(
+            [CLAUDE, "--version"], capture_output=True, text=True
+        )
+        if result.returncode != 0:
+            raise RuntimeError(
+                f"Claude Code not found, please install it first:\n"
+            )
+        logger.info(f"Claude Code found: {result.stdout.strip()}")
+    except Exception as e:
+        logger.error(f"Claude Code check failed: {e}")
+        raise RuntimeError(
+            "Claude Code CLI not found. Please install it first:\n"
+            "npm install -g @anthropic-ai/claude-code"
+        )
+
+
+# Health check
+@app.get("/health")
+async def health_check():
+    return {"status": "healthy", "service": "claude-code-cli-proxy"}
+
+
+@app.post("/v1/query")
+async def query(request: ClaudeCodeRequest):
+
+    endpoint = ClaudeCodeCLIEndpoint()
+    try:
+        return await endpoint._call(payload={"request": request}, headers={})
+    except Exception as e:
+        logger.error(f"Error processing request: {e}")
+        raise HTTPException(
+            status_code=500, detail=f"Internal server error: {str(e)}"
+        )
+
+
+# Error handling
+@app.exception_handler(Exception)
+async def general_exception_handler(request: Request, exc: Exception):
+    logger.error(f"Unhandled exception: {exc}")
+    return Response(
+        content=json.dumps(
+            {
+                "error": {
+                    "message": str(exc),
+                    "type": "internal_error",
+                    "param": None,
+                    "code": None,
+                }
+            }
+        ),
+        status_code=500,
+        media_type="application/json",
+    )
+
+
+if __name__ == "__main__":
+    # Parse command line arguments
+    import argparse
+
+    parser = argparse.ArgumentParser(
+        description="Claude Code Proxy API Server"
+    )
+    parser.add_argument("--host", default="127.0.0.1", help="Host to bind to")
+    parser.add_argument(
+        "--port", type=int, default=8000, help="Port to bind to"
+    )
+    parser.add_argument(
+        "--reload", action="store_true", help="Enable auto-reload"
+    )
+    args = parser.parse_args()
+
+    # Run server
+    uvicorn.run(
+        "claude_code_proxy:app" if args.reload else app,
+        host=args.host,
+        port=args.port,
+        reload=args.reload,
+        log_level="info",
+    )