@ugm/desiagent 0.1.36 → 0.2.0

package/README.md

@@ -1,278 +1,346 @@
 # desiAgent
 
-A library-first async agent system for building autonomous workflows with TypeScript.
+A library-first async agent system for building autonomous workflows with TypeScript. Give it a goal in plain English and it decomposes it into a DAG of tasks, executes them with built-in tools, and streams results back — all in a few lines of code.
 
 ## Features
 
-- **Multiple LLM Providers**: OpenAI, OpenRouter, and Ollama support
-- **Autonomous Execution**: Let agents execute objectives autonomously
-- **DAG Workflows**: Decompose complex objectives into directed acyclic graphs
-- **Built-in Tools**: Web scraping, file operations, bash commands, and more
-- **Event Streaming**: Track execution progress in real-time
-- **SQLite Storage**: Persistent storage using bun:sqlite
+- **Goal → DAG → Execution** — Describe what you want; desiAgent plans a directed acyclic graph (DAG) of sub-tasks and executes them autonomously.
+- **Multiple LLM Providers** — OpenAI, OpenRouter, and Ollama (local) out of the box.
+- **Built-in Tools** — Web scraping, file I/O, bash commands, email (SMTP/IMAP), PDF parsing, and more.
+- **Event Streaming** — `for await` over execution events to track progress in real-time.
+- **Clarification Flow** — If the goal is ambiguous the agent asks for clarification before proceeding.
+- **In-Memory or Persistent Storage** — Use `:memory:` for quick experiments or a SQLite file for production.
+- **Cron Scheduling** — Schedule DAGs to run on a cron expression with timezone support.
+- **Artifacts** — Tools can write output files (reports, code, images) that are automatically stored and retrievable.
+- **Cost Tracking** — Token usage and USD cost are recorded per execution step.
+- **Experiments API** — Compare models and temperatures on the same goal in one call.
 
 ## Installation
 
 ### Prerequisites
 
