Configuration

from nomos import Agent, AgentConfig
from nomos.models.agent import Step, Route
from nomos.models.flow import FlowConfig
from nomos.llms import LLMConfig
from math import sqrt, pow

def get_time():
    """Get the current time.

    Returns:
        str: The current time in string format.
    """
    from datetime import datetime
    return f"Current time: {datetime.now()}"

steps = [
    Step(
        step_id="start",
        description="Greet and offer to tell the time or perform calculations.",
        available_tools=["get_time", "sqrt"],
        routes=[Route(target="calculation", condition="User wants to do math"),
                Route(target="end", condition="User is done")],
    ),
    Step(
        step_id="calculation",
        description="Perform mathematical calculations for the user.",
        available_tools=["sqrt", "pow"],
        routes=[Route(target="end", condition="Calculation is complete")],
    ),
    Step(
        step_id="end",
        description="Say goodbye.",
    ),
]

# Define flows for better organization
flows = [
    FlowConfig(
        flow_id="math_workflow",
        description="Handle mathematical calculations",
        enters=["calculation"],
        exits=["end"],
        components={
            "memory": {
                "llm": {"provider": "openai", "model": "gpt-4o-mini"},
                "retriever": {"method": "embedding", "kwargs": {"k": 3}}
            }
        }
    )
]

config = AgentConfig(
    name="clockbot",
    llm=LLMConfig(provider="openai", model="gpt-4o-mini"),
    steps=steps,
    flows=flows,
    start_step_id="start",
    persona="You are a friendly assistant that can tell time and perform calculations.",
    max_errors=3,
    max_iter=10,
)

agent = Agent.from_config(config, tools=[get_time, sqrt, pow])
session = agent.create_session()
# ... interact with session.next(user_input)

name: utility-bot
persona: You are a helpful utility bot that can perform various calculations and data operations.
steps:
  - step_id: start
    description: Handle user requests for mathematical operations or data processing.
    available_tools:
      - sqrt
      - load_json
      - combinations
    routes:
      - target: end
        condition: User is done with calculations
  - step_id: end
    description: Say goodbye to the user.
start_step_id: start
max_errors: 3  # Maximum consecutive errors before stopping
max_iter: 10  # Maximum iterations per interaction

# Server configuration (optional)
server:
  port: 8000
  host: "0.0.0.0"
  workers: 1
  security:
    # Authentication
    enable_auth: false
    auth_type: jwt  # or "api_key"
    jwt_secret_key: "$JWT_SECRET_KEY"

    # Rate limiting
    enable_rate_limiting: false
    redis_url: "$REDIS_URL"  # Redis URL for rate limiting
    rate_limit_times: 100  # Number of allowed requests
    rate_limit_seconds: 60  # Time period for rate limiting in seconds

    # CSRF protection
    enable_csrf_protection: false
    csrf_secret_key: "$CSRF_SECRET_KEY"

    # CORS configuration
    allowed_origins:
      - "*"  # Use specific domains in production

server:
  port: 8000
  host: "0.0.0.0"
  workers: 1

  security:
    # Authentication configuration
    enable_auth: true
    auth_type: jwt  # Options: "jwt", "api_key"
    jwt_secret_key: "$JWT_SECRET_KEY"
    # api_key_url: "https://your-api.com/validate-key"  # For API key auth

    # Rate limiting configuration
    enable_rate_limiting: true
    redis_url: "$REDIS_URL"  # Redis URL for rate limiting
    rate_limit_times: 100  # Number of allowed requests
    rate_limit_seconds: 60  # Time period for rate limiting in seconds

    # CSRF protection configuration
    enable_csrf_protection: true
    csrf_secret_key: "$CSRF_SECRET_KEY"

    # CORS configuration
    allowed_origins:
      - "https://your-frontend.com"
      - "http://localhost:3000"  # For development

  # Session store configuration
  session:
    store_type: "memory"  # Options: "memory", "production"
    default_ttl: 3600
    cache_ttl: 3600
    # For production session store:
    # database_url: "$DATABASE_URL"
    # redis_url: "$REDIS_URL"
    # kafka_brokers: "$KAFKA_BROKERS"
    # events_enabled: true

llm:
  provider: openai
  model: gpt-4o-mini

nomos schema --output agent.schema.json

name: my-agent
persona: A helpful assistant
steps:
  - step_id: start
    # ... step configuration
start_step_id: start

# Tool configuration - NEW in v0.2.4
tools:
  tool_files:
    - "barista_tools.py"    # Python module
    - "tools.my_tools"      # File path
  external_tools:
    - tag: "@pkg/itertools.combinations"
      name: "combinations"
    - tag: "@crewai/FileReadTool"
      name: "file_read_tool"
    - tag: "@langchain/google_search"
      name: "google_search"
  tool_arg_descriptions:
    add_to_cart:
      coffee_type: "Coffee type (e.g., Espresso, Latte, Cappuccino)"
      size: "Size of the coffee (Small, Medium, Large)"

# tools/my_tools.py
def greet(name: str) -> str:
    """Return a simple greeting."""
    return f"Hello {name}!"

tools = [greet]

steps:
  - step_id: start
    description: Initial step
    examples:
      - context: "User asks for the time"
        decision: "Answer with the current time."
        visibility: always
      - context: "sqrt 4"
        decision: "Call sqrt tool"
        visibility: dynamic

name: my-agent
# ... other config
max_errors: 3  # Maximum consecutive errors before stopping
max_iter: 10   # Maximum iterations per interaction

name: my-secure-agent
llm:
  provider: openai
  api_key: "$OPENAI_API_KEY"  # References environment variable

server:
  security:
    jwt_secret_key: "$JWT_SECRET_KEY"
    csrf_secret_key: "$CSRF_SECRET_KEY"
  session:
    database_url: "$DATABASE_URL"
    redis_url: "$REDIS_URL"