run_graphiti_benchmark.py•573 B
#!/usr/bin/env python3
"""Run Graphiti QA benchmark."""
from qa.qa_benchmark_graphiti import QABenchmarkGraphiti, GraphitiConfig
def main():
"""Run Graphiti benchmark."""
config = GraphitiConfig(
corpus_limit=None, # Small test
qa_limit=None,
print_results=True,
)
benchmark = QABenchmarkGraphiti.from_jsons(
corpus_file="hotpot_qa_24_corpus.json",
qa_pairs_file="hotpot_qa_24_qa_pairs.json",
config=config,
)
results = benchmark.run()
return results
if __name__ == "__main__":
main()