@promptbook/cli 0.103.0-53 → 0.103.0-54

package/apps/agents-server/src/app/admin/api-tokens/ApiTokensClient.tsx

@@ -0,0 +1,186 @@
+'use client';
+
+import { Copy, Plus, Trash } from 'lucide-react';
+import { useEffect, useState } from 'react';
+
+type ApiTokenEntry = {
+    id: number;
+    token: string;
+    note: string | null;
+    createdAt: string;
+    updatedAt: string;
+    isRevoked: boolean;
+};
+
+export function ApiTokensClient() {
+    const [tokens, setTokens] = useState<ApiTokenEntry[]>([]);
+    const [loading, setLoading] = useState(true);
+    const [error, setError] = useState<string | null>(null);
+    const [note, setNote] = useState('');
+    const [isCreating, setIsCreating] = useState(false);
+
+    const fetchTokens = async () => {
+        try {
+            setLoading(true);
+            const response = await fetch('/api/api-tokens');
+            if (!response.ok) {
+                throw new Error('Failed to fetch tokens');
+            }
+            const data: ApiTokenEntry[] = await response.json();
+            setTokens(data);
+        } catch (err) {
+            setError(err instanceof Error ? err.message : 'An error occurred');
+        } finally {
+            setLoading(false);
+        }
+    };
+
+    useEffect(() => {
+        fetchTokens();
+    }, []);
+
+    const handleCreate = async (e: React.FormEvent) => {
+        e.preventDefault();
+        setError(null);
+        setIsCreating(true);
+
+        try {
+            const response = await fetch('/api/api-tokens', {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({ note }),
+            });
+
+            if (!response.ok) {
+                const data = await response.json();
+                throw new Error(data.error || 'Failed to create token');
+            }
+
+            setNote('');
+            fetchTokens();
+        } catch (err) {
+            setError(err instanceof Error ? err.message : 'An error occurred');
+        } finally {
+            setIsCreating(false);
+        }
+    };
+
+    const handleDelete = async (id: number) => {
+        if (!confirm('Are you sure you want to delete this token?')) return;
+
+        try {
+            const response = await fetch(`/api/api-tokens?id=${id}`, {
+                method: 'DELETE',
+            });
+
+            if (!response.ok) {
+                const data = await response.json();
+                throw new Error(data.error || 'Failed to delete token');
+            }
+
+            fetchTokens();
+        } catch (err) {
+            setError(err instanceof Error ? err.message : 'An error occurred');
+        }
+    };
+
+    const copyToClipboard = (text: string) => {
+        navigator.clipboard.writeText(text);
+        // You could add a toast notification here
+    };
+
+    if (loading && tokens.length === 0) {
+        return <div className="p-8 text-center">Loading tokens...</div>;
+    }
+
+    return (
+        <div className="container mx-auto p-8 max-w-4xl">
+            <h1 className="text-3xl font-bold mb-8">API Tokens</h1>
+
+            {error && (
+                <div className="bg-red-100 border border-red-400 text-red-700 px-4 py-3 rounded mb-6">{error}</div>
+            )}
+
+            <div className="bg-white shadow rounded-lg p-6 mb-8">
+                <h2 className="text-xl font-semibold mb-4">Create New Token</h2>
+                <form onSubmit={handleCreate} className="flex gap-4">
+                    <input
+                        type="text"
+                        value={note}
+                        onChange={(e) => setNote(e.target.value)}
+                        className="flex-grow px-3 py-2 border border-gray-300 rounded-md focus:outline-none focus:ring-2 focus:ring-blue-500"
+                        placeholder="Description (optional)"
+                    />
+                    <button
+                        type="submit"
+                        disabled={isCreating}
+                        className="bg-blue-600 text-white py-2 px-4 rounded-md hover:bg-blue-700 transition-colors flex items-center gap-2 whitespace-nowrap"
+                    >
+                        <Plus className="w-4 h-4" />
+                        {isCreating ? 'Creating...' : 'Create Token'}
+                    </button>
+                </form>
+            </div>
+
+            <div className="bg-white shadow rounded-lg overflow-hidden">
+                <table className="min-w-full divide-y divide-gray-200">
+                    <thead className="bg-gray-50">
+                        <tr>
+                            <th className="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">
+                                Token
+                            </th>
+                            <th className="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">
+                                Note
+                            </th>
+                            <th className="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">
+                                Created At
+                            </th>
+                            <th className="px-6 py-3 text-right text-xs font-medium text-gray-500 uppercase tracking-wider">
+                                Actions
+                            </th>
+                        </tr>
+                    </thead>
+                    <tbody className="bg-white divide-y divide-gray-200">
+                        {tokens.length === 0 ? (
+                            <tr>
+                                <td colSpan={4} className="px-6 py-4 text-center text-gray-500">
+                                    No tokens found.
+                                </td>
+                            </tr>
+                        ) : (
+                            tokens.map((entry) => (
+                                <tr key={entry.id}>
+                                    <td className="px-6 py-4 whitespace-nowrap text-sm font-medium text-gray-900 font-mono">
+                                        <div className="flex items-center gap-2">
+                                            <span>{entry.token}</span>
+                                            <button
+                                                onClick={() => copyToClipboard(entry.token)}
+                                                className="text-gray-400 hover:text-gray-600"
+                                                title="Copy to clipboard"
+                                            >
+                                                <Copy className="w-4 h-4" />
+                                            </button>
+                                        </div>
+                                    </td>
+                                    <td className="px-6 py-4 text-sm text-gray-500">{entry.note || '-'}</td>
+                                    <td className="px-6 py-4 whitespace-nowrap text-sm text-gray-500">
+                                        {new Date(entry.createdAt).toLocaleDateString()}
+                                    </td>
+                                    <td className="px-6 py-4 whitespace-nowrap text-right text-sm font-medium">
+                                        <button
+                                            onClick={() => handleDelete(entry.id)}
+                                            className="text-red-600 hover:text-red-900"
+                                            title="Delete token"
+                                        >
+                                            <Trash className="w-4 h-4" />
+                                        </button>
+                                    </td>
+                                </tr>
+                            ))
+                        )}
+                    </tbody>
+                </table>
+            </div>
+        </div>
+    );
+}

