@olane/o-tool-registry 0.6.2 → 0.6.3

package/dist/src/intelligence/gemini-intelligence.tool.js

@@ -0,0 +1,267 @@
+import { oAddress } from '@olane/o-core';
+import { oVirtualTool } from '@olane/o-tool';
+import { INTELLIGENCE_PARAMS } from './methods/intelligence.methods.js';
+export class GeminiIntelligenceTool extends oVirtualTool {
+    constructor(config) {
+        super({
+            ...config,
+            address: new oAddress('o://gemini'),
+            description: 'Intelligence tool using Google Gemini suite of models',
+            methods: INTELLIGENCE_PARAMS,
+            dependencies: [],
+        });
+    }
+    /**
+     * Chat completion with Gemini
+     */
+    async _tool_completion(request) {
+        try {
+            const params = request.params;
+            const { model = this.defaultModel, messages, ...options } = params;
+            if (!this.apiKey) {
+                return {
+                    success: false,
+                    error: 'Gemini API key is required',
+                };
+            }
+            if (!messages || !Array.isArray(messages)) {
+                return {
+                    success: false,
+                    error: '"messages" array is required',
+                };
+            }
+            // Convert messages to Gemini format
+            const contents = messages.map((msg) => ({
+                role: msg.role === 'assistant' ? 'model' : 'user',
+                parts: [{ text: msg.content }],
+            }));
+            const chatRequest = {
+                contents,
+                generationConfig: {
+                    temperature: options.temperature,
+                    topK: options.topK,
+                    topP: options.topP,
+                    maxOutputTokens: options.maxOutputTokens,
+                    stopSequences: options.stopSequences,
+                },
+                safetySettings: options.safetySettings,
+            };
+            const response = await fetch(`${this.baseUrl}/models/${model}:generateContent?key=${this.apiKey}`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify(chatRequest),
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                return {
+                    success: false,
+                    error: `Gemini API error: ${response.status} - ${errorText}`,
+                };
+            }
+            const result = (await response.json());
+            if (!result.candidates || result.candidates.length === 0) {
+                return {
+                    success: false,
+                    error: 'No response generated from Gemini',
+                };
+            }
+            return {
+                success: true,
+                response: result.candidates[0].content.parts[0]?.text || '',
+                model: model,
+                usage: result.usageMetadata,
+                finish_reason: result.candidates[0].finishReason,
+                safety_ratings: result.candidates[0].safetyRatings,
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: `Failed to complete chat: ${error.message}`,
+            };
+        }
+    }
+    /**
+     * Generate text with Gemini
+     */
+    async _tool_generate(request) {
+        try {
+            const params = request.params;
+            const { model = this.defaultModel, prompt, system, ...options } = params;
+            if (!this.apiKey) {
+                return {
+                    success: false,
+                    error: 'Gemini API key is required',
+                };
+            }
+            if (!prompt) {
+                return {
+                    success: false,
+                    error: 'Prompt is required',
+                };
+            }
+            // Combine system and user prompt
+            const fullPrompt = system ? `${system}\n\n${prompt}` : prompt;
+            const generateRequest = {
+                contents: [
+                    {
+                        parts: [{ text: fullPrompt }],
+                    },
+                ],
+                generationConfig: {
+                    temperature: options.temperature,
+                    topK: options.topK,
+                    topP: options.topP,
+                    maxOutputTokens: options.maxOutputTokens,
+                    stopSequences: options.stopSequences,
+                },
+                safetySettings: options.safetySettings,
+            };
+            const response = await fetch(`${this.baseUrl}/models/${model}:generateContent?key=${this.apiKey}`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify(generateRequest),
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                return {
+                    success: false,
+                    error: `Gemini API error: ${response.status} - ${errorText}`,
+                };
+            }
+            const result = (await response.json());
+            if (!result.candidates || result.candidates.length === 0) {
+                return {
+                    success: false,
+                    error: 'No response generated from Gemini',
+                };
+            }
+            return {
+                success: true,
+                response: result.candidates[0].content.parts[0]?.text || '',
+                model: model,
+                usage: result.usageMetadata,
+                finish_reason: result.candidates[0].finishReason,
+                safety_ratings: result.candidates[0].safetyRatings,
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: `Failed to generate text: ${error.message}`,
+            };
+        }
+    }
+    /**
+     * List available models
+     */
+    async _tool_list_models(request) {
+        try {
+            if (!this.apiKey) {
+                return {
+                    success: false,
+                    error: 'Gemini API key is required',
+                };
+            }
+            const response = await fetch(`${this.baseUrl}/models?key=${this.apiKey}`, {
+                method: 'GET',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                return {
+                    success: false,
+                    error: `Gemini API error: ${response.status} - ${errorText}`,
+                };
+            }
+            const result = (await response.json());
+            return {
+                success: true,
+                models: result.models,
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: `Failed to list models: ${error.message}`,
+            };
+        }
+    }
+    /**
+     * Get model information
+     */
+    async _tool_model_info(request) {
+        try {
+            const params = request.params;
+            const { model = this.defaultModel } = params;
+            if (!this.apiKey) {
+                return {
+                    success: false,
+                    error: 'Gemini API key is required',
+                };
+            }
+            const response = await fetch(`${this.baseUrl}/models/${model}?key=${this.apiKey}`, {
+                method: 'GET',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                return {
+                    success: false,
+                    error: `Gemini API error: ${response.status} - ${errorText}`,
+                };
+            }
+            const result = (await response.json());
+            return {
+                success: true,
+                model_info: result,
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: `Failed to get model info: ${error.message}`,
+            };
+        }
+    }
+    /**
+     * Check Gemini API status
+     */
+    async _tool_status(request) {
+        try {
+            if (!this.apiKey) {
+                return {
+                    success: false,
+                    status: 'offline',
+                    error: 'Gemini API key is required',
+                };
+            }
+            const response = await fetch(`${this.baseUrl}/models?key=${this.apiKey}`, {
+                method: 'GET',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+            });
+            return {
+                success: response.ok,
+                status: response.ok ? 'online' : 'offline',
+                status_code: response.status,
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                status: 'offline',
+                error: `Connection failed: ${error.message}`,
+            };
+        }
+    }
+}

package/dist/src/intelligence/index.d.ts

@@ -0,0 +1,6 @@
+export * from './ollama-intelligence.tool.js';
+export * from './openai-intelligence.tool.js';
+export * from './anthropic-intelligence.tool.js';
+export * from './gemini-intelligence.tool.js';
+export * from './intelligence.tool.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/src/intelligence/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/intelligence/index.ts"],"names":[],"mappings":"AAAA,cAAc,+BAA+B,CAAC;AAC9C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,kCAAkC,CAAC;AACjD,cAAc,+BAA+B,CAAC;AAC9C,cAAc,wBAAwB,CAAC"}

package/dist/src/intelligence/index.js

@@ -0,0 +1,5 @@
+export * from './ollama-intelligence.tool.js';
+export * from './openai-intelligence.tool.js';
+export * from './anthropic-intelligence.tool.js';
+export * from './gemini-intelligence.tool.js';
+export * from './intelligence.tool.js';

package/dist/src/intelligence/intelligence.tool.d.ts

@@ -0,0 +1,11 @@
+import { oToolConfig, oVirtualTool } from '@olane/o-tool';
+import { oRequest } from '@olane/o-core';
+import { ToolResult } from '@olane/o-tool';
+export declare class IntelligenceTool extends oVirtualTool {
+    private roundRobinIndex;
+    constructor(config: oToolConfig);
+    requestMissingData(): Promise<ToolResult>;
+    chooseIntelligence(request: oRequest): Promise<ToolResult>;
+    _tool_prompt(request: oRequest): Promise<ToolResult>;
+}
+//# sourceMappingURL=intelligence.tool.d.ts.map

package/dist/src/intelligence/intelligence.tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"intelligence.tool.d.ts","sourceRoot":"","sources":["../../../src/intelligence/intelligence.tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE1D,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAO3C,qBAAa,gBAAiB,SAAQ,YAAY;IAChD,OAAO,CAAC,eAAe,CAAK;gBAChB,MAAM,EAAE,WAAW;IAmDzB,kBAAkB,IAAI,OAAO,CAAC,UAAU,CAAC;IAoCzC,kBAAkB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IAqC1D,YAAY,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;CAiB3D"}

package/dist/src/intelligence/intelligence.tool.js

@@ -0,0 +1,134 @@
+import { oVirtualTool } from '@olane/o-tool';
+import { oAddress } from '@olane/o-core';
+import { AnthropicIntelligenceTool } from './anthropic-intelligence.tool.js';
+import { OpenAIIntelligenceTool } from './openai-intelligence.tool.js';
+import { OllamaIntelligenceTool } from './ollama-intelligence.tool.js';
+import { PerplexityIntelligenceTool } from './perplexity-intelligence.tool.js';
+import { INTELLIGENCE_PARAMS } from './methods/intelligence.methods.js';
+export class IntelligenceTool extends oVirtualTool {
+    constructor(config) {
+        super({
+            ...config,
+            address: new oAddress('o://intelligence'),
+            methods: INTELLIGENCE_PARAMS,
+            description: config.description ||
+                'Tool to help route LLM requests to the best intelligence tool',
+            dependencies: [
+                {
+                    address: 'o://setup',
+                    parameters: [
+                        {
+                            name: 'intelligence',
+                            type: 'string',
+                            description: 'The intelligence tool to use',
+                        },
+                    ],
+                },
+            ],
+        });
+        this.roundRobinIndex = 0;
+        this.addChildNode(new AnthropicIntelligenceTool({
+            ...config,
+            parent: null,
+            leader: null,
+        }));
+        this.addChildNode(new OpenAIIntelligenceTool({
+            ...config,
+            parent: null,
+            leader: null,
+        }));
+        this.addChildNode(new OllamaIntelligenceTool({
+            ...config,
+            parent: null,
+            leader: null,
+        }));
+        this.addChildNode(new PerplexityIntelligenceTool({
+            ...config,
+            parent: null,
+            leader: null,
+        }));
+    }
+    async requestMissingData() {
+        // check to see if the anthropic key is provided in the ENV vars
+        if (process.env.ANTHROPIC_API_KEY) {
+            return {
+                choice: 'o://anthropic',
+                apiKey: process.env.ANTHROPIC_API_KEY,
+            };
+        }
+        // if the anthropic key is not in the vault, ask the human
+        this.logger.info('Anthropic API key not found in vault, asking human');
+        const humanResponse = await this.use(new oAddress('o://human'), {
+            method: 'question',
+            params: {
+                question: 'Enter the anthropic api key',
+            },
+        });
+        // process the human response
+        const { answer } = humanResponse.result.data;
+        this.logger.info('Human answer: ', answer);
+        await this.use(new oAddress('o://memory'), {
+            method: 'put',
+            params: {
+                key: 'anthropic-api-key',
+                value: answer,
+            },
+        });
+        return {
+            choice: 'o://anthropic',
+            apiKey: answer,
+        };
+    }
+    async chooseIntelligence(request) {
+        // check to see if anthropic key is in vault
+        const preference = await this.use(new oAddress('o://memory'), {
+            method: 'get',
+            params: {
+                key: 'intelligence-preference',
+            },
+        });
+        const { value } = preference.result.data;
+        if (value) {
+            return {
+                choice: value,
+                apiKey: '',
+            };
+        }
+        const response = await this.use(new oAddress('o://memory'), {
+            method: 'get',
+            params: {
+                key: 'anthropic-api-key',
+            },
+        });
+        // if the anthropic key is in the vault, use it
+        if (response.result.data) {
+            const { value } = response.result.data;
+            if (value) {
+                return {
+                    choice: 'o://anthropic',
+                    apiKey: value,
+                };
+            }
+        }
+        const result = await this.requestMissingData();
+        return result;
+    }
+    // we cannot wrap this tool use in a plan because it is a core dependency in all planning
+    async _tool_prompt(request) {
+        const { prompt } = request.params;
+        const intelligence = await this.chooseIntelligence(request);
+        const response = await this.use(new oAddress(intelligence.choice), {
+            method: 'completion',
+            params: {
+                apiKey: intelligence.apiKey,
+                messages: [
+                    {
+                        role: 'user',
+                        content: prompt,
+                    },
+                ],
+            },
+        });
+        return response.result.data;
+    }
+}

package/dist/src/intelligence/methods/intelligence.methods.d.ts

@@ -0,0 +1,5 @@
+import { oMethod } from '@olane/o-protocol';
+export declare const INTELLIGENCE_PARAMS: {
+    [key: string]: oMethod;
+};
+//# sourceMappingURL=intelligence.methods.d.ts.map

package/dist/src/intelligence/methods/intelligence.methods.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"intelligence.methods.d.ts","sourceRoot":"","sources":["../../../../src/intelligence/methods/intelligence.methods.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAE5C,eAAO,MAAM,mBAAmB,EAAE;IAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CAmIzD,CAAC"}

package/dist/src/intelligence/methods/intelligence.methods.js

@@ -0,0 +1,132 @@
+export const INTELLIGENCE_PARAMS = {
+    completion: {
+        name: 'completion',
+        description: 'Completion',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'model',
+                type: 'string',
+                value: 'string',
+                description: 'The model to use for generation',
+                required: false,
+            },
+            {
+                name: 'messages',
+                type: 'array',
+                value: 'string[]',
+                description: 'The messages to use for generation',
+            },
+            {
+                name: 'options',
+                type: 'object',
+                value: 'object',
+                description: 'The options to use for generation',
+                required: false,
+            },
+        ],
+    },
+    generate: {
+        name: 'generate',
+        description: 'Generate',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'model',
+                type: 'string',
+                value: 'string',
+                description: 'The model to use for generation',
+            },
+        ],
+    },
+    list_models: {
+        name: 'list_models',
+        description: 'List models',
+        dependencies: [],
+        parameters: [],
+    },
+    pull_model: {
+        name: 'pull_model',
+        description: 'Pull model',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'model',
+                type: 'string',
+                value: 'string',
+                description: 'The model to pull',
+            },
+            {
+                name: 'insecure',
+                type: 'boolean',
+                value: 'boolean',
+                description: 'Whether to allow insecure connections',
+            },
+        ],
+    },
+    delete_model: {
+        name: 'delete_model',
+        description: 'Delete model',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'model',
+                type: 'string',
+                value: 'string',
+                description: 'The model to delete',
+            },
+        ],
+    },
+    model_info: {
+        name: 'model_info',
+        description: 'Model info',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'model',
+                type: 'string',
+                value: 'string',
+                description: 'The model to get info for',
+            },
+        ],
+    },
+    status: {
+        name: 'status',
+        description: 'Status',
+        dependencies: [],
+        parameters: [],
+    },
+    prompt: {
+        name: 'prompt',
+        description: 'Generate a response using AI based on a prompt',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'prompt',
+                type: 'string',
+                value: 'string',
+                description: 'The prompt to send to the AI model',
+            },
+        ],
+    },
+    search: {
+        name: 'search',
+        description: 'Search for information using AI search capabilities',
+        dependencies: [],
+        parameters: [
+            {
+                name: 'query',
+                type: 'string',
+                value: 'string',
+                description: 'The search query to execute',
+            },
+            {
+                name: 'focus',
+                type: 'string',
+                value: 'string',
+                description: 'The focus area for the search',
+                required: false,
+            },
+        ],
+    },
+};

