diff --git a/generic_rag/app.py b/generic_rag/app.py index aabbee1..999b910 100644 --- a/generic_rag/app.py +++ b/generic_rag/app.py @@ -91,14 +91,14 @@ vector_store = Chroma( if args.use_conditional_graph: graph = CondRetGenLangGraph( - vector_store, + vector_store=vector_store, chat_model=get_chat_model(args.chat_backend), embedding_model=get_embedding_model(args.emb_backend), system_prompt=system_prompt, ) else: graph = RetGenLangGraph( - vector_store, + vector_store=vector_store, chat_model=get_chat_model(args.chat_backend), embedding_model=get_embedding_model(args.emb_backend), system_prompt=system_prompt,