import nyraxis_sdk
from langchain_openai import ChatOpenAI
from langchain.agents import create_react_agent

nyraxis_sdk.init(
    api_key="nyx_...",
    instrument=True,
    enforce=True,
    agent_name="research-agent",
)

llm = ChatOpenAI(model="gpt-4o")
# All LangChain calls are now traced and governed

from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI

nyraxis_sdk.init(api_key="nyx_...", enforce=True)

prompt = ChatPromptTemplate.from_messages([
    ("system", "You are a helpful assistant."),
    ("user", "{input}"),
])

chain = prompt | ChatOpenAI(model="gpt-4o")
result = chain.invoke({"input": user_message})
# Prompt injection in user_message → NyraxisBlockedError

from langgraph.prebuilt import create_react_agent
from langchain_openai import ChatOpenAI

nyraxis_sdk.init(api_key="nyx_...", enforce=True, instrument=True)

agent = create_react_agent(ChatOpenAI(model="gpt-4o"), tools=[...])
result = agent.invoke({"messages": [("user", user_input)]})
# Every LLM call in the graph is evaluated