epsimo-agent 0.1.0

package/epsimo/resources/credits.py

@@ -0,0 +1,16 @@
+class Credits:
+    def __init__(self, client):
+        self.client = client
+
+    def get_balance(self):
+        """Retrieve the current thread and credit balance."""
+        # Using the thread-info endpoint as it contains current balance data
+        return self.client.request("GET", "/auth/thread-info")
+
+    def create_checkout_session(self, quantity, total_amount):
+        """Create a checkout session to buy credits."""
+        payload = {
+            "quantity": quantity,
+            "total_amount": float(total_amount)
+        }
+        return self.client.request("POST", "/checkout/create-checkout-session", json=payload)

package/epsimo/resources/db.py

@@ -0,0 +1,31 @@
+class Database:
+    """
+    The Database resource allows using Epsimo threads as a virtual structured storage.
+    It wraps thread state management into a familiar key-value or document-based interface.
+    """
+    def __init__(self, client):
+        self.client = client
+
+    def get_all(self, project_id, thread_id):
+        """Retrieve all structured data stored in the thread state."""
+        state = self.client.threads.get_state(project_id, thread_id)
+        return state.get("values", {})
+
+    def get(self, project_id, thread_id, key, default=None):
+        """Retrieve a specific key from the thread state."""
+        values = self.get_all(project_id, thread_id)
+        if isinstance(values, dict):
+            return values.get(key, default)
+        return default
+
+    def set(self, project_id, thread_id, key, value):
+        """
+        Store a value in the thread state. 
+        Note: This currently attempts a manual state update which may be restricted 
+        depending on assistant configuration.
+        """
+        return self.client.threads.set_state(project_id, thread_id, {key: value})
+
+    def update(self, project_id, thread_id, data):
+        """Bulk update the thread state with a dictionary of values."""
+        return self.client.threads.set_state(project_id, thread_id, data)

package/epsimo/resources/files.py

@@ -0,0 +1,39 @@
+import os
+
+class Files:
+    def __init__(self, client):
+        self.client = client
+
+    def list(self, project_id, assistant_id):
+        """List files attached to an assistant."""
+        headers = self.client.get_project_headers(project_id)
+        return self.client.request("GET", f"/assistants/{assistant_id}/files", headers=headers)
+
+    def upload(self, project_id, assistant_id, file_path):
+        """Upload a file to an assistant."""
+        headers = self.client.get_project_headers(project_id)
+        # requests handles Content-Type for files
+        # We need to act directly on the session or expose a clean method in client
+        
+        # client.request takes json/params but not files explicitly in my simple wrapper
+        # Let's bypass wrapper slightly or extend it, but here we construct full URL
+        # reusing client.base_url
+        
+        url = f"{self.client.base_url}/assistants/{assistant_id}/files"
+        
+        # We need to strip standard JSON content type if present
+        # but headers only has Auth here.
+        
+        with open(file_path, 'rb') as f:
+            files = {'files': (os.path.basename(file_path), f)}
+            # accessing _session directly for file upload flexibility
+            resp = self.client._session.post(url, headers=headers, files=files)
+            
+        if not resp.ok:
+            resp.raise_for_status()
+        return resp.json()
+
+    def delete(self, project_id, assistant_id, file_id):
+        """Delete a file from an assistant."""
+        headers = self.client.get_project_headers(project_id)
+        return self.client.request("DELETE", f"/assistants/{assistant_id}/files/{file_id}", headers=headers)

package/epsimo/resources/projects.py

@@ -0,0 +1,30 @@
+class Projects:
+    def __init__(self, client):
+        self.client = client
+
+    def list(self):
+        """List all projects."""
+        return self.client.request("GET", "/projects/")
+
+    def create(self, name, description="Epsimo Project"):
+        """Create a new project."""
+        payload = {"name": name, "description": description}
+        return self.client.request("POST", "/projects/", json=payload)
+
+    def get(self, project_id):
+        """Get project details (and token context switching)."""
+        return self.client.request("GET", f"/projects/{project_id}")
+
+    def update(self, project_id, name=None, description=None):
+        """Update a project."""
+        payload = {}
+        if name: payload["name"] = name
+        if description: payload["description"] = description
+        return self.client.request("PUT", f"/projects/{project_id}", json=payload)
+
+    def delete(self, project_id, confirm=False):
+        """Delete a project."""
+        url = f"/projects/{project_id}"
+        if confirm:
+            url += "?confirm=true"
+        return self.client.request("DELETE", url)