package/apps/agents-server/src/app/admin/api-tokens/page.tsx

@@ -0,0 +1,13 @@
+import { ForbiddenPage } from '../../../components/ForbiddenPage/ForbiddenPage';
+import { isUserAdmin } from '../../../utils/isUserAdmin';
+import { ApiTokensClient } from './ApiTokensClient';
+
+export default async function ApiTokensPage() {
+    const isAdmin = await isUserAdmin();
+
+    if (!isAdmin) {
+        return <ForbiddenPage />;
+    }
+
+    return <ApiTokensClient />;
+}

package/apps/agents-server/src/app/agents/[agentName]/AgentChatWrapper.tsx

@@ -8,12 +8,13 @@ import { string_agent_url } from '../../../../../../src/types/typeAliases';
 
 type AgentChatWrapperProps = {
     agentUrl: string_agent_url;
+    defaultMessage?: string;
 };
 
 // TODO: [🐱‍🚀] Rename to AgentChatSomethingWrapper
 
 export function AgentChatWrapper(props: AgentChatWrapperProps) {
-    const { agentUrl } = props;
+    const { agentUrl, defaultMessage } = props;
 
     const agentPromise = useMemo(
         () =>
@@ -60,7 +61,14 @@ export function AgentChatWrapper(props: AgentChatWrapperProps) {
         return <>{/* <- TODO: [🐱‍🚀] <PromptbookLoading /> */}</>;
     }
 
-    return <AgentChat className={`w-full h-full`} agent={agent} onFeedback={handleFeedback} />;
+    return (
+        <AgentChat
+            className={`w-full h-full`}
+            agent={agent}
+            onFeedback={handleFeedback}
+            defaultMessage={defaultMessage}
+        />
+    );
 }
 
 /**

package/apps/agents-server/src/app/agents/[agentName]/api/openai/chat/completions/route.ts

@@ -0,0 +1,176 @@
+import { $provideAgentCollectionForServer } from '@/src/tools/$provideAgentCollectionForServer';
+import { $provideExecutionToolsForServer } from '@/src/tools/$provideExecutionToolsForServer';
+import { Agent } from '@promptbook-local/core';
+import { ChatMessage, ChatPromptResult, Prompt, TODO_any } from '@promptbook-local/types';
+import { NextRequest, NextResponse } from 'next/server';
+
+export async function POST(
+    request: NextRequest,
+    { params }: { params: Promise<{ agentName: string }> },
+) {
+    const { agentName } = await params;
+    
+    // Note: Authentication is handled by middleware
+    // If we are here, the request is either authenticated or public access is allowed (but middleware blocks it if not)
+
+    try {
+        const body = await request.json();
+        const { messages, stream, model } = body;
+
+        if (!messages || !Array.isArray(messages) || messages.length === 0) {
+            return NextResponse.json(
+                { error: { message: 'Messages array is required and cannot be empty.', type: 'invalid_request_error' } },
+                { status: 400 },
+            );
+        }
+
+        const collection = await $provideAgentCollectionForServer();
+        let agentSource;
+        try {
+            agentSource = await collection.getAgentSource(agentName);
+        } catch (error) {
+            return NextResponse.json(
+                { error: { message: `Agent '${agentName}' not found.`, type: 'invalid_request_error' } },
+                { status: 404 },
+            );
+        }
+
+        if (!agentSource) {
+            return NextResponse.json(
+                { error: { message: `Agent '${agentName}' not found.`, type: 'invalid_request_error' } },
+                { status: 404 },
+            );
+        }
+
+        const executionTools = await $provideExecutionToolsForServer();
+        const agent = new Agent({
+            agentSource,
+            executionTools,
+            isVerbose: true, // or false
+        });
+
+        // Prepare thread and content
+        const lastMessage = messages[messages.length - 1];
+        const previousMessages = messages.slice(0, -1);
+
+        const thread: ChatMessage[] = previousMessages.map((msg: TODO_any, index: number) => ({
+            id: `msg-${index}`, // Placeholder ID
+            from: msg.role === 'assistant' ? 'agent' : 'user', // Mapping standard OpenAI roles
+            content: msg.content,
+            isComplete: true,
+            date: new Date(), // We don't have the real date, using current
+        }));
+
+        const prompt: Prompt = {
+            title: 'OpenAI API Chat Completion',
+            content: lastMessage.content,
+            modelRequirements: {
+                modelVariant: 'CHAT',
+                // We could pass 'model' from body if we wanted to enforce it, but Agent usually has its own config
+            },
+            parameters: {},
+            thread,
+        } as Prompt; 
+        // Note: Casting as Prompt because the type definition might require properties we don't strictly use or that are optional but TS complains
+
+        if (stream) {
+            const encoder = new TextEncoder();
+            const readableStream = new ReadableStream({
+                async start(controller) {
+                    const runId = `chatcmpl-${Math.random().toString(36).substring(2, 15)}`;
+                    const created = Math.floor(Date.now() / 1000);
+
+                    let previousContent = '';
+
+                    try {
+                        await agent.callChatModelStream(prompt, (chunk: ChatPromptResult) => {
+                            const fullContent = chunk.content;
+                            const deltaContent = fullContent.substring(previousContent.length);
+                            previousContent = fullContent;
+
+                            if (deltaContent) {
+                                const chunkData = {
+                                    id: runId,
+                                    object: 'chat.completion.chunk',
+                                    created,
+                                    model: model || 'promptbook-agent',
+                                    choices: [
+                                        {
+                                            index: 0,
+                                            delta: {
+                                                content: deltaContent,
+                                            },
+                                            finish_reason: null,
+                                        },
+                                    ],
+                                };
+                                controller.enqueue(encoder.encode(`data: ${JSON.stringify(chunkData)}\n\n`));
+                            }
+                        });
+
+                        const doneChunkData = {
+                            id: runId,
+                            object: 'chat.completion.chunk',
+                            created,
+                            model: model || 'promptbook-agent',
+                            choices: [
+                                {
+                                    index: 0,
+                                    delta: {},
+                                    finish_reason: 'stop',
+                                },
+                            ],
+                        };
+                        controller.enqueue(encoder.encode(`data: ${JSON.stringify(doneChunkData)}\n\n`));
+                        controller.enqueue(encoder.encode('[DONE]'));
+                    } catch (error) {
+                        console.error('Error during streaming:', error);
+                        // OpenAI stream doesn't usually send error JSON in stream, just closes or sends error text?
+                        // But we should try to close gracefully or error.
+                        controller.error(error);
+                    }
+                    controller.close();
+                },
+            });
+
+            return new Response(readableStream, {
+                headers: {
+                    'Content-Type': 'text/event-stream',
+                    'Cache-Control': 'no-cache',
+                    Connection: 'keep-alive',
+                },
+            });
+        } else {
+             const result = await agent.callChatModel(prompt);
+             
+             return NextResponse.json({
+                id: `chatcmpl-${Math.random().toString(36).substring(2, 15)}`,
+                object: 'chat.completion',
+                created: Math.floor(Date.now() / 1000),
+                model: model || 'promptbook-agent',
+                choices: [
+                    {
+                        index: 0,
+                        message: {
+                            role: 'assistant',
+                            content: result.content,
+                        },
+                        finish_reason: 'stop',
+                    },
+                ],
+                usage: {
+                    prompt_tokens: result.usage?.input?.tokensCount?.value || 0,
+                    completion_tokens: result.usage?.output?.tokensCount?.value || 0,
+                    total_tokens: (result.usage?.input?.tokensCount?.value || 0) + (result.usage?.output?.tokensCount?.value || 0),
+                },
+             });
+        }
+
+    } catch (error) {
+        console.error('Error in OpenAI API handler:', error);
+        return NextResponse.json(
+            { error: { message: (error as Error).message || 'Internal Server Error', type: 'server_error' } },
+            { status: 500 },
+        );
+    }
+}

package/apps/agents-server/src/app/agents/[agentName]/page.tsx

@@ -3,7 +3,7 @@
 // import { BookEditor } from '@promptbook-local/components';
 import { $provideServer } from '@/src/tools/$provideServer';
 import { PROMPTBOOK_COLOR } from '@promptbook-local/core';
-import { CodeIcon, HistoryIcon, NotebookPenIcon } from 'lucide-react';
+import { CodeIcon, HistoryIcon, MessageCircleQuestionIcon, MessageSquareIcon, NotebookPenIcon } from 'lucide-react';
 import { headers } from 'next/headers';
 import { notFound } from 'next/navigation';
 import { Color } from '../../../../../../src/utils/color/Color';
@@ -26,13 +26,20 @@ import { isUserAdmin } from '../../../utils/isUserAdmin';
 
 export const generateMetadata = generateAgentMetadata;
 
-export default async function AgentPage({ params }: { params: Promise<{ agentName: string }> }) {
+export default async function AgentPage({
+    params,
+    searchParams,
+}: {
+    params: Promise<{ agentName: string }>;
+    searchParams: Promise<{ [key: string]: string | string[] | undefined }>;
+}) {
     // const [apiKey, setApiKey] = useStateInLocalStorage<string>('openai-apiKey', () => '');
     // const [isApiKeyVisible, setIsApiKeyVisible] = useState(false);
     // const [isApiKeySectionCollapsed, setIsApiKeySectionCollapsed] = useState(!!apiKey);
 
     $sideEffect(headers());
 
+    const { message } = await searchParams;
     const agentName = await getAgentName(params);
     const isAdmin = await isUserAdmin();
 
@@ -155,6 +162,20 @@ export default async function AgentPage({ params }: { params: Promise<{ agentNam
                                 Maintenance
                             </h2>
                             <div className="flex flex-col gap-2">
+                                <a
+                                    href={`/admin/chat-history?agentName=${encodeURIComponent(agentName)}`}
+                                    className="inline-flex items-center justify-center whitespace-nowrap rounded-md border border-gray-300 bg-white px-3 py-1.5 text-xs font-semibold text-gray-700 shadow-sm hover:bg-gray-50"
+                                >
+                                    <MessageSquareIcon className="mr-2 w-3 h-3" />
+                                    Chat history
+                                </a>
+                                <a
+                                    href={`/admin/chat-feedback?agentName=${encodeURIComponent(agentName)}`}
+                                    className="inline-flex items-center justify-center whitespace-nowrap rounded-md border border-gray-300 bg-white px-3 py-1.5 text-xs font-semibold text-gray-700 shadow-sm hover:bg-gray-50"
+                                >
+                                    <MessageCircleQuestionIcon className="mr-2 w-3 h-3" />
+                                    Chat feedback
+                                </a>
                                 <ClearAgentChatHistoryButton agentName={agentName} />
                                 <ClearAgentChatFeedbackButton agentName={agentName} />
                             </div>
@@ -181,7 +202,7 @@ export default async function AgentPage({ params }: { params: Promise<{ agentNam
 
             {/* Main content: Chat */}
             <div className="flex-1 relative h-full bg-white">
