from egregore import Agent

# Create an agent with OpenAI's GPT-4
agent = Agent(provider="openai:gpt-4")

# Make your first call
response = agent.call("Hello! What's the weather like?")
print(response)

# The agent maintains context automatically
response = agent.call("What did I just ask you about?")
print(response)  # Agent remembers the previous question

from egregore import Agent
from egregore.core.context_scaffolds import InternalNotesScaffold

# Create scaffold for persistent notes
notes = InternalNotesScaffold()

# Create agent with scaffold
agent = Agent(
    provider="openai:gpt-4",
    scaffolds=[notes]
)

# The agent now has access to note-taking tools
response = agent.call(
    "Remember that my favorite color is blue. "
    "Also, I prefer working in the mornings."
)

# Later in the conversation...
response = agent.call("What do you know about my preferences?")
# Agent can retrieve notes from its persistent memory

from egregore import Agent

agent = Agent(provider="openai:gpt-4")

# Stream the response
for chunk in agent.stream("Write a poem about AI"):
    print(chunk, end="", flush=True)

from egregore import Agent
from egregore.core.context_management import TextContent

agent = Agent(provider="openai:gpt-4")

# Insert a persistent reminder at system depth
reminder = TextContent(content="Always be concise and helpful.")
agent.context.pact_insert("d-1,0", reminder)

# The reminder persists across all turns
response = agent.call("Explain quantum computing")
# Response will be concise due to the system-level reminder

from egregore import Agent

def get_weather(location: str) -> str:
    """Get the current weather for a location."""
    # In reality, you'd call a weather API
    return f"The weather in {location} is sunny, 72°F"

# Create agent with custom tool
agent = Agent(
    provider="openai:gpt-4",
    tools=[get_weather]
)

# Agent can now use the weather tool
response = agent.call("What's the weather in San Francisco?")
print(response)

from egregore import Agent

agent = Agent(provider="openai:gpt-4")

# Add a hook to log all tool calls
@agent.hooks.tool.pre_call
def log_tool_call(ctx):
    print(f"🔧 Calling tool: {ctx.tool_name}")
    print(f"   Parameters: {ctx.parameters}")

@agent.hooks.tool.post_call
def log_tool_result(ctx):
    print(f"✅ Tool result: {ctx.tool_result}")

# Tool calls will now be logged
response = agent.call("What's the weather in NYC?")

agent = Agent(provider="openai:gpt-4")

# Context is automatically managed
agent.call("My name is Alice")
agent.call("I like Python programming")
agent.call("What do you know about me?")
# Agent remembers: name and Python preference

from egregore.core.context_management import TextContent

agent = Agent(provider="openai:gpt-4")

# Add a reminder that expires after 3 turns
reminder = TextContent(content="User is in a hurry", ttl=3)
agent.context.pact_insert("d0,1", reminder)

# Reminder available for 3 turns, then automatically removed

import asyncio
from egregore import Agent

async def main():
    agent = Agent(provider="openai:gpt-4")

    # Async call
    response = await agent.acall("Hello!")

    # Async streaming
    async for chunk in agent.astream("Tell me a story"):
        print(chunk, end="", flush=True)

asyncio.run(main())