package/epsimo/resources/threads.py

@@ -0,0 +1,83 @@
+import json
+
+class Threads:
+    def __init__(self, client):
+        self.client = client
+
+    def list(self, project_id):
+        """List threads in a project."""
+        headers = self.client.get_project_headers(project_id)
+        return self.client.request("GET", "/threads/", headers=headers)
+
+    def create(self, project_id, name, assistant_id, metadata=None):
+        """Create a new thread."""
+        headers = self.client.get_project_headers(project_id)
+        payload = {
+            "name": name,
+            "assistant_id": assistant_id,
+            "metadata": metadata or {"type": "thread"}
+        }
+        return self.client.request("POST", "/threads/", json=payload, headers=headers)
+
+    def get(self, project_id, thread_id):
+        """Get thread details."""
+        headers = self.client.get_project_headers(project_id)
+        return self.client.request("GET", f"/threads/{thread_id}", headers=headers)
+
+    def get_state(self, project_id, thread_id):
+        """Retrieve the structured state (values) of a thread."""
+        headers = self.client.get_project_headers(project_id)
+        return self.client.request("GET", f"/threads/{thread_id}/state", headers=headers)
+
+    def set_state(self, project_id, thread_id, values, config=None):
+        """Update the structured state (values) of a thread."""
+        headers = self.client.get_project_headers(project_id)
+        payload = {
+            "values": values,
+            "config": config or {}
+        }
+        return self.client.request("POST", f"/threads/{thread_id}/state", json=payload, headers=headers)
+
+    # --- Runs (Streaming) ---
+    # Putting this here for convenience as Runs are usually per-thread/assistant
+    
+    def run_stream(self, project_id, thread_id, assistant_id, message, stream_mode=None):
+        """
+        Stream a run and yield chunks.
+        
+        Args:
+            project_id: Project ID
+            thread_id: Thread ID
+            assistant_id: Assistant ID
+            message: User message input string
+            stream_mode: List of modes, e.g. ["messages", "values"]
+            
+        Yields:
+            Parsed JSON chunks from the SSE stream.
+        """
+        headers = self.client.get_project_headers(project_id)
+        headers["Accept"] = "text/event-stream"
+        
+        payload = {
+            "thread_id": thread_id,
+            "assistant_id": assistant_id,
+            "input": [{"role": "user", "content": message, "type": "human"}],
+            "stream_mode": stream_mode or ["messages", "values"]
+        }
+        
+        # Bypass client.request to handle streaming
+        url = f"{self.client.base_url}/runs/stream"
+        response = self.client._session.post(url, json=payload, headers=headers, stream=True)
+        response.raise_for_status()
+
+        for line in response.iter_lines():
+            if line:
+                decoded = line.decode('utf-8')
+                if decoded.startswith("data:"):
+                    data_str = decoded[5:].strip()
+                    if data_str == "[DONE]":
+                        break
+                    try:
+                        yield json.loads(data_str)
+                    except json.JSONDecodeError:
+                        yield {"raw": data_str, "error": "json_decode_error"}

package/epsimo/templates/components/AuthModal/AuthModal.module.css

