Skip to content

Commit

Permalink
fix(solver): fix solver config (#192)
Browse files Browse the repository at this point in the history
* update reader register names

* add processed chunk id checkpoint

* add example config file

* update solver pipeline config

* fix project create

* fix main solver conf
  • Loading branch information
zhuzhongshu123 authored Dec 31, 2024
1 parent fee2f3f commit c63959d
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion kag/solver/main_solver.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,31 +49,39 @@ def invoke(
"recall_num": 10,
"rerank_topk": 10,
"type": "default_chunk_retriever",
"llm_client": llm_client,
},
"exact_kg_retriever": {
"el_num": 5,
"graph_api": {"type": "openspg_graph_api"},
"search_api": {"type": "openspg_search_api"},
"type": "default_exact_kg_retriever",
"llm_client": llm_client,
},
"force_chunk_retriever": True,
"fuzzy_kg_retriever": {
"el_num": 5,
"graph_api": {"type": "openspg_graph_api"},
"search_api": {"type": "openspg_search_api"},
"type": "default_fuzzy_kg_retriever",
"llm_client": llm_client,
},
"merger": {
"chunk_retriever": {
"recall_num": 10,
"rerank_topk": 10,
"llm_client": llm_client,
"type": "default_chunk_retriever",
},
"type": "default_lf_sub_query_res_merger",
},
"llm_client": llm_client,
"type": "default_lf_executor",
},
"lf_planner": {"type": "default_lf_planner"},
"lf_planner": {
"type": "default_lf_planner",
"llm_client": llm_client,
},
"llm_client": llm_client,
"type": "default_reasoner",
},
Expand Down

0 comments on commit c63959d

Please sign in to comment.