import os
from pyTigerGraph import TigerGraphConnection

host = "http://localhost"

username = "tigergraph"
password = "tigergraph"

# We first create a connection to the database
conn = TigerGraphConnection(
    host=host,
    username=username, 
    password=password,
    restppPort=14240
    )

conn.graphname = "TigerGraphRAG"

# And then add GraphRAG's address to the connection. This address
# is the host's address where the GraphRAG container is running.
conn.ai.configureGraphRAGHost(f"{host}:8000")

query = "how to load data to tigergraph vector store, give an example in Python"
print(f"""Fetching answer for question: {query}""")

resp = conn.ai.answerQuestion(
    query,
    method="hybrid",
    method_parameters = {
        "indices": ["Document", "DocumentChunk", "Entity", "Relationship"],
        "top_k": 2,
        "num_hops": 2,
        "num_seen_min": 2,
        "verbose": True
    })

print(f"""\nAnswer using Hybrid Search:\n{resp["response"]}""")

resp = conn.ai.answerQuestion(
    query,
    method="community",
    method_parameters={
        "community_level": 2,
        "combine": False,
        "top_k": 5,
        "verbose": True
    })

print(f"""\nAnswer using Community Search:\n{resp["response"]}""")