octagon-investors-mcp/dist/index.js

MCP server for Investors Analysis. Provides specialized AI-powered investor information and investment activity analysis with comprehensive business intelligence, investment criteria, and market research.

#!/usr/bin/env node
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import dotenv from "dotenv";
import { readFile } from "fs/promises";
import OpenAI from "openai";
import path from "path";
import { fileURLToPath } from "url";
import { z } from "zod";
// Get package.json info
const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);
const packageJsonPath = path.join(__dirname, "..", "package.json");
const packageJsonContent = await readFile(packageJsonPath, "utf8");
const packageInfo = JSON.parse(packageJsonContent);
// Load environment variables
dotenv.config();
// Check for required environment variables
const OCTAGON_API_KEY = process.env.OCTAGON_API_KEY;
const OCTAGON_API_BASE_URL = process.env.OCTAGON_API_BASE_URL || "https://api.octagonagents.com/v1";
if (!OCTAGON_API_KEY) {
    console.error("Error: OCTAGON_API_KEY is not set in the environment variables");
    console.error("Please set the OCTAGON_API_KEY environment variable or use 'env OCTAGON_API_KEY=your_key npx -y octagon-private-companies-mcp'");
    process.exit(1);
}
// Initialize OpenAI client with Octagon API
const octagonClient = new OpenAI({
    apiKey: OCTAGON_API_KEY,
    baseURL: OCTAGON_API_BASE_URL,
    defaultHeaders: {
        "User-Agent": `${packageInfo.name}/${packageInfo.version} (Node.js/${process.versions.node})`
    },
});
// Create MCP server
const server = new McpServer({
    name: packageInfo.name,
    version: packageInfo.version,
});
// Helper function to process streaming responses
async function processStreamingResponse(stream) {
    let fullResponse = "";
    let citations = [];
    try {
        // Process the streaming response
        for await (const chunk of stream) {
            // For Chat Completions API
            if (chunk.choices && chunk.choices[0]?.delta?.content) {
                fullResponse += chunk.choices[0].delta.content;
                // Check for citations in the final chunk
                if (chunk.choices[0]?.finish_reason === "stop" && chunk.choices[0]?.citations) {
                    citations = chunk.choices[0].citations;
                }
            }
            // For Responses API
            if (chunk.type === "response.output_text.delta") {
                fullResponse += chunk.text?.delta || "";
            }
        }
        return fullResponse;
    }
    catch (error) {
        console.error("Error processing streaming response:", error);
        throw error;
    }
}
// Define a schema for the 'prompt' parameter that all tools will use
const promptSchema = z.object({
    prompt: z.string().describe("Your natural language query or request for the agent"),
});
// Investors Agent
server.tool("octagon-investors-agent", "[PRIVATE MARKET INTELLIGENCE] A specialized database agent for looking up information on investors including angel investors, venture capital, and private equity. Capabilities: Retrieve information about investors, their investment criteria, and past activities. Best for: Research on investors and details about their investment activities. NOTE: For better and more accurate results, provide the investor's website URL instead of just the investor name. Example queries: 'What is the latest investment criteria of Insight Partners (insightpartners.com)?', 'How many investments did Andreessen Horowitz (a16z.com) make in the last 6 months', 'What is the typical check size for QED Investors (qedinvestors.com)'.", {
    prompt: z.string().describe("Your natural language query or request for the agent"),
}, async ({ prompt }) => {
    try {
        const response = await octagonClient.chat.completions.create({
            model: "octagon-investors-agent",
            messages: [{ role: "user", content: prompt }],
            stream: true,
            metadata: { tool: "mcp" }
        });
        const result = await processStreamingResponse(response);
        return {
            content: [
                {
                    type: "text",
                    text: result,
                },
            ],
        };
    }
    catch (error) {
        console.error("Error calling Investors agent:", error);
        return {
            isError: true,
            content: [
                {
                    type: "text",
                    text: `Error: Failed to process investors query. ${error}`,
                },
            ],
        };
    }
});
// Start the server with stdio transport
async function main() {
    try {
        const transport = new StdioServerTransport();
        await server.connect(transport);
    }
    catch (error) {
        process.exit(1);
    }
}
main();