ragstack-ai-knowledge-store

import os from ragstack_ai.knowledge_store import AstraDBKnowledgeStore from langchain_core.documents import Document # Ensure Astra DB credentials are set as environment variables # ASTRA_DB_APPLICATION_TOKEN, ASTRA_DB_ID, ASTRA_DB_KEYSPACE (optional, defaults to 'default') astra_db_token = os.environ.get("ASTRA_DB_APPLICATION_TOKEN", "YOUR_ASTRA_DB_TOKEN") astra_db_id = os.environ.get("ASTRA_DB_ID", "YOUR_ASTRA_DB_ID") astra_db_keyspace = os.environ.get("ASTRA_DB_KEYSPACE", "default") if not all([astra_db_token, astra_db_id]): print("Please set ASTRA_DB_APPLICATION_TOKEN and ASTRA_DB_ID environment variables.") exit() try: # Initialize the Knowledge Store kg_store = AstraDBKnowledgeStore( astra_db_id=astra_db_id, astra_db_application_token=astra_db_token, astra_db_keyspace=astra_db_keyspace, graph_name="my_rag_knowledge_graph" # A unique name for your graph ) # Add documents to the knowledge store documents = [ Document(page_content="Leonardo da Vinci was an Italian polymath.", metadata={"source": "wiki"}), Document(page_content="The Mona Lisa is a half-length portrait painting by Leonardo da Vinci.", metadata={"source": "wiki"}) ] print("Adding documents...") kg_store.add_documents(documents=documents) print("Documents added successfully.") # Example: Simple graph traversal (conceptual, actual queries depend on graph structure) # Note: Complex graph queries typically use specific methods or a dedicated graph query engine. # This demonstrates basic interaction. print(f"Knowledge Store initialized for graph: {kg_store.graph_name}") except Exception as e: print(f"An error occurred: {e}") print("Ensure your Astra DB credentials are correct and the database is accessible.")