import os
from crewai import Agent, Task, Crew, tool
from oculus_sdk import OculusClient

oculus = OculusClient(
    client_id=os.environ["AGENT_CLIENT_ID"],
    client_secret=os.environ["AGENT_CLIENT_SECRET"],
)

@tool("Read customer data")
def read_customer(customer_id: str) -> str:
    """Read customer data from the database."""
    import httpx
    token = oculus.get_token(scope="db:read")
    resp = httpx.get(
        f"https://api.myservice.com/customers/{customer_id}",
        headers={"Authorization": f"Bearer {token}"},
    )
    return resp.text

analyst = Agent(
    role="Customer Analyst",
    goal="Analyze customer data and provide insights",
    backstory="You are a data analyst with access to customer records.",
    tools=[read_customer],
    verbose=True,
)

task = Task(
    description="Analyze the top 10 customers by revenue",
    agent=analyst,
)

crew = Crew(agents=[analyst], tasks=[task])
result = crew.kickoff()