Auto-generate AI agent from OpenAPI specs

from typing import Any

from fastapi import FastAPI, HTTPException
from fastapi.routing import APIRoute
from pydantic import BaseModel

app = FastAPI(
    title="User Management API Liman Demo",
    description="FastAPI implementation based of Liman Simple OpenAPI samle",
    version="1.0.0",
)


class User(BaseModel):
    id: str
    name: str
    email: str
    region: str
    is_admin: bool


class UpdateUserRequest(BaseModel):
    name: str | None = None
    email: str | None = None
    is_admin: bool | None = None


users = {
    "12345": {
        "id": "12345",
        "name": "John",
        "email": "john@example.com",
        "region": "US",
        "is_admin": True,
    },
    "67890": {
        "id": "67890",
        "name": "Max",
        "email": "max@example.com",
        "region": "EU",
        "is_admin": True,
    },
    "54321": {
        "id": "54321",
        "name": "Alice",
        "email": "alice@example.com",
        "region": "US",
        "is_admin": False,
    },
    "17821": {
        "id": "17821",
        "name": "Sofia",
        "email": "sofia@example.com",
        "region": "EU",
        "is_admin": False,
    },
    "99821": {
        "id": "99821",
        "name": "Bob",
        "email": "bob@example.com",
        "region": "EU",
        "is_admin": False,
    },
}


@app.get("/users/{user_id}", response_model=User)
def get_user(user_id: str) -> User:
    """Get a user by their ID"""
    user = users.get(user_id)
    if not user:
        raise HTTPException(status_code=404, detail=f"User with ID {user_id} not found")
    return User.model_validate(user)


@app.get("/users", response_model=list[User])
def list_users() -> list[User]:
    """List all users in the system"""
    return [User.model_validate(user) for user in users.values()]


@app.get("/users/search/name/{name}", response_model=User)
def search_user(name: str) -> User:
    """Search a user by name"""
    for user in users.values():
        if user["name"].lower() == name.lower():
            return User.model_validate(user)
    raise HTTPException(status_code=404, detail=f"No user found with name {name}")

@app.put("/users/{user_id}", response_model=dict[str, Any])
def update_user(user_id: str, request: UpdateUserRequest) -> dict[str, Any]:
	"""Update a user by ID"""
	if user_id not in users:
		raise HTTPException(status_code=404, detail=f"User with ID {user_id} not found")

	user = users[user_id]
	changed = False

	for field in ["name", "email", "is_admin"]:
		value = getattr(request, field, None)
		if value is not None and value != user.get(field):
			user[field] = value
			changed = True

	return {"was_changed": changed, "user": user}


@app.get("/users/search/region/{region}", response_model=list[str])
def search_by_region(region: str) -> list[str]:
    """Find user IDs by region"""
    user_ids = [
        user["id"]
        for user in users.values()
        if user["region"].lower() == region.lower()
    ]
    if not user_ids:
        raise HTTPException(
            status_code=404, detail=f"No users found in region {region}"
        )
    return user_ids


@app.delete("/users/{user_id}")
def delete_user(user_id: str) -> dict[str, str]:
    """Delete a user by ID"""
    if user_id in users:
        del users[user_id]
    else:
        raise HTTPException(status_code=404, detail=f"User with ID {user_id} not found")

    return {"message": f"User with ID {user_id} has been deleted"}


def use_route_names_as_operation_ids(app: FastAPI) -> None:
    """
    Simplify operation IDs so that generated API clients have simpler function
    names.

    Should be called only after all routes have been added.
    """
    for route in app.routes:
        if isinstance(route, APIRoute):
            route.operation_id = route.name


use_route_names_as_operation_ids(app)

fastapi run server.py

kind: LLMNode
name: chat
prompts:
  system:
    en: |
      You are a helpful assistant that works through chat.
      You can execute tools in parallel or sequentially as needed.
      Always reconsider your approach if the initial solution doesn't work.

python main.py

Input: What can you do?
Agent: I can interact with various tools to help you manage user data. Here are some of the things I can do:

1. **Get a User by ID**: Retrieve detailed information about a user by their unique ID.
2. **Search Users by Name**: Find users based on their name.
3. **Find User IDs by Region**: Search for user IDs in a specific region.
4. **List All Users**: Provide a list of all users in the system.

Let me know if you need help with any of these tasks!
--------------------
Input: Find the users Alice and Pavel
Agent: I found the user Alice in the system:

- **Alice**
  - ID: 54321
  - Email: alice@example.com
  - Region: US
  - Is Admin: No

However, I couldn't find any user with the name Pavel in the system. If you have more details or another query, let me know!
--------------------
Input:

def confirm_action(title: str, action: str, user_id: str) -> str:
    print(f"{title}? Y/N", end=": ")
    res = input()
    if res.lower() == "y":
        return f"{action} confirmed. User ID: {user_id}"
    else:
        return "{action} cancelled."

python main.py

Input: Revoke all admins rights for users in EU region, and explain the decision steps you did 1 by 1.
Revoke admin rights for Max (User ID: 67890) in the EU region? Y/N: y
Agent: The admin rights for Max (User ID: 67890) have been successfully revoked. Here's a summary of the action taken:

1. **Identified Users in EU Region:** Retrieved users with IDs 67890, 17821, and 99821 in the EU region.
2. **Checked Admin Status:** Confirmed that only Max had admin rights.
3. **Confirmed Action:** Secured confirmation to proceed with revoking admin rights for Max.
4. **Updated User:** Successfully revoked admin rights for Max, changing his admin status to `false`.

If you need further actions or assistance, feel free to ask!
--------------------
Input: Delete all users that name starts from A
Delete user Alice (User ID: 54321)? Y/N: y
Agent: Alice (User ID: 54321) has been successfully deleted from the system. If you need any additional changes or have further queries, feel free to ask!