CLI Tracing Setup

"""
Auto-generated Handit.ai configuration
Generated by handit-cli setup on 2024-01-15
"""
import os
from handit import HanditTracker
 
# Initialize tracker with your project settings
tracker = HanditTracker()
tracker.config(
    api_key=os.getenv("HANDIT_API_KEY"),
    project_name="my-ai-agent",
    environment=os.getenv("ENVIRONMENT", "development")
)
 
# Export for use across your application
__all__ = ['tracker']

# Handit.ai Configuration
HANDIT_API_KEY=your_integration_token_here
ENVIRONMENT=development

# Add your existing environment variables below

/**
 * Auto-generated Handit.ai configuration
 * Generated by handit-cli setup on 2024-01-15
 */
import { config } from '@handit.ai/node';
 
// Initialize Handit.ai with your project settings
config({
    apiKey: process.env.HANDIT_API_KEY,
    projectName: 'my-ai-agent',
    environment: process.env.NODE_ENV || 'development'
});
 
export default config;

# Handit.ai Configuration
HANDIT_API_KEY=your_integration_token_here
NODE_ENV=development

# Add your existing environment variables below

"""
Auto-generated tracing wrapper for CustomerServiceAgent
Generated by handit-cli setup
"""
from handit_config import tracker
from .customer_service import CustomerServiceAgent as OriginalAgent
 
class CustomerServiceAgent(OriginalAgent):
    """Traced version of CustomerServiceAgent with automatic observability"""
    
    @tracker.start_agent_tracing()
    async def process_customer_request(self, user_message: str, context: dict = None):
        """
        Process customer request with automatic tracing
        Original function wrapped with Handit.ai observability
        """
        return await super().process_customer_request(user_message, context)
    
    @tracker.trace_agent_node("response_generation")
    async def generate_response(self, message: str, history: list = None):
        """
        Generate response with node-level tracing
        Automatically tracks inputs, outputs, and performance
        """
        return await super().generate_response(message, history)

/**
 * Auto-generated tracing wrapper for CustomerServiceAgent
 * Generated by handit-cli setup
 */
import { startAgentTracing, traceAgentNode } from '@handit.ai/node';
import { CustomerServiceAgent as OriginalAgent } from './customerService.js';
import './handit.config.js';
 
export class CustomerServiceAgent extends OriginalAgent {
    /**
     * Process customer request with automatic tracing
     * Original function wrapped with Handit.ai observability
     */
    processCustomerRequest = startAgentTracing(async (userMessage, context = {}) => {
        return await super.processCustomerRequest(userMessage, context);
    });
 
    /**
     * Generate response with node-level tracing
     * Automatically tracks inputs, outputs, and performance
     */
    generateResponse = traceAgentNode({
        agentNodeId: 'response_generation',
        callback: async (message, history = []) => {
            return await super.generateResponse(message, history);
        }
    });
}

# Add custom tracing to the generated wrapper
@tracker.trace_agent_node("intent_analysis")
async def analyze_intent(self, user_message: str):
    """Custom tracing for intent analysis"""
    return await super().analyze_intent(user_message)
 
@tracker.trace_agent_node("knowledge_retrieval")
async def retrieve_knowledge(self, query: str):
    """Custom tracing for knowledge base queries"""
    return await super().retrieve_knowledge(query)

// Add custom tracing to the generated wrapper
analyzeIntent = traceAgentNode({
    agentNodeId: 'intent_analysis',
    callback: async (userMessage) => {
        return await super.analyzeIntent(userMessage);
    }
});
 
retrieveKnowledge = traceAgentNode({
    agentNodeId: 'knowledge_retrieval', 
    callback: async (query) => {
        return await super.retrieveKnowledge(query);
    }
});