Django Integration

pip install AgentToolProtocol django

# settings.py
INSTALLED_APPS = [
    'django.contrib.admin',
    'django.contrib.auth',
    'django.contrib.contenttypes',
    'django.contrib.sessions',
    'django.contrib.messages',
    'django.contrib.staticfiles',
    'django_atp',  # Add this
    'myapp',
]

# urls.py
from django.contrib import admin
from django.urls import path, include

urlpatterns = [
    path('admin/', admin.site.urls),
    path('', include('django_atp.urls')),  # Add this
]

# myapp/atp_tools.py
from atp_sdk.clients import ToolKitClient
from django_atp.registry import register_client

# Initialize the client
client = ToolKitClient(
    api_key="YOUR_ATP_TOOLKIT_API_KEY",
    app_name="my_toolkit"
)

# Register the client
register_client("my_toolkit", client)

# Register tools
@client.register_tool(
    function_name="hello_world",
    params=['name'],
    required_params=['name'],
    description="Returns a greeting.",
    auth_provider=None,
    auth_type=None,
    auth_with=None
)
def hello_world(**kwargs):
    return {"message": f"Hello, {kwargs.get('name')}!"}

@client.register_tool(
    function_name="get_users",
    params=[],
    required_params=[],
    description="Returns a list of users from the database.",
    auth_provider=None,
    auth_type=None,
    auth_with=None
)
def get_users(**kwargs):
    from django.contrib.auth.models import User
    users = User.objects.all().values('id', 'username', 'email')
    return {"users": list(users)}

# myapp/apps.py
from django.apps import AppConfig

class MyAppConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'myapp'
    
    def ready(self):
        import myapp.atp_tools  # Import to register tools

# myapp/__init__.py
default_app_config = 'myapp.apps.MyAppConfig'

curl http://localhost:8000/atp/my_toolkit/

{
  "toolkit": "my_toolkit",
  "tools": ["hello_world", "get_users"]
}

curl http://localhost:8000/atp/my_toolkit/hello_world/

{
  "function_name": "hello_world",
  "params": ["name"],
  "required_params": ["name"],
  "description": "Returns a greeting.",
  "auth_provider": null,
  "auth_type": null,
  "auth_with": null
}

curl -X POST http://localhost:8000/atp/my_toolkit/hello_world/ \
  -H "Content-Type: application/json" \
  -d '{"name": "Alice"}'

{
  "result": {
    "message": "Hello, Alice!"
  }
}

# myapp/atp_tools.py
from atp_sdk.clients import ToolKitClient
from django_atp.registry import register_client
from django.contrib.auth.models import User
import requests

# Initialize the client
client = ToolKitClient(
    api_key="YOUR_ATP_TOOLKIT_API_KEY",
    app_name="django_toolkit"
)

# Register the client
register_client("django_toolkit", client)

# Tool 1: Hello World
@client.register_tool(
    function_name="hello_world",
    params=['name'],
    required_params=['name'],
    description="Returns a greeting.",
    auth_provider=None,
    auth_type=None,
    auth_with=None
)
def hello_world(**kwargs):
    return {"message": f"Hello, {kwargs.get('name')}!"}

# Tool 2: Get Users
@client.register_tool(
    function_name="get_users",
    params=[],
    required_params=[],
    description="Returns a list of users from the database.",
    auth_provider=None,
    auth_type=None,
    auth_with=None
)
def get_users(**kwargs):
    users = User.objects.all().values('id', 'username', 'email')
    return {"users": list(users)}

# Tool 3: Create User
@client.register_tool(
    function_name="create_user",
    params=['username', 'email', 'password'],
    required_params=['username', 'email', 'password'],
    description="Creates a new user in the database.",
    auth_provider=None,
    auth_type=None,
    auth_with=None
)
def create_user(**kwargs):
    user = User.objects.create_user(
        username=kwargs.get('username'),
        email=kwargs.get('email'),
        password=kwargs.get('password')
    )
    return {"user": {"id": user.id, "username": user.username, "email": user.email}}

@client.register_tool(
    function_name="get_posts",
    params=[],
    required_params=[],
    description="Returns a list of blog posts.",
    auth_provider=None,
    auth_type=None,
    auth_with=None
)
def get_posts(**kwargs):
    from myapp.models import Post
    posts = Post.objects.all().values('id', 'title', 'content', 'created_at')
    return {"posts": list(posts)}

@client.register_tool(
    function_name="get_current_user",
    params=['user_id'],
    required_params=['user_id'],
    description="Returns the current user's details.",
    auth_provider=None,
    auth_type=None,
    auth_with=None
)
def get_current_user(**kwargs):
    from django.contrib.auth.models import User
    user = User.objects.get(id=kwargs.get('user_id'))
    return {"user": {"id": user.id, "username": user.username, "email": user.email}}