Squashed 'vendor/ruvector/' content from commit b64c2172

git-subtree-dir: vendor/ruvector git-subtree-split: b64c21726f2bb37286d9ee36a7869fef60cc6900
2026-02-28 14:39:40 -05:00
commit d803bfe2b1
7854 changed files with 3522914 additions and 0 deletions
--- a/npm/packages/graph-data-generator/src/generators/knowledge-graph.js
+++ b/npm/packages/graph-data-generator/src/generators/knowledge-graph.js
@@ -0,0 +1,192 @@
+"use strict";
+/**
+ * Knowledge graph generator using OpenRouter/Kimi K2
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KnowledgeGraphGenerator = void 0;
+exports.createKnowledgeGraphGenerator = createKnowledgeGraphGenerator;
+const DEFAULT_ENTITY_TYPES = [
+    'Person',
+    'Organization',
+    'Location',
+    'Event',
+    'Concept',
+    'Technology',
+    'Product'
+];
+const DEFAULT_RELATIONSHIP_TYPES = [
+    'WORKS_FOR',
+    'LOCATED_IN',
+    'CREATED_BY',
+    'PART_OF',
+    'RELATED_TO',
+    'INFLUENCES',
+    'DEPENDS_ON'
+];
+class KnowledgeGraphGenerator {
+    constructor(client) {
+        this.client = client;
+    }
+    /**
+     * Generate a knowledge graph
+     */
+    async generate(options) {
+        const startTime = Date.now();
+        // Generate entities first
+        const entities = await this.generateEntities(options);
+        // Generate relationships between entities
+        const relationships = await this.generateRelationships(entities, options);
+        // Convert to graph structure
+        const nodes = entities.map((entity, idx) => ({
+            id: entity.id || `entity_${idx}`,
+            labels: [entity.type || 'Entity'],
+            properties: {
+                name: entity.name,
+                ...entity.properties
+            }
+        }));
+        const edges = relationships.map((rel, idx) => ({
+            id: `rel_${idx}`,
+            type: rel.type,
+            source: rel.source,
+            target: rel.target,
+            properties: rel.properties || {}
+        }));
+        const data = {
+            nodes,
+            edges,
+            metadata: {
+                domain: options.domain,
+                generated_at: new Date(),
+                total_nodes: nodes.length,
+                total_edges: edges.length
+            }
+        };
+        return {
+            data,
+            metadata: {
+                generated_at: new Date(),
+                model: this.client.getConfig().model || 'moonshot/kimi-k2-instruct',
+                duration: Date.now() - startTime
+            }
+        };
+    }
+    /**
+     * Generate entities for the knowledge graph
+     */
+    async generateEntities(options) {
+        const entityTypes = options.entityTypes || DEFAULT_ENTITY_TYPES;
+        const systemPrompt = `You are an expert knowledge graph architect. Generate realistic entities for a knowledge graph about ${options.domain}.`;
+        const userPrompt = `Generate ${options.entities} diverse entities for a knowledge graph about ${options.domain}.
+
+Entity types to include: ${entityTypes.join(', ')}
+
+For each entity, provide:
+- id: unique identifier (use snake_case)
+- name: entity name
+- type: one of the specified entity types
+- properties: relevant properties (at least 2-3 properties per entity)
+
+Return a JSON array of entities. Make them realistic and relevant to ${options.domain}.
+
+Example format:
+\`\`\`json
+[
+  {
+    "id": "john_doe",
+    "name": "John Doe",
+    "type": "Person",
+    "properties": {
+      "role": "Software Engineer",
+      "expertise": "AI/ML",
+      "years_experience": 5
+    }
+  }
+]
+\`\`\``;
+        const entities = await this.client.generateStructured(systemPrompt, userPrompt, {
+            temperature: 0.8,
+            maxTokens: Math.min(8000, options.entities * 100)
+        });
+        return entities;
+    }
+    /**
+     * Generate relationships between entities
+     */
+    async generateRelationships(entities, options) {
+        const relationshipTypes = options.relationshipTypes || DEFAULT_RELATIONSHIP_TYPES;
+        const systemPrompt = `You are an expert at creating meaningful relationships in knowledge graphs. Create realistic relationships that make sense for ${options.domain}.`;
+        const entityList = entities.slice(0, 50).map(e => `- ${e.id}: ${e.name} (${e.type})`).join('\n');
+        const userPrompt = `Given these entities from a ${options.domain} knowledge graph:
+
+${entityList}
+
+Generate ${options.relationships} meaningful relationships between them.
+
+Relationship types to use: ${relationshipTypes.join(', ')}
+
+For each relationship, provide:
+- source: source entity id
+- target: target entity id
+- type: relationship type (use one of the specified types)
+- properties: optional properties describing the relationship
+
+Return a JSON array of relationships. Make them logical and realistic.
+
+Example format:
+\`\`\`json
+[
+  {
+    "source": "john_doe",
+    "target": "acme_corp",
+    "type": "WORKS_FOR",
+    "properties": {
+      "since": "2020",
+      "position": "Senior Engineer"
+    }
+  }
+]
+\`\`\``;
+        const relationships = await this.client.generateStructured(systemPrompt, userPrompt, {
+            temperature: 0.7,
+            maxTokens: Math.min(8000, options.relationships * 80)
+        });
+        return relationships;
+    }
+    /**
+     * Generate knowledge triples (subject-predicate-object)
+     */
+    async generateTriples(domain, count) {
+        const systemPrompt = `You are an expert at extracting knowledge triples from domains. Generate meaningful subject-predicate-object triples about ${domain}.`;
+        const userPrompt = `Generate ${count} knowledge triples about ${domain}.
+
+Each triple should have:
+- subject: the entity or concept
+- predicate: the relationship or property
+- object: the related entity, value, or concept
+- confidence: confidence score (0-1)
+
+Return a JSON array of triples.
+
+Example format:
+\`\`\`json
+[
+  {
+    "subject": "Einstein",
+    "predicate": "developed",
+    "object": "Theory of Relativity",
+    "confidence": 1.0
+  }
+]
+\`\`\``;
+        return this.client.generateStructured(systemPrompt, userPrompt, { temperature: 0.7, maxTokens: count * 100 });
+    }
+}
+exports.KnowledgeGraphGenerator = KnowledgeGraphGenerator;
+/**
+ * Create a knowledge graph generator
+ */
+function createKnowledgeGraphGenerator(client) {
+    return new KnowledgeGraphGenerator(client);
+}
+//# sourceMappingURL=knowledge-graph.js.map