@@ -0,0 +1,39 @@
+.footer {
+    padding: 1rem 0 0;
+    margin-top: 1rem;
+    border-top: 1px solid var(--border-color);
+    text-align: center;
+    font-size: 0.875rem;
+    color: var(--text-secondary);
+}
+
+.switchBtn {
+    background: none;
+    border: none;
+    color: var(--color-primary);
+    font-weight: 500;
+    cursor: pointer;
+    padding: 0 4px;
+}
+
+.switchBtn:hover {
+    text-decoration: underline;
+}
+
+.form {
+    display: flex;
+    flex-direction: column;
+    gap: 1rem;
+}
+
+.errorAlert {
+    display: flex;
+    align-items: center;
+    gap: 8px;
+    padding: 10px;
+    background-color: rgba(220, 38, 38, 0.1);
+    border: 1px solid rgba(220, 38, 38, 0.2);
+    border-radius: var(--radius-md);
+    color: #ef4444;
+    font-size: 0.875rem;
+}

package/epsimo/templates/components/AuthModal/AuthModal.tsx

@@ -0,0 +1,138 @@
+"use client";
+
+import React, { useState } from 'react';
+import { useAuth } from '@/contexts/AuthContext'; // Assumes user has this context
+import { api } from '@/lib/api-client'; // Assumes user has this client
+import { Modal } from '@/components/ui/Modal/Modal';
+import { Input } from '@/components/ui/Input/Input';
+import { Button } from '@/components/ui/Button/Button';
+import styles from './AuthModal.module.css';
+
+interface AuthModalProps {
+    isOpen: boolean;
+    onClose: () => void;
+    onSuccess?: () => void;
+    defaultTab?: 'login' | 'signup';
+}
+
+export const AuthModal: React.FC<AuthModalProps> = ({
+    isOpen,
+    onClose,
+    onSuccess,
+    defaultTab = 'login'
+}) => {
+    const [mode, setMode] = useState<'login' | 'signup'>(defaultTab);
+    const [email, setEmail] = useState('');
+    const [password, setPassword] = useState('');
+    const [name, setName] = useState('');
+    const [isLoading, setIsLoading] = useState(false);
+    const [error, setError] = useState<string | null>(null);
+
+    const { login } = useAuth();
+
+    const handleSubmit = async (e: React.FormEvent) => {
+        e.preventDefault();
+        setIsLoading(true);
+        setError(null);
+
+        try {
+            if (mode === 'login') {
+                const response = await api.post('/auth/login', {
+                    body: { email, password }
+                });
+                login(response.jwt_token, response.main_agent_id, response.main_thread_id);
+            } else {
+                const response = await api.post('/auth/signup', {
+                    body: { email, password, name }
+                });
+                // Auto-login after signup
+                if (response.jwt_token) {
+                    login(response.jwt_token, response.main_agent_id, response.main_thread_id);
+                }
+            }
+
+            if (onSuccess) onSuccess();
+            onClose();
+        } catch (err: any) {
+            console.error('Auth error:', err);
+            setError(err.message || `Failed to ${mode}. Please check your details.`);
+        } finally {
+            setIsLoading(false);
+        }
+    };
+
+    return (
+        <Modal
+            isOpen={isOpen}
+            onClose={onClose}
+            title={mode === 'login' ? 'Welcome Back' : 'Create Account'}
+            size="sm"
+            footer={
+                <div className={styles.footer}>
+                    <p>
+                        {mode === 'login' ? "Don't have an account? " : "Already have an account? "}
+                        <button
+                            type="button"
+                            className={styles.switchBtn}
+                            onClick={() => {
+                                setMode(mode === 'login' ? 'signup' : 'login');
+                                setError(null);
+                            }}
+                        >
+                            {mode === 'login' ? 'Sign up' : 'Log in'}
+                        </button>
+                    </p>
+                </div>
+            }
+        >
+            <form onSubmit={handleSubmit} className={styles.form}>
+                {mode === 'signup' && (
+                    <Input
+                        label="Full Name"
+                        placeholder="John Doe"
+                        value={name}
+                        onChange={(e) => setName(e.target.value)}
+                        required
+                        fullWidth
+                    />
+                )}
+
+                <Input
+                    type="email"
+                    label="Email"
+                    placeholder="name@example.com"
+                    value={email}
+                    onChange={(e) => setEmail(e.target.value)}
+                    required
+                    fullWidth
+                />
+
+                <Input
+                    type="password"
+                    label="Password"
+                    placeholder="••••••••"
+                    value={password}
+                    onChange={(e) => setPassword(e.target.value)}
+                    required
+                    fullWidth
+                    minLength={8}
+                />
+
+                {error && (
+                    <div className={styles.errorAlert}>
+                        <svg width="16" height="16" viewBox="0 0 24 24" fill="none" stroke="currentColor" strokeWidth="2"><circle cx="12" cy="12" r="10"></circle><line x1="12" y1="8" x2="12" y2="12"></line><line x1="12" y1="16" x2="12.01" y2="16"></line></svg>
+                        <span>{error}</span>
+                    </div>
+                )}
+
+                <Button
+                    type="submit"
+                    fullWidth
+                    isLoading={isLoading}
+                >
+                    {mode === 'login' ? 'Sign In' : 'Create Account'}
+                </Button>
+            </form>
+        </Modal>
+    );
+};

