a1-ai 2.8.0

A1 — The cryptographic identity and authorization layer that turns anonymous AI agents into accountable, verifiable entities. One Identity. Full Provenance.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

"""
examples/integrations/langchain_example.py

Complete LangChain integration showing how to gate a trading tool with
a1 authorization.  Run the gateway first:

    docker run -p 8080:8080 ghcr.io/dyologician/a1-gateway:2

Install deps:

    pip install a1 langchain langchain-openai
"""
from __future__ import annotations

import json
import os

from a1 import A1Client, IntentSpec
from a1.langchain_tool import a1_tool

# ── Setup ─────────────────────────────────────────────────────────────────────

client = A1Client(os.getenv("A1_GATEWAY_URL", "http://localhost:8080"))

# In production these come from your identity provider / secrets manager.
AGENT_PK_HEX: str = os.environ["AGENT_PK_HEX"]
# The signed chain is obtained from the previous delegation step and passed
# through your agent's session context.
SIGNED_CHAIN: dict = json.loads(os.environ["AGENT_SIGNED_CHAIN"])

# ── Define a guarded LangChain tool ──────────────────────────────────────────

@a1_tool(
    name="execute_trade",
    description="Execute an equity trade on the user's behalf. Input: JSON with symbol and qty.",
    intent_name="trade.equity",
    client=client,
    chain=SIGNED_CHAIN,
    executor_pk_hex=AGENT_PK_HEX,
)
def execute_trade(tool_input: str) -> str:
    args = json.loads(tool_input)
    symbol: str = args["symbol"]
    qty: int    = int(args["qty"])

    # Replace with your real broker call.
    print(f"[broker] BUY {qty} × {symbol}")
    return json.dumps({"status": "filled", "symbol": symbol, "qty": qty})


# ── Wire up to a LangChain agent ─────────────────────────────────────────────

def main() -> None:
    from langchain.agents import AgentExecutor, create_openai_tools_agent
    from langchain_openai import ChatOpenAI
    from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder

    llm = ChatOpenAI(model="gpt-4o", temperature=0)

    prompt = ChatPromptTemplate.from_messages([
        ("system", "You are a trading assistant. Use execute_trade to place orders."),
        ("human", "{input}"),
        MessagesPlaceholder("agent_scratchpad"),
    ])

    agent = create_openai_tools_agent(llm, [execute_trade], prompt)
    executor = AgentExecutor(agent=agent, tools=[execute_trade], verbose=True)

    result = executor.invoke({"input": "Buy 10 shares of AAPL for me."})
    print(result["output"])


if __name__ == "__main__":
    main()