graphiti_core_temporal_knowledge_graph_ingest_and_search.py

1import asyncio
2from graphiti import Graphiti
3
4async def main():
5    # Initialize the Graphiti client
6    # Expects NEO4J_URI, NEO4J_USERNAME, NEO4J_PASSWORD, and OPENAI_API_KEY env vars
7    graphiti = Graphiti()
8
9    # Add data to the graph
10    # Graphiti automatically extracts entities and relationships
11    await graphiti.build_indices()
12    
13    # Ingesting a narrative
14    text = "Alice is a software engineer who moved to San Francisco in 2023. She started working at TechCorp."
15    await graphiti.add_block(text)
16
17    # Search the knowledge graph
18    results = await graphiti.search("Where does Alice work?")
19    
20    for result in results:
21        print(f"Result: {result.text}")
22
23if __name__ == "__main__":
24    asyncio.run(main())