-                <AgentChatWrapper agentUrl={agentUrl} />
+                <AgentChatWrapper agentUrl={agentUrl} defaultMessage={message as string} />
             </div>
         </div>
     );

package/apps/agents-server/src/app/api/api-tokens/route.ts

@@ -0,0 +1,76 @@
+import { $getTableName } from '../../../database/$getTableName';
+import { $provideSupabase } from '../../../database/$provideSupabase';
+import { isUserAdmin } from '../../../utils/isUserAdmin';
+import { randomUUID } from 'crypto';
+import { NextRequest, NextResponse } from 'next/server';
+
+export async function GET(request: NextRequest) {
+    if (!(await isUserAdmin())) {
+        return NextResponse.json({ error: 'Unauthorized' }, { status: 401 });
+    }
+
+    const supabase = $provideSupabase();
+    const table = await $getTableName('ApiTokens');
+
+    const { data, error } = await supabase.from(table).select('*').order('createdAt', { ascending: false });
+
+    if (error) {
+        return NextResponse.json({ error: error.message }, { status: 500 });
+    }
+
+    return NextResponse.json(data);
+}
+
+export async function POST(request: NextRequest) {
+    if (!(await isUserAdmin())) {
+        return NextResponse.json({ error: 'Unauthorized' }, { status: 401 });
+    }
+
+    try {
+        const body = await request.json();
+        const { note } = body;
+
+        const token = `ptbk_${randomUUID().replace(/-/g, '')}`;
+
+        const supabase = $provideSupabase();
+        const table = await $getTableName('ApiTokens');
+
+        const { data, error } = await supabase
+            .from(table)
+            .insert({ token, note })
+            .select()
+            .single();
+
+        if (error) {
+            return NextResponse.json({ error: error.message }, { status: 500 });
+        }
+
+        return NextResponse.json(data);
+    } catch (e) {
+        return NextResponse.json({ error: 'Invalid request body' }, { status: 400 });
+    }
+}
+
+export async function DELETE(request: NextRequest) {
+    if (!(await isUserAdmin())) {
+        return NextResponse.json({ error: 'Unauthorized' }, { status: 401 });
+    }
+
+    const searchParams = request.nextUrl.searchParams;
+    const id = searchParams.get('id');
+
+    if (!id) {
+        return NextResponse.json({ error: 'ID is required' }, { status: 400 });
+    }
+
+    const supabase = $provideSupabase();
+    const table = await $getTableName('ApiTokens');
+
+    const { error } = await supabase.from(table).delete().eq('id', parseInt(id, 10));
+
+    if (error) {
+        return NextResponse.json({ error: error.message }, { status: 500 });
+    }
+
+    return NextResponse.json({ success: true });
+}

