praisonai 1.5.4 → 1.7.0

package/README.md

@@ -33,7 +33,7 @@ Here are examples of different ways to use PraisonAI:
 ### 1. Single Agent Example
 
 ```typescript
-import { Agent, PraisonAIAgents } from 'praisonai';
+import { Agent, AgentTeam } from 'praisonai';
 
 async function main() {
     // Create a simple agent (no task specified)
@@ -43,8 +43,8 @@ async function main() {
         verbose: true
     });
 
-    // Run the agent
-    const praisonAI = new PraisonAIAgents({
+    // Run the agent with AgentTeam
+    const team = new AgentTeam({
         agents: [agent],
         tasks: ["Explain the process of photosynthesis in detail."],
         verbose: true
@@ -52,7 +52,7 @@ async function main() {
 
     try {
         console.log('Starting single agent example...');
-        const results = await praisonAI.start();
+        const results = await team.start();
         console.log('\nFinal Results:', results);
     } catch (error) {
         console.error('Error:', error);
@@ -65,7 +65,7 @@ main();
 ### 2. Multi-Agent Example
 
 ```typescript
-import { Agent, PraisonAIAgents } from 'praisonai';
+import { Agent, AgentTeam } from 'praisonai';
 
 async function main() {
     // Create multiple agents with different roles
@@ -87,8 +87,8 @@ async function main() {
         verbose: true
     });
 
