forked from AI_team/Philosophy-RAG-demo
define param explicit
This commit is contained in:
parent
939a7044b5
commit
5a15f1bd16
@ -91,14 +91,14 @@ vector_store = Chroma(
|
|||||||
|
|
||||||
if args.use_conditional_graph:
|
if args.use_conditional_graph:
|
||||||
graph = CondRetGenLangGraph(
|
graph = CondRetGenLangGraph(
|
||||||
vector_store,
|
vector_store=vector_store,
|
||||||
chat_model=get_chat_model(args.chat_backend),
|
chat_model=get_chat_model(args.chat_backend),
|
||||||
embedding_model=get_embedding_model(args.emb_backend),
|
embedding_model=get_embedding_model(args.emb_backend),
|
||||||
system_prompt=system_prompt,
|
system_prompt=system_prompt,
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
graph = RetGenLangGraph(
|
graph = RetGenLangGraph(
|
||||||
vector_store,
|
vector_store=vector_store,
|
||||||
chat_model=get_chat_model(args.chat_backend),
|
chat_model=get_chat_model(args.chat_backend),
|
||||||
embedding_model=get_embedding_model(args.emb_backend),
|
embedding_model=get_embedding_model(args.emb_backend),
|
||||||
system_prompt=system_prompt,
|
system_prompt=system_prompt,
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user