package/apps/agents-server/src/app/docs/[docId]/page.tsx

@@ -30,6 +30,7 @@ export default async function DocPage(props: DocPageProps) {
                     <div className="p-8 border-b border-gray-100 bg-gray-50/50">
                         <div className="flex items-center gap-4 mb-4">
                             <h1 className="text-4xl font-bold text-gray-900 tracking-tight">
+                                <span className="mr-3">{primary.icon}</span>
                                 {primary.type}
                                 {aliases.length > 0 && (
                                     <span className="text-gray-400 font-normal ml-4 text-2xl">

package/apps/agents-server/src/app/docs/page.tsx

@@ -14,6 +14,7 @@ export default function DocsPage() {
                         <Link key={primary.type} href={`/docs/${primary.type}`} className="block h-full group">
                             <Card className="h-full group-hover:border-blue-500 transition-colors">
                                 <h3 className="text-xl font-semibold mb-2 group-hover:text-blue-600 transition-colors">
+                                    <span className="mr-2">{primary.icon}</span>
                                     {primary.type}
                                     {aliases.length > 0 && (
                                         <span className="text-gray-400 font-normal text-lg">

package/apps/agents-server/src/components/Header/Header.tsx

@@ -209,6 +209,11 @@ export function Header(props: HeaderProps) {
                           } as SubMenuItem,
                       ],
                   },
+                  {
+                      type: 'link' as const,
+                      label: 'API Tokens',
+                      href: '/admin/api-tokens',
+                  },
                   {
                       type: 'link' as const,
                       label: 'Metadata',

package/apps/agents-server/src/components/LayoutWrapper/LayoutWrapper.tsx

@@ -31,6 +31,7 @@ export function LayoutWrapper({
     const isAdminChatPage =
         pathname?.startsWith('/admin/chat-history') || pathname?.startsWith('/admin/chat-feedback');
     const isHeaderHidden = pathname?.includes('/chat') && !isAdminChatPage;
+    const isFooterHiddenOnPage = pathname ? /^\/agents\/[^/]+\/book(\+chat)?$/.test(pathname) : false;
 
     if (isHeaderHidden) {
         return <main className={`pt-0`}>{children}</main>;
@@ -46,7 +47,7 @@ export function LayoutWrapper({
                 agents={agents}
             />
             <main className={`pt-[60px]`}>{children}</main>
-            {isFooterShown && <Footer extraLinks={footerLinks} />}
+            {isFooterShown && !isFooterHiddenOnPage && <Footer extraLinks={footerLinks} />}
         </>
     );
 }

package/apps/agents-server/src/database/migrations/2025-12-0010-llm-cache.sql

@@ -0,0 +1,12 @@
+CREATE TABLE IF NOT EXISTS "prefix_LlmCache" (
+    "id" BIGINT GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY,
+    "createdAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(),
+    "updatedAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(),
+
+    "hash" TEXT NOT NULL,
+    "value" JSONB NOT NULL
+);
+
+CREATE UNIQUE INDEX IF NOT EXISTS "prefix_LlmCache_hash_idx" ON "prefix_LlmCache" ("hash");
+
+ALTER TABLE "prefix_LlmCache" ENABLE ROW LEVEL SECURITY;

package/apps/agents-server/src/database/migrations/2025-12-0060-api-tokens.sql

@@ -0,0 +1,13 @@
+CREATE TABLE IF NOT EXISTS "prefix_ApiTokens" (
+    "id" BIGINT GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY,
+    "createdAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(),
+    "updatedAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT now(),
+
+    "token" TEXT NOT NULL,
+    "note" TEXT,
+    "isRevoked" BOOLEAN NOT NULL DEFAULT FALSE
+);
+
+CREATE UNIQUE INDEX IF NOT EXISTS "prefix_ApiTokens_token_idx" ON "prefix_ApiTokens" ("token");
+
+ALTER TABLE "prefix_ApiTokens" ENABLE ROW LEVEL SECURITY;