-- [Bun](https://bun.sh) 1.3.5 or higher
+- [Bun](https://bun.sh) ≥ 1.3.5
 
 ### Install
 
 ```bash
-bun add desiagent
+bun add @ugm/desiagent
 ```
 
-Or with npm/pnpm:
+Or with npm / pnpm:
 
 ```bash
-npm install desiagent
+npm install @ugm/desiagent
 # or
-pnpm add desiagent
+pnpm add @ugm/desiagent
 ```
 
-## Environment Variables
+### Environment Variables
 
-Create a `.env` file in your project root with the appropriate API keys for your chosen provider:
-
-### OpenAI
+Create a `.env` file with your provider's API key:
 
 ```bash
-OPENAI_API_KEY=sk-...
-```
+# OpenRouter (recommended — access to many models via one key)
+OPENROUTER_API_KEY=sk-or-...
 
-### OpenRouter
+# OpenAI
+OPENAI_API_KEY=sk-...
 
-```bash
-OPENROUTER_API_KEY=sk-or-...
+# Ollama (local, no key needed)
+OLLAMA_BASE_URL=http://localhost:11434   # optional, this is the default
 ```
 
-### Ollama (Local)
+## Examples
 
-```bash
-OLLAMA_BASE_URL=http://localhost:11434  # Optional, this is the default
-```
+> All examples below use OpenRouter and an in-memory database (`:memory:`) so you can run them without any local SQLite file.
 
-## Quick Start
+### 1. Goal → Execute in One Call
 
-### Basic Example (OpenAI)
+The fastest way to go from idea to result. `createAndExecuteFromGoal` plans the DAG **and** executes it in a single call.
 
 ```typescript
-import { setupDesiAgent } from 'desiagent';
+import { setupDesiAgent } from '@ugm/desiagent';
 
 const client = await setupDesiAgent({
-  llmProvider: 'openai',
-  openaiApiKey: process.env.OPENAI_API_KEY,
-  modelName: 'gpt-4o',
+  llmProvider: 'openrouter',
+  openrouterApiKey: process.env.OPENROUTER_API_KEY,
+  modelName: 'google/gemini-2.5-flash-lite-preview-09-2025',
+  databasePath: ':memory:',
+  skipGenerationStats: true,
 });
 
-// Create and execute a DAG
-const execution = await client.dags.createAndExecute('Summarize the latest tech news');
-console.log(`Execution: ${execution.id}, Status: ${execution.status}`);
+const result = await client.dags.createAndExecuteFromGoal({
+  goalText: 'Research the top 5 trends in AI agents for 2025 and write a concise briefing document to ai-trends.md',
+  agentName: 'DecomposerV8',
+  temperature: 0.7,
+});
+
+if (result.status === 'clarification_required') {
+  console.log('Agent needs more info:', result.clarificationQuery);
+} else {
+  console.log('Execution started:', result.executionId);
+
+  // Stream events until completion
+  for await (const event of client.executions.streamEvents(result.executionId)) {
+    console.log(event.type, event.data);
+  }
+
+  // Retrieve final result
+  const details = await client.executions.getWithSubSteps(result.executionId);
+  console.log('Final result:\n', details.finalResult);
+}
 
 await client.shutdown();
 ```
 
-### Using OpenRouter
+### 2. Plan First, Execute Later
+
+Separate planning from execution so you can inspect or modify the DAG before running it.
 
 ```typescript
-import { setupDesiAgent } from 'desiagent';
+import { setupDesiAgent } from '@ugm/desiagent';
 
 const client = await setupDesiAgent({
   llmProvider: 'openrouter',
   openrouterApiKey: process.env.OPENROUTER_API_KEY,
-  modelName: 'google/gemini-2.5-flash-lite-preview-06-2025',
+  modelName: 'google/gemini-2.5-flash-lite-preview-09-2025',
+  databasePath: ':memory:',
+  skipGenerationStats: true,
+});
+
+// Step 1 — Plan
+const plan = await client.dags.createFromGoal({
+  goalText: 'Create a tutorial on processing driftwood into handicrafts — cover cleaning, tools, finishes — and write it to driftwood.md',
+  agentName: 'DecomposerV8',
+  temperature: 0.7,
 });
 
-const execution = await client.dags.createAndExecute('Research best practices for API design');
+if (plan.status !== 'success') {
+  console.log('Planning issue:', plan.status);
+  await client.shutdown();
+  process.exit(1);
+}
+
+console.log('DAG created:', plan.dagId);
+
+// Step 2 — Execute
+const execution = await client.dags.execute(plan.dagId);
+console.log('Execution ID:', execution.id);
+
+for await (const event of client.executions.streamEvents(execution.id)) {
+  console.log(event.type, event.data);
+}
+
+const details = await client.executions.getWithSubSteps(execution.id);
+console.log('Final result:\n', details.finalResult);
+
 await client.shutdown();
 ```
 
-### Using Ollama (Local LLM)
+### 3. List Agents and Tools
+
+Explore what's available in the system.
 
 ```typescript
-import { setupDesiAgent } from 'desiagent';
+import { setupDesiAgent } from '@ugm/desiagent';
 
 const client = await setupDesiAgent({
-  llmProvider: 'ollama',
-  ollamaBaseUrl: process.env.OLLAMA_BASE_URL || 'http://localhost:11434',
-  modelName: 'llama3.2',
+  llmProvider: 'openrouter',
+  openrouterApiKey: process.env.OPENROUTER_API_KEY,
+  modelName: 'openai/gpt-4o',
+  databasePath: ':memory:',
+  skipGenerationStats: true,
+  logLevel: 'warn',
 });
 
-const execution = await client.dags.createAndExecute('Explain quantum computing in simple terms');
+// List seeded agents
+const agents = await client.agents.list();
+for (const a of agents) {
+  console.log(`${a.name} (${a.provider}/${a.model}) — ${a.description}`);
+}
+
+// List available tools
+const tools = await client.tools.list();
+for (const t of tools) {
+  console.log(t.function.name);
+}
+
 await client.shutdown();
 ```
 
-## Configuration Options
+### 4. Handle Clarifications
 
-```typescript
-interface DesiAgentConfig {
-  // LLM Provider (required)
-  llmProvider: 'openai' | 'openrouter' | 'ollama';
-  modelName: string;
+When the agent decides the goal is ambiguous, it returns a clarification query instead of creating a DAG.
 
-  // Provider-specific API keys
-  openaiApiKey?: string;      // Required for 'openai' provider
-  openrouterApiKey?: string;  // Required for 'openrouter' provider
-  ollamaBaseUrl?: string;     // Optional for 'ollama', defaults to http://localhost:11434
+```typescript
+import { setupDesiAgent } from '@ugm/desiagent';
 
-  // Database (optional)
-  databasePath?: string;      // Default: ~/.desiAgent/data/agent.db
+const client = await setupDesiAgent({
+  llmProvider: 'openrouter',
+  openrouterApiKey: process.env.OPENROUTER_API_KEY,
+  modelName: 'google/gemini-2.5-flash-lite-preview-09-2025',
+  databasePath: ':memory:',
+  skipGenerationStats: true,
+});
 
-  // Agent definitions (optional)
-  agentDefinitionsPath?: string;  // Default: ~/.desiAgent/agents
+const plan = await client.dags.createFromGoal({
+  goalText: 'Build the app',
+  agentName: 'DecomposerV8',
+});
 
-  // Logging (optional)
-  logLevel?: 'debug' | 'info' | 'warn' | 'error';  // Default: 'info'
+if (plan.status === 'clarification_required') {
+  console.log('Agent asks:', plan.clarificationQuery);
 
-  // Callbacks (optional)
-  onExecutionStart?: (executionId: string) => void;
-  onExecutionEnd?: (executionId: string, result: any) => void;
+  // Provide the answer and retry
+  const resumed = await client.dags.resumeFromClarification(
+    plan.dagId,
+    'A Pomodoro timer web app using HTML, CSS, and vanilla JS',
+  );
+  console.log('Resumed status:', resumed.status);
 }
+
+await client.shutdown();
 ```
 
-## API Reference
+### 5. Custom Inference (No DAG)
 
-### Agents
+Use a named agent directly for a single LLM call — useful for summarisation, translation, or any one-shot task.
 
 ```typescript
-// Create a new agent
-const agent = await client.agents.create(name, version, prompt, params?);
-
-// List all agents
-const agents = await client.agents.list(filter?);
-
-// Get a specific agent
-const agent = await client.agents.get(id);
+import { setupDesiAgent } from '@ugm/desiagent';
 
-// Update an agent
-await client.agents.update(id, updates);
+const client = await setupDesiAgent({
+  llmProvider: 'openrouter',
+  openrouterApiKey: process.env.OPENROUTER_API_KEY,
+  modelName: 'openai/gpt-4o',
+  databasePath: ':memory:',
+  skipGenerationStats: true,
+});
 
-// Activate/Delete an agent
-await client.agents.activate(id);
-await client.agents.delete(id);
+// Resolve an agent by name and call it directly
+const agent = await client.agents.resolve('Summarizer');
+// ... or use the lower-level inference API in scripts/infer.ts
 
-// Resolve agent by name
-const agent = await client.agents.resolve(name);
+await client.shutdown();
 ```
 
-### DAGs (Directed Acyclic Graphs)
+### Running the Bundled Examples
 
-```typescript
-// Create a DAG from an objective
-const dag = await client.dags.create(objective, params?);
-
-// Create and immediately execute
-const execution = await client.dags.createAndExecute(objective, params?);
+The [`examples/`](./examples/) directory contains runnable scripts:
 
-// Execute an existing DAG
-const execution = await client.dags.execute(dagId, params?);
+```bash
+# Execute a goal from a file
+bun run examples/execute-goal.ts -f examples/goals/pomodoro-timer.txt
 
-// List DAGs
-const dags = await client.dags.list(filter?);
-const scheduled = await client.dags.listScheduled();
+# In-memory database smoke test
+bun run examples/init_6_memory_db.ts
 
-// Get/Update/Delete
-const dag = await client.dags.get(id);
-await client.dags.update(id, updates);
-await client.dags.delete(id);
+# List all agents
+bun run examples/list-agents.ts
 
-// Resume a paused execution
-await client.dags.resume(executionId);
+# List all tools
+bun run examples/list-tools.ts --names
 ```
 
-### Executions
+## Configuration Reference
 
 ```typescript
-// List executions
-const executions = await client.executions.list(filter?);
+interface DesiAgentConfig {
+  llmProvider: 'openai' | 'openrouter' | 'ollama';
+  modelName: string;
 
-// Get execution details
-const execution = await client.executions.get(id);
-const subSteps = await client.executions.getSubSteps(id);
+  // Provider keys
+  openaiApiKey?: string;
+  openrouterApiKey?: string;
+  ollamaBaseUrl?: string;       // default: http://localhost:11434
 
-// Delete an execution
-await client.executions.delete(id);
+  // Storage
+  databasePath?: string;        // default: ~/.desiAgent/data/agent.db
+                                 // use ':memory:' for ephemeral experiments
+  artifactsDir?: string;        // default: sibling of database file
 
-// Stream execution events (AsyncIterable)
-for await (const event of client.executions.streamEvents(id)) {
-  console.log(event.type, event.data);
-}
-```
+  // Agent definitions
+  agentDefinitionsPath?: string; // default: ~/.desiAgent/agents
 
-## CLI
+  // Logging
+  logLevel?: 'debug' | 'info' | 'warn' | 'error' | 'silent';
 
-The `desi` CLI ships with the standalone `desiCLI` package. Command groups support default subcommands for convenience:
+  // Lifecycle hooks
+  onExecutionStart?: (executionId: string) => void;
+  onExecutionEnd?: (executionId: string, result: Record<string, any>) => void;
 
-```bash
-# Artifacts
-desi artifacts              # Same as: desi artifacts list
-desi artifacts report.json  # Same as: desi artifacts get report.json
+  // Feature flags
+  autoStartScheduler?: boolean;  // default: true
+  enableToolValidation?: boolean; // default: true
+}
+```
 
-# Agents
-desi agents                 # Same as: desi agents list
+## Contributing
 
-# Results
-desi results exec_123       # Same as: desi results view exec_123
-```
+We welcome contributions of all kinds — bug fixes, new tools, documentation improvements, and feature ideas.
 
-## Examples
+### Getting Started
 
-### News Bulletin Generator
+1. **Fork & clone** the repository.
 
-See [`examples/news-bulletin.ts`](./examples/news-bulletin.ts) for a complete example that creates a news bulletin with multiple categories.
+   ```bash
+   git clone https://github.com/<your-username>/desiAgent.git
+   cd desiAgent
+   ```
 
-```bash
-# Run with Ollama
-bun run examples/news-bulletin.ts
+2. **Install dependencies** (requires Bun ≥ 1.3.5).
 
-# Run with OpenRouter
-OPENROUTER_API_KEY=your-key bun run examples/news-bulletin-openrouter.ts
-```
+   ```bash
+   bun install
+   ```
 
-## Manual Abort Tests
+3. **Create a branch** for your change.
 
-Use these scripts to manually verify that abort signals cancel in-flight LLM calls.
+   ```bash
+   git checkout -b feat/my-awesome-feature
+   ```
 
-```bash
-# OpenRouter abort test
-OPENROUTER_API_KEY=sk-or-... ABORT_AFTER_MS=2000 bun run test:abort:openrouter
+4. **Make your changes**, then verify:
 
-# Ollama abort test
-OLLAMA_BASE_URL=http://localhost:11434 OLLAMA_MODEL=mistral ABORT_AFTER_MS=2000 bun run test:abort:ollama
-```
+   ```bash
+   bun run type-check   # TypeScript must compile cleanly
+   bun test             # All tests must pass
+   ```
 
-## Development
+### Code Guidelines
 
-### Setup
+- **TypeScript only** — no plain JS files.
+- **Follow existing patterns** — look at neighbouring files before writing new code. Match naming conventions, imports, and error handling style.
+- **Keep PRs focused** — one logical change per pull request. Small, reviewable diffs are merged faster.
+- **Write tests** — if you add a feature or fix a bug, add or update a test in the corresponding `*.test.ts` file. Run `bun test` to verify.
+- **No secrets** — never commit API keys, tokens, or credentials. Use environment variables and `.env` files (already in `.gitignore`).
 
-```bash
-# Install dependencies
-pnpm install
+### Commit Messages
 
-# Run in development mode
-bun run dev
+Follow the [Conventional Commits](https://www.conventionalcommits.org/) format:
+
+```
+feat: add PDF attachment support to inference
+fix: handle empty goal text in DAG creation
+docs: update README examples
+chore: bump drizzle-orm to 0.46
 ```
 
-### Commands
+### Pull Request Process
 
-```bash
-# Build
-pnpm build
+1. Ensure your branch is up to date with `main`.
+2. Open a PR against `main` with a clear title and description of **what** and **why**.
+3. Link any related issues (e.g., `Closes #42`).
+4. A maintainer will review your PR. Address feedback promptly — we aim to merge within a few days.
 
-# Type check
-pnpm type-check
+### Reporting Issues
 
-# Run tests
-pnpm test
-pnpm test:watch
-pnpm test:coverage
+- Use [GitHub Issues](https://github.com/ugmurthy/desiAgent/issues) to report bugs or request features.
+- Include steps to reproduce, expected vs. actual behaviour, and your environment (OS, Bun version, provider used).
 
-# Database commands
-pnpm db:generate
-pnpm db:push
-pnpm db:studio
-```
+### Code of Conduct
+
+Be respectful and constructive. We follow the [Contributor Covenant](https://www.contributor-covenant.org/) code of conduct.
 
 ## License
 

package/dist/core/execution/dagExecutor.d.ts

@@ -45,7 +45,10 @@ export interface DAGExecutorConfig {
     db: DrizzleDB;
     llmProvider: LLMProvider;
     toolRegistry: ToolRegistry;
-    artifactsDir?: string;
+    artifactsDir: string;
+    apiKey?: string;
+    ollamaBaseUrl?: string;
+    skipGenerationStats?: boolean;
 }
 /**
  * Execution configuration for performance tuning
@@ -85,6 +88,9 @@ export declare class DAGExecutor {
     private llmProvider;
     private toolRegistry;
     private artifactsDir;
+    private apiKey?;
+    private ollamaBaseUrl?;
+    private skipGenerationStats?;
     private logger;
     constructor(config: DAGExecutorConfig);
     private extractUrls;

package/dist/core/execution/dagExecutor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"dagExecutor.d.ts","sourceRoot":"","sources":["../../../src/core/execution/dagExecutor.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAEpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAQzD,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,GAAG,WAAW,CAAC;IAClC,cAAc,EAAE;QACd,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC9B,CAAC;IACF,eAAe,EAAE,MAAM,CAAC;IACxB,YAAY,EAAE,MAAM,EAAE,CAAC;CACxB;AAED,MAAM,WAAW,aAAa;IAC5B,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,EAAE;QACN,OAAO,EAAE,MAAM,CAAC;QAChB,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB,CAAC;IACF,QAAQ,EAAE,KAAK,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,cAAc,EAAE,MAAM,CAAC;KACxB,CAAC,CAAC;IACH,SAAS,EAAE,OAAO,EAAE,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE;QACV,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,MAAM,EAAE,CAAC;QACf,kBAAkB,EAAE,MAAM,EAAE,CAAC;KAC9B,CAAC;IACF,oBAAoB,EAAE,OAAO,CAAC;IAC9B,mBAAmB,CAAC,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,SAAS,CAAC;IACd,WAAW,EAAE,WAAW,CAAC;IACzB,YAAY,EAAE,YAAY,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B;;;OAGG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB;;;OAGG;IACH,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB;;OAEG;IACH,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAYD,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,GAAG,CAAC;IACb,KAAK,CAAC,EAAE;QACN,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACvC;AAMD,qBAAa,WAAW;IACtB,OAAO,CAAC,EAAE,CAAY;IACtB,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,MAAM,CAAe;gBAEjB,MAAM,EAAE,iBAAiB;IAWrC,OAAO,CAAC,WAAW;IAYnB,OAAO,CAAC,kBAAkB;IAgB1B,OAAO,CAAC,oBAAoB;IAkC5B,OAAO,CAAC,mBAAmB;IAe3B,OAAO,CAAC,qBAAqB;IA6B7B,OAAO,CAAC,mBAAmB;IA+B3B,OAAO,CAAC,uBAAuB;IAqB/B,OAAO,CAAC,gBAAgB;IA0BxB,OAAO,CAAC,yBAAyB;IA0BjC;;;OAGG;YACW,cAAc;IAsC5B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAMpB,OAAO,CACX,GAAG,EAAE,aAAa,EAClB,WAAW,EAAE,MAAM,EACnB,KAAK,CAAC,EAAE,MAAM,EACd,eAAe,CAAC,EAAE,MAAM,EACxB,MAAM,GAAE,eAAoB,GAC3B,OAAO,CAAC,MAAM,CAAC;YA2aJ,gBAAgB;IAuB9B,OAAO,CAAC,qBAAqB;YA6Bf,UAAU;YAmEV,QAAQ;IAKtB,OAAO,CAAC,cAAc;IAkBtB,OAAO,CAAC,aAAa;CAatB"}
+{"version":3,"file":"dagExecutor.d.ts","sourceRoot":"","sources":["../../../src/core/execution/dagExecutor.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAEpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAQzD,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,GAAG,WAAW,CAAC;IAClC,cAAc,EAAE;QACd,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC9B,CAAC;IACF,eAAe,EAAE,MAAM,CAAC;IACxB,YAAY,EAAE,MAAM,EAAE,CAAC;CACxB;AAED,MAAM,WAAW,aAAa;IAC5B,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,EAAE;QACN,OAAO,EAAE,MAAM,CAAC;QAChB,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB,CAAC;IACF,QAAQ,EAAE,KAAK,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,cAAc,EAAE,MAAM,CAAC;KACxB,CAAC,CAAC;IACH,SAAS,EAAE,OAAO,EAAE,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE;QACV,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,MAAM,EAAE,CAAC;QACf,kBAAkB,EAAE,MAAM,EAAE,CAAC;KAC9B,CAAC;IACF,oBAAoB,EAAE,OAAO,CAAC;IAC9B,mBAAmB,CAAC,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,SAAS,CAAC;IACd,WAAW,EAAE,WAAW,CAAC;IACzB,YAAY,EAAE,YAAY,CAAC;IAC3B,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B;;;OAGG;IACH,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB;;;OAGG;IACH,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB;;OAEG;IACH,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAYD,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,GAAG,CAAC;IACb,KAAK,CAAC,EAAE;QACN,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACvC;AAMD,qBAAa,WAAW;IACtB,OAAO,CAAC,EAAE,CAAY;IACtB,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,aAAa,CAAC,CAAS;IAC/B,OAAO,CAAC,mBAAmB,CAAC,CAAU;IACtC,OAAO,CAAC,MAAM,CAAe;gBAEjB,MAAM,EAAE,iBAAiB;IAcrC,OAAO,CAAC,WAAW;IAYnB,OAAO,CAAC,kBAAkB;IAgB1B,OAAO,CAAC,oBAAoB;IAkC5B,OAAO,CAAC,mBAAmB;IAe3B,OAAO,CAAC,qBAAqB;IA6B7B,OAAO,CAAC,mBAAmB;IA+B3B,OAAO,CAAC,uBAAuB;IAqB/B,OAAO,CAAC,gBAAgB;IA0BxB,OAAO,CAAC,yBAAyB;IA0BjC;;;OAGG;YACW,cAAc;IAsC5B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAMpB,OAAO,CACX,GAAG,EAAE,aAAa,EAClB,WAAW,EAAE,MAAM,EACnB,KAAK,CAAC,EAAE,MAAM,EACd,eAAe,CAAC,EAAE,MAAM,EACxB,MAAM,GAAE,eAAoB,GAC3B,OAAO,CAAC,MAAM,CAAC;YA2aJ,gBAAgB;IAuB9B,OAAO,CAAC,qBAAqB;YA6Bf,UAAU;YAmEV,QAAQ;IAKtB,OAAO,CAAC,cAAc;IAkBtB,OAAO,CAAC,aAAa;CAatB"}

package/dist/core/execution/dagExecutor.js

@@ -19,12 +19,18 @@ export class DAGExecutor {
     llmProvider;
     toolRegistry;
     artifactsDir;
+    apiKey;
+    ollamaBaseUrl;
+    skipGenerationStats;
     logger = getLogger();
     constructor(config) {
         this.db = config.db;
         this.llmProvider = config.llmProvider;
         this.toolRegistry = config.toolRegistry;
-        this.artifactsDir = config.artifactsDir || process.env.ARTIFACTS_DIR || './artifacts';
+        this.artifactsDir = config.artifactsDir;
+        this.apiKey = config.apiKey;
+        this.ollamaBaseUrl = config.ollamaBaseUrl;
+        this.skipGenerationStats = config.skipGenerationStats;
         this.logger.debug({
             provider: this.llmProvider.name,
         }, 'DAGExecutor created');
@@ -342,7 +348,7 @@ Respond with ONLY the expected output format. Build upon dependencies for cohere
                     if (!agent) {
                         throw new Error(`No agent found with name: ${agentName} (not in pre-fetch cache)`);
                     }
-                    const llmExecuteTool = new LlmExecuteTool();
+                    const llmExecuteTool = new LlmExecuteTool({ apiKey: this.apiKey, baseUrl: this.ollamaBaseUrl, skipGenerationStats: this.skipGenerationStats });
                     const result = await llmExecuteTool.execute({
                         provider: agent.provider,
                         model: agent.model,