mirror of
https://github.com/OpenSPG/KAG.git
synced 2025-06-27 03:20:08 +00:00
32 lines
783 B
Python
32 lines
783 B
Python
import json
|
|
import logging
|
|
import os
|
|
import time
|
|
from concurrent.futures import ThreadPoolExecutor, as_completed
|
|
|
|
from tqdm import tqdm
|
|
|
|
from kag.common.benchmarks.evaluate import Evaluate
|
|
from kag.solver.logic.solver_pipeline import SolverPipeline
|
|
from kag.common.conf import KAG_CONFIG
|
|
from kag.common.registry import import_modules_from_path
|
|
|
|
from kag.common.checkpointer import CheckpointerManager
|
|
|
|
|
|
def qa(query):
|
|
resp = SolverPipeline.from_config(KAG_CONFIG.all_config["kag_solver_pipeline"])
|
|
answer, traceLog = resp.run(query)
|
|
|
|
print(f"\n\nso the answer for '{query}' is: {answer}\n\n") #
|
|
print(traceLog)
|
|
return answer, traceLog
|
|
|
|
|
|
if __name__ == "__main__":
|
|
queries = [
|
|
"皮质激素有什么作用",
|
|
]
|
|
for q in queries:
|
|
qa(q)
|