anekameni commited on
Commit
c365331
·
1 Parent(s): 91dda71

Add default config for context len in chat

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -38,7 +38,8 @@ class ChatInterface:
38
 
39
  # Usage example:
40
  if __name__ == "__main__":
41
- rag_system = RAGSystem(top_k_documents=12)
 
42
  rag_system.initialize_vector_store()
43
 
44
  chat_interface = ChatInterface(rag_system)
 
38
 
39
  # Usage example:
40
  if __name__ == "__main__":
41
+ top_k_docs = int(os.getenv("N_CONTEXT") or 8)
42
+ rag_system = RAGSystem(top_k_documents=top_k_docs)
43
  rag_system.initialize_vector_store()
44
 
45
  chat_interface = ChatInterface(rag_system)