octagon-funding-data-mcp/dist/index.js

MCP server for Funding Data Analysis. Provides specialized AI-powered company funding transaction analysis with comprehensive venture capital research, funding rounds, investor activity, and investment trends.

#!/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-funding-data-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"),
});
// Funding Agent
server.tool("octagon-funding-agent", "[PRIVATE MARKET INTELLIGENCE] A specialized database agent for company funding transactions. Capabilities: Extract information about funding rounds, investors, valuations, and investment trends. Best for: Researching startup funding history, investor activity, and venture capital patterns. NOTE: For better and more accurate results, provide the company's website URL instead of just the company name. Example queries: 'What was Anthropic's latest funding round size, valuation, and key investors (anthropic.com)?', 'How much has OpenAI raised in total funding and at what valuation (openai.com)?', 'Who were the lead investors in Databricks' Series G round and what was the post-money valuation (databricks.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-funding-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 Funding agent:", error);
        return {
            isError: true,
            content: [
                {
                    type: "text",
                    text: `Error: Failed to process funding data 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();