-    // Run the agents in sequence
-    const praisonAI = new PraisonAIAgents({
+    // Run the agents in sequence with AgentTeam
+    const team = new AgentTeam({
         agents: [researchAgent, summaryAgent, recommendationAgent],
         tasks: [
             "Research and analyze current renewable energy technologies and their implementation.",
@@ -100,7 +100,7 @@ async function main() {
 
     try {
         console.log('Starting multi-agent example...');
-        const results = await praisonAI.start();
+        const results = await team.start();
         console.log('\nFinal Results:', results);
     } catch (error) {
         console.error('Error:', error);
@@ -113,7 +113,7 @@ main();
 ### 3. Task-Based Agent Example
 
 ```typescript
-import { Agent, Task, PraisonAIAgents } from 'praisonai';
+import { Agent, Task, AgentTeam } from 'praisonai';
 
 async function main() {
     // Create agents first
@@ -160,15 +160,15 @@ The blog post should:
         dependencies: [createRecipesTask]  // This task depends on the recipes being created first
     });
 
-    // Run the tasks
-    const praisonAI = new PraisonAIAgents({
+    // Run the tasks with AgentTeam
+    const team = new AgentTeam({
         tasks: [createRecipesTask, writeBlogTask],
         verbose: true
     });
 
     try {
         console.log('Starting task-based example...');
-        const results = await praisonAI.start();
+        const results = await team.start();
         console.log('\nFinal Results:', results);
     } catch (error) {
         console.error('Error:', error);

package/dist/agent/code.d.ts

@@ -0,0 +1,155 @@
+/**
+ * CodeAgent - Code generation, execution, review, and refactoring
+ *
+ * Python parity with praisonaiagents/agent/code_agent.py
+ */
+/**
+ * Configuration for CodeAgent.
+ * Python parity with CodeConfig dataclass.
+ */
+export interface CodeConfig {
+    /** Enable sandboxed execution (default: true for safety) */
+    sandbox?: boolean;
+    /** Execution timeout in seconds */
+    timeout?: number;
+    /** List of allowed programming languages */
+    allowedLanguages?: string[];
+    /** Maximum output length in characters */
+    maxOutputLength?: number;
+    /** Working directory for code execution */
+    workingDirectory?: string;
+    /** Environment variables for execution */
+    environment?: Record<string, string>;
+}
+/**
+ * Result of code execution.
+ */
+export interface CodeExecutionResult {
+    /** Whether execution was successful */
+    success: boolean;
+    /** Standard output from execution */
+    output: string;
+    /** Error message if execution failed */
+    error?: string;
+    /** Exit code from execution */
+    exitCode: number;
+    /** Execution time in seconds */
+    executionTime: number;
+}
+/**
+ * Configuration for creating a CodeAgent.
+ */
+export interface CodeAgentConfig {
+    /** Agent name */
+    name?: string;
+    /** LLM model (default: gpt-4o-mini) */
+    llm?: string;
+    /** Code configuration (bool, object, or CodeConfig) */
+    code?: boolean | CodeConfig;
+    /** System instructions */
+    instructions?: string;
+    /** Enable verbose output */
+    verbose?: boolean;
+}
+/**
+ * Agent for code generation, execution, review, and refactoring.
+ *
+ * This agent provides capabilities for:
+ * - Generating code from natural language descriptions
+ * - Executing code in a sandboxed environment
+ * - Reviewing code for issues and improvements
+ * - Refactoring and fixing code
+ * - Explaining code functionality
+ *
+ * @example
+ * ```typescript
+ * import { CodeAgent } from 'praisonai';
+ *
+ * const agent = new CodeAgent({ name: 'Coder' });
+ *
+ * // Generate code
+ * const code = await agent.generate('Write a function to calculate fibonacci');
+ *
+ * // Execute code
+ * const result = await agent.execute("console.log('Hello, World!')");
+ *
+ * // Review code
+ * const review = await agent.review(code);
+ * ```
+ */
+export declare class CodeAgent {
+    readonly name: string;
+    private readonly llm;
+    private readonly instructions?;
+    private readonly verbose;
+    private readonly codeConfig;
+    private readonly agent;
+    constructor(config: CodeAgentConfig);
+    private buildSystemPrompt;
+    /**
+     * Generate code from natural language description.
+     *
+     * @param prompt - Natural language description of desired code
+     * @param language - Target programming language (default: python)
+     * @returns Generated code as string
+     */
+    generate(prompt: string, language?: string): Promise<string>;
+    /**
+     * Alias for generate() method.
+     */
+    generateCode(prompt: string, language?: string): Promise<string>;
+    /**
+     * Execute code in a sandboxed environment.
+     *
+     * @param code - Code to execute
+     * @param language - Programming language (default: python)
+     * @returns Execution result
+     */
+    execute(code: string, language?: string): Promise<CodeExecutionResult>;
+    /**
+     * Alias for execute() method.
+     */
+    executeCode(code: string, language?: string): Promise<CodeExecutionResult>;
+    /**
+     * Review code for issues and improvements.
+     *
+     * @param code - Code to review
+     * @param language - Programming language
+     * @returns Review feedback
+     */
+    review(code: string, language?: string): Promise<string>;
+    /**
+     * Alias for review() method.
+     */
+    reviewCode(code: string, language?: string): Promise<string>;
+    /**
+     * Refactor code for better quality.
+     *
+     * @param code - Code to refactor
+     * @param instructions - Specific refactoring instructions
+     * @returns Refactored code
+     */
+    refactor(code: string, instructions?: string): Promise<string>;
+    /**
+     * Fix bugs in code.
+     *
+     * @param code - Code with bugs
+     * @param errorMessage - Optional error message to help identify the bug
+     * @returns Fixed code
+     */
+    fix(code: string, errorMessage?: string): Promise<string>;
+    /**
+     * Explain code functionality.
+     *
+     * @param code - Code to explain
+     * @returns Explanation of the code
+     */
+    explain(code: string): Promise<string>;
+}
+/**
+ * Create a CodeAgent instance.
+ *
+ * @param config - CodeAgent configuration
+ * @returns CodeAgent instance
+ */
+export declare function createCodeAgent(config: CodeAgentConfig): CodeAgent;

package/dist/agent/code.js

@@ -0,0 +1,246 @@
+"use strict";
+/**
+ * CodeAgent - Code generation, execution, review, and refactoring
+ *
+ * Python parity with praisonaiagents/agent/code_agent.py
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CodeAgent = void 0;
+exports.createCodeAgent = createCodeAgent;
+const simple_1 = require("./simple");
+// ============================================================================
+// Default Configuration
+// ============================================================================
+const DEFAULT_CODE_CONFIG = {
+    sandbox: true,
+    timeout: 30,
+    allowedLanguages: ['python'],
+    maxOutputLength: 10000,
+    workingDirectory: process.cwd(),
+    environment: {},
+};
+// ============================================================================
+// CodeAgent Class
+// ============================================================================
+/**
+ * Agent for code generation, execution, review, and refactoring.
+ *
+ * This agent provides capabilities for:
+ * - Generating code from natural language descriptions
+ * - Executing code in a sandboxed environment
+ * - Reviewing code for issues and improvements
+ * - Refactoring and fixing code
+ * - Explaining code functionality
+ *
+ * @example
+ * ```typescript
+ * import { CodeAgent } from 'praisonai';
+ *
+ * const agent = new CodeAgent({ name: 'Coder' });
+ *
+ * // Generate code
+ * const code = await agent.generate('Write a function to calculate fibonacci');
+ *
+ * // Execute code
+ * const result = await agent.execute("console.log('Hello, World!')");
+ *
+ * // Review code
+ * const review = await agent.review(code);
+ * ```
+ */
+class CodeAgent {
+    constructor(config) {
+        this.name = config.name || 'CodeAgent';
+        this.llm = config.llm || process.env.OPENAI_MODEL_NAME || 'gpt-4o-mini';
+        this.instructions = config.instructions;
+        this.verbose = config.verbose ?? true;
+        // Resolve code configuration
+        if (config.code === undefined || config.code === true) {
+            this.codeConfig = { ...DEFAULT_CODE_CONFIG };
+        }
+        else if (typeof config.code === 'object') {
+            this.codeConfig = { ...DEFAULT_CODE_CONFIG, ...config.code };
+        }
+        else {
+            this.codeConfig = { ...DEFAULT_CODE_CONFIG };
+        }
+        // Create underlying agent for LLM calls
+        this.agent = new simple_1.Agent({
+            name: this.name,
+            instructions: this.buildSystemPrompt(),
+            llm: this.llm,
+            verbose: this.verbose,
+        });
+    }
+    buildSystemPrompt() {
+        let prompt = `You are an expert programmer and code assistant.
+You can generate, review, refactor, fix, and explain code.
+Follow best practices and coding standards.
+Write clean, well-documented, production-ready code.`;
+        if (this.instructions) {
+            prompt += `\n\nAdditional instructions: ${this.instructions}`;
+        }
+        return prompt;
+    }
+    // =========================================================================
+    // Code Generation Methods
+    // =========================================================================
+    /**
+     * Generate code from natural language description.
+     *
+     * @param prompt - Natural language description of desired code
+     * @param language - Target programming language (default: python)
+     * @returns Generated code as string
+     */
+    async generate(prompt, language = 'python') {
+        const systemPrompt = `You are an expert ${language} programmer.
+Generate clean, well-documented, production-ready code.
+Only output the code, no explanations unless asked.
+Follow best practices and coding standards.`;
+        const response = await this.agent.chat(`Generate ${language} code for: ${prompt}`);
+        return response;
+    }
+    /**
+     * Alias for generate() method.
+     */
+    async generateCode(prompt, language = 'python') {
+        return this.generate(prompt, language);
+    }
+    // =========================================================================
+    // Code Execution Methods
+    // =========================================================================
+    /**
+     * Execute code in a sandboxed environment.
+     *
+     * @param code - Code to execute
+     * @param language - Programming language (default: python)
+     * @returns Execution result
+     */
+    async execute(code, language = 'python') {
+        // Check if language is allowed
+        if (!this.codeConfig.allowedLanguages.includes(language)) {
+            return {
+                success: false,
+                output: '',
+                error: `Language '${language}' is not allowed. Allowed: ${this.codeConfig.allowedLanguages.join(', ')}`,
+                exitCode: 1,
+                executionTime: 0,
+            };
+        }
+        const startTime = Date.now();
+        try {
+            // For safety, we use the sandbox executor if available
+            // This is a simplified implementation - full sandbox would use child_process
+            if (language === 'javascript' || language === 'typescript') {
+                // Use eval for simple JS (NOT SAFE FOR PRODUCTION - use sandbox)
+                if (!this.codeConfig.sandbox) {
+                    const result = eval(code);
+                    return {
+                        success: true,
+                        output: String(result ?? ''),
+                        exitCode: 0,
+                        executionTime: (Date.now() - startTime) / 1000,
+                    };
+                }
+            }
+            // For other languages or sandboxed execution, return placeholder
+            return {
+                success: false,
+                output: '',
+                error: `Sandboxed execution for '${language}' requires additional setup. Use sandbox executor.`,
+                exitCode: 1,
+                executionTime: (Date.now() - startTime) / 1000,
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                output: '',
+                error: error instanceof Error ? error.message : String(error),
+                exitCode: 1,
+                executionTime: (Date.now() - startTime) / 1000,
+            };
+        }
+    }
+    /**
+     * Alias for execute() method.
+     */
+    async executeCode(code, language = 'python') {
+        return this.execute(code, language);
+    }
+    // =========================================================================
+    // Code Review Methods
+    // =========================================================================
+    /**
+     * Review code for issues and improvements.
+     *
+     * @param code - Code to review
+     * @param language - Programming language
+     * @returns Review feedback
+     */
+    async review(code, language) {
+        const langHint = language ? ` (${language})` : '';
+        const response = await this.agent.chat(`Review the following code${langHint} for issues, bugs, and improvements:\n\n\`\`\`\n${code}\n\`\`\``);
+        return response;
+    }
+    /**
+     * Alias for review() method.
+     */
+    async reviewCode(code, language) {
+        return this.review(code, language);
+    }
+    // =========================================================================
+    // Code Refactoring Methods
+    // =========================================================================
+    /**
+     * Refactor code for better quality.
+     *
+     * @param code - Code to refactor
+     * @param instructions - Specific refactoring instructions
+     * @returns Refactored code
+     */
+    async refactor(code, instructions) {
+        const prompt = instructions
+            ? `Refactor the following code according to these instructions: ${instructions}\n\n\`\`\`\n${code}\n\`\`\``
+            : `Refactor the following code for better readability, performance, and maintainability:\n\n\`\`\`\n${code}\n\`\`\``;
+        const response = await this.agent.chat(prompt);
+        return response;
+    }
+    /**
+     * Fix bugs in code.
+     *
+     * @param code - Code with bugs
+     * @param errorMessage - Optional error message to help identify the bug
+     * @returns Fixed code
+     */
+    async fix(code, errorMessage) {
+        const prompt = errorMessage
+            ? `Fix the bug in the following code. Error: ${errorMessage}\n\n\`\`\`\n${code}\n\`\`\``
+            : `Fix any bugs in the following code:\n\n\`\`\`\n${code}\n\`\`\``;
+        const response = await this.agent.chat(prompt);
+        return response;
+    }
+    /**
+     * Explain code functionality.
+     *
+     * @param code - Code to explain
+     * @returns Explanation of the code
+     */
+    async explain(code) {
+        const response = await this.agent.chat(`Explain what the following code does in detail:\n\n\`\`\`\n${code}\n\`\`\``);
+        return response;
+    }
+}
+exports.CodeAgent = CodeAgent;
+// ============================================================================
+// Factory Function
+// ============================================================================
+/**
+ * Create a CodeAgent instance.
+ *
+ * @param config - CodeAgent configuration
+ * @returns CodeAgent instance
+ */
+function createCodeAgent(config) {
+    return new CodeAgent(config);
+}

package/dist/agent/embedding.d.ts

@@ -0,0 +1,118 @@
+/**
+ * EmbeddingAgent - Text embedding generation agent
+ *
+ * Python parity with praisonaiagents/agent/embedding_agent.py
+ * Generates embeddings for text using embedding models.
+ */
+/**
+ * Configuration for Embedding settings.
+ */
+export interface EmbeddingConfig {
+    /** Embedding model to use */
+    model?: string;
+    /** Dimensions for the embedding */
+    dimensions?: number;
+    /** Batch size for processing multiple texts */
+    batchSize?: number;
+    /** Timeout in seconds */
+    timeout?: number;
+}
+/**
+ * Result of embedding generation.
+ */
+export interface EmbeddingResult {
+    /** The embedding vector */
+    embedding: number[];
+    /** Model used */
+    model: string;
+    /** Token usage */
+    usage?: {
+        promptTokens: number;
+        totalTokens: number;
+    };
+}
+/**
+ * Result of batch embedding generation.
+ */
+export interface BatchEmbeddingResult {
+    /** Array of embeddings */
+    embeddings: number[][];
+    /** Model used */
+    model: string;
+    /** Total token usage */
+    usage?: {
+        promptTokens: number;
+        totalTokens: number;
+    };
+}
+/**
+ * Configuration for creating an EmbeddingAgent.
+ */
+export interface EmbeddingAgentConfig {
+    /** Agent name */
+    name?: string;
+    /** Embedding model */
+    llm?: string;
+    /** Alias for llm */
+    model?: string;
+    /** Embedding configuration */
+    embedding?: boolean | EmbeddingConfig;
+    /** Enable verbose output */
+    verbose?: boolean;
+}
+/**
+ * Agent for generating text embeddings.
+ *
+ * @example
+ * ```typescript
+ * import { EmbeddingAgent } from 'praisonai';
+ *
+ * const agent = new EmbeddingAgent({});
+ *
+ * // Generate embedding for text
+ * const result = await agent.embed('Hello, world!');
+ * console.log(result.embedding.length); // 1536
+ *
+ * // Generate embeddings for multiple texts
+ * const results = await agent.embedMany(['Hello', 'World']);
+ * ```
+ */
+export declare class EmbeddingAgent {
+    static readonly DEFAULT_MODEL = "text-embedding-3-small";
+    readonly name: string;
+    private readonly model;
+    private readonly verbose;
+    private readonly embeddingConfig;
+    constructor(config: EmbeddingAgentConfig);
+    private log;
+    /**
+     * Generate embedding for a single text.
+     *
+     * @param text - Text to embed
+     * @returns EmbeddingResult with the embedding vector
+     */
+    embed(text: string): Promise<EmbeddingResult>;
+    /**
+     * Generate embeddings for multiple texts.
+     *
+     * @param texts - Array of texts to embed
+     * @returns BatchEmbeddingResult with all embeddings
+     */
+    embedMany(texts: string[]): Promise<BatchEmbeddingResult>;
+    /**
+     * Calculate cosine similarity between two embeddings.
+     */
+    cosineSimilarity(a: number[], b: number[]): number;
+    /**
+     * Find the most similar text from a list.
+     */
+    findMostSimilar(query: string, candidates: string[]): Promise<{
+        text: string;
+        similarity: number;
+        index: number;
+    }>;
+}
+/**
+ * Create an EmbeddingAgent instance.
+ */
+export declare function createEmbeddingAgent(config: EmbeddingAgentConfig): EmbeddingAgent;