package/epsimo/templates/components/BuyCredits/BuyCreditsModal.module.css

@@ -0,0 +1,96 @@
+.pricingGrid {
+    display: grid;
+    grid-template-columns: repeat(3, 1fr);
+    gap: 16px;
+    margin-top: 24px;
+}
+
+.pricingCard {
+    background: var(--background-surface);
+    border: 1px solid var(--border-color);
+    border-radius: var(--radius-lg);
+    padding: 24px;
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    text-align: center;
+    transition: all 0.2s ease;
+    cursor: pointer;
+    position: relative;
+    overflow: hidden;
+}
+
+.pricingCard:hover {
+    border-color: var(--color-primary);
+    transform: translateY(-2px);
+    box-shadow: var(--shadow-md);
+}
+
+.selectedCard {
+    border-color: var(--color-primary);
+    background: rgba(var(--color-primary-rgb), 0.03);
+    box-shadow: 0 0 0 2px var(--color-primary-light);
+}
+
+.popularBadge {
+    position: absolute;
+    top: 12px;
+    right: 12px;
+    background: var(--color-primary);
+    color: white;
+    font-size: 0.7rem;
+    padding: 2px 8px;
+    border-radius: 12px;
+    font-weight: 600;
+}
+
+.threads {
+    font-size: 2rem;
+    font-weight: 700;
+    color: var(--text-primary);
+    margin: 8px 0;
+}
+
+.price {
+    font-size: 1.1rem;
+    color: var(--text-secondary);
+    margin-bottom: 24px;
+}
+
+.customInput {
+    margin-top: 24px;
+    width: 100%;
+    padding-top: 24px;
+    border-top: 1px solid var(--border-color);
+}
+
+.loadingOverlay {
+    position: absolute;
+    inset: 0;
+    background: rgba(255, 255, 255, 0.8);
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    z-index: 10;
+}
+
+.spinner {
+    width: 24px;
+    height: 24px;
+    border: 2px solid var(--border-color);
+    border-top: 2px solid var(--color-primary);
+    border-radius: 50%;
+    animation: spin 1s linear infinite;
+}
+
+@keyframes spin {
+    to {
+        transform: rotate(360deg);
+    }
+}
+
+@media (max-width: 640px) {
+    .pricingGrid {
+        grid-template-columns: 1fr;
+    }
+}

package/epsimo/templates/components/BuyCredits/BuyCreditsModal.tsx

