techconsptrs commited on
Commit
6ddf79a
·
1 Parent(s): 64b006f

UPDATE: format docs

Browse files
Files changed (2) hide show
  1. config.ini +2 -2
  2. src/components/rag/RAG.py +7 -1
config.ini CHANGED
@@ -1,6 +1,6 @@
1
  [EMBEDDINGS]
2
  embeddingModel = BAAI/bge-large-en-v1.5
3
- device = cuda
4
  normalize_embeddings = true
5
 
6
  [VECTORSTORE]
@@ -22,4 +22,4 @@ fetchK = 10
22
  timeout = 30
23
 
24
  [EASYOCR]
25
- gpu = true
 
1
  [EMBEDDINGS]
2
  embeddingModel = BAAI/bge-large-en-v1.5
3
+ device = cpu
4
  normalize_embeddings = true
5
 
6
  [VECTORSTORE]
 
22
  timeout = 30
23
 
24
  [EASYOCR]
25
+ gpu = false
src/components/rag/RAG.py CHANGED
@@ -25,7 +25,13 @@ class Chain:
25
  Returns:
26
  str: Formatted string with documents or a placeholder if empty.
27
  """
28
- context = "\n\n\n".join(docs) or "No Context Found"
 
 
 
 
 
 
29
  return context
30
 
31
  def returnChain(self, text: str):
 
25
  Returns:
26
  str: Formatted string with documents or a placeholder if empty.
27
  """
28
+ context = ""
29
+ for doc in docs:
30
+ context += f"{doc}\n\n\n"
31
+ if context == "":
32
+ context = "No Context Found"
33
+ else:
34
+ pass
35
  return context
36
 
37
  def returnChain(self, text: str):