package/dist/src/intelligence/ollama-intelligence.tool.d.ts

@@ -0,0 +1,36 @@
+import { oRequest } from '@olane/o-core';
+import { oToolConfig, oVirtualTool, ToolResult } from '@olane/o-tool';
+export declare class OllamaIntelligenceTool extends oVirtualTool {
+    static defaultModel: string;
+    static defaultUrl: string;
+    constructor(config: oToolConfig);
+    /**
+     * Chat completion with Ollama
+     */
+    _tool_completion(request: oRequest): Promise<ToolResult>;
+    /**
+     * Generate text with Ollama
+     */
+    _tool_generate(request: oRequest): Promise<ToolResult>;
+    /**
+     * List available models
+     */
+    _tool_list_models(request: oRequest): Promise<ToolResult>;
+    /**
+     * Pull a model from Ollama library
+     */
+    _tool_pull_model(request: oRequest): Promise<ToolResult>;
+    /**
+     * Delete a model
+     */
+    _tool_delete_model(request: oRequest): Promise<ToolResult>;
+    /**
+     * Get model information
+     */
+    _tool_model_info(request: oRequest): Promise<ToolResult>;
+    /**
+     * Check Ollama server status
+     */
+    _tool_status(request: oRequest): Promise<ToolResult>;
+}
+//# sourceMappingURL=ollama-intelligence.tool.d.ts.map

package/dist/src/intelligence/ollama-intelligence.tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama-intelligence.tool.d.ts","sourceRoot":"","sources":["../../../src/intelligence/ollama-intelligence.tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAY,QAAQ,EAAgB,MAAM,eAAe,CAAC;AACjE,OAAO,EAAS,WAAW,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AA+G7E,qBAAa,sBAAuB,SAAQ,YAAY;IACtD,MAAM,CAAC,YAAY,SAAoB;IACvC,MAAM,CAAC,UAAU,SAA4B;gBAEjC,MAAM,EAAE,WAAW;IAY/B;;OAEG;IACG,gBAAgB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IA4D9D;;OAEG;IACG,cAAc,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IAgE5D;;OAEG;IACG,iBAAiB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IAmC/D;;OAEG;IACG,gBAAgB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IAiE9D;;OAEG;IACG,kBAAkB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IA6ChE;;OAEG;IACG,gBAAgB,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;IAwC9D;;OAEG;IACG,YAAY,CAAC,OAAO,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC;CAyB3D"}