@@ -0,0 +1,132 @@
+"use client";
+
+import React, { useState } from 'react';
+import { api } from '@/lib/api-client';
+import { Modal } from '@/components/ui/Modal/Modal';
+import { Button } from '@/components/ui/Button/Button';
+import { Input } from '@/components/ui/Input/Input';
+import styles from './BuyCreditsModal.module.css';
+
+interface BuyCreditsModalProps {
+    isOpen: boolean;
+    onClose: () => void;
+    tokenPrice?: number; // Price per token/thread
+}
+
+export const BuyCreditsModal: React.FC<BuyCreditsModalProps> = ({
+    isOpen,
+    onClose,
+    tokenPrice = 0.10 // Default fallback price
+}) => {
+    const [quantity, setQuantity] = useState<number>(100);
+    const [isCustom, setIsCustom] = useState(false);
+    const [isLoading, setIsLoading] = useState(false);
+    const [error, setError] = useState<string | null>(null);
+
+    const packages = [
+        { name: 'Starter', threads: 100, price: 10, popular: false },
+        { name: 'Pro', threads: 500, price: 45, popular: true },
+        { name: 'Max', threads: 1000, price: 80, popular: false },
+    ];
+
+    const currentPrice = isCustom ? (quantity * tokenPrice).toFixed(2) : (packages.find(p => p.threads === quantity)?.price || (quantity * tokenPrice)).toFixed(2);
+
+    const handleCheckout = async () => {
+        setIsLoading(true);
+        setError(null);
+
+        try {
+            const response: any = await api.post('/checkout/create-checkout-session', {
+                body: {
+                    quantity,
+                    total_amount: parseFloat(currentPrice)
+                }
+            });
+
+            if (response && response.url) {
+                window.location.href = response.url;
+            } else {
+                throw new Error('Invalid checkout response');
+            }
+        } catch (err: any) {
+            console.error('Checkout error:', err);
+            setError(err.message || 'Failed to start checkout');
+            setIsLoading(false);
+        }
+    };
+
+    return (
+        <Modal
+            isOpen={isOpen}
+            onClose={onClose}
+            title="Add More Threads"
+            size="lg"
+            footer={
+                <div style={{ display: 'flex', justifyContent: 'flex-end', gap: '12px', alignItems: 'center', width: '100%' }}>
+                    <div style={{ marginRight: 'auto', fontWeight: 'bold' }}>
+                        Total: €{currentPrice}
+                    </div>
+                    <Button variant="ghost" onClick={onClose} disabled={isLoading}>
+                        Cancel
+                    </Button>
+                    <Button onClick={handleCheckout} isLoading={isLoading}>
+                        Proceed to Checkout
+                    </Button>
+                </div>
+            }
+        >
+            <div className={styles.container}>
+                {error && (
+                    <div className="bg-red-50 text-red-600 p-3 rounded-md mb-4 text-sm">
+                        {error}
+                    </div>
+                )}
+
+                <p style={{ color: 'var(--text-secondary)', marginBottom: '20px' }}>
+                    Purchase additional threads to continue your conversations.
+                </p>
+
+                <div className={styles.pricingGrid}>
+                    {packages.map((pkg) => (
+                        <div
+                            key={pkg.name}
+                            className={`${styles.pricingCard} ${quantity === pkg.threads && !isCustom ? styles.selectedCard : ''}`}
+                            onClick={() => {
+                                setQuantity(pkg.threads);
+                                setIsCustom(false);
+                            }}
+                        >
+                            {pkg.popular && <span className={styles.popularBadge}>Most Popular</span>}
+                            <div className={styles.threads}>{pkg.threads}</div>
+                            <div className={styles.price}>Threads</div>
+                            <div style={{ marginTop: 'auto', fontSize: '1.25rem', fontWeight: 'bold' }}>€{pkg.price}</div>
+                        </div>
+                    ))}
+                </div>
+
+                <div className={styles.customInput}>
+                    <Button
+                        variant="ghost"
+                        fullWidth
+                        onClick={() => setIsCustom(!isCustom)}
+                        style={{ marginBottom: isCustom ? '16px' : '0' }}
+                    >
+                        {isCustom ? 'Hide Custom Amount' : 'Enter Custom Amount'}
+                    </Button>
+
+                    {isCustom && (
+                        <Input
+                            type="number"
+                            label="Number of Threads"
+                            value={quantity}
+                            onChange={(e) => setQuantity(Math.max(1, parseInt(e.target.value) || 0))}
+                            min={1}
+                            placeholder={`Approx. €${(quantity * tokenPrice).toFixed(2)}`}
+                            fullWidth
+                        />
+                    )}
+                </div>
+            </div>
+        </Modal>
+    );
+};