(db_name: str = "paper")
| 60 | |
| 61 | |
| 62 | def example_multi_db(db_name: str = "paper"): |
| 63 | # Step 1: Build factory config |
| 64 | config = GraphDBConfigFactory( |
| 65 | backend="neo4j", |
| 66 | config={ |
| 67 | "uri": NEO4J_URI, |
| 68 | "user": NEO4J_USER, |
| 69 | "password": NEO4J_PASSWORD, |
| 70 | "db_name": db_name, |
| 71 | "auto_create": True, |
| 72 | "embedding_dimension": EMBEDDING_DIMENSION, |
| 73 | "use_multi_db": True, |
| 74 | }, |
| 75 | ) |
| 76 | |
| 77 | # Step 2: Instantiate the graph store |
| 78 | graph = GraphStoreFactory.from_config(config) |
| 79 | graph.clear() |
| 80 | |
| 81 | # Step 3: Create topic node |
| 82 | topic = TextualMemoryItem( |
| 83 | memory="This research addresses long-term multi-UAV navigation for energy-efficient communication coverage.", |
| 84 | metadata=TreeNodeTextualMemoryMetadata( |
| 85 | memory_type="LongTermMemory", |
| 86 | key="Multi-UAV Long-Term Coverage", |
| 87 | hierarchy_level="topic", |
| 88 | type="fact", |
| 89 | memory_time="2024-01-01", |
| 90 | source="file", |
| 91 | sources=["paper://multi-uav-coverage/intro"], |
| 92 | status="activated", |
| 93 | confidence=95.0, |
| 94 | tags=["UAV", "coverage", "multi-agent"], |
| 95 | entities=["UAV", "coverage", "navigation"], |
| 96 | visibility="public", |
| 97 | updated_at=datetime.now().isoformat(), |
| 98 | embedding=embed_memory_item( |
| 99 | "This research addresses long-term " |
| 100 | "multi-UAV navigation for " |
| 101 | "energy-efficient communication " |
| 102 | "coverage." |
| 103 | ), |
| 104 | ), |
| 105 | ) |
| 106 | |
| 107 | graph.add_node( |
| 108 | id=topic.id, memory=topic.memory, metadata=topic.metadata.model_dump(exclude_none=True) |
| 109 | ) |
| 110 | |
| 111 | # Step 4: Define and write concept nodes |
| 112 | concepts = [ |
| 113 | TextualMemoryItem( |
| 114 | memory="The reward function combines multiple objectives: coverage maximization, energy consumption minimization, and overlap penalty.", |
| 115 | metadata=TreeNodeTextualMemoryMetadata( |
| 116 | memory_type="LongTermMemory", |
| 117 | key="Reward Function Design", |
| 118 | hierarchy_level="concept", |
| 119 | type="fact", |
no test coverage detected