jgrivolla commited on
Commit
bc7c73a
·
verified ·
1 Parent(s): ce269b1

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -54,7 +54,7 @@ class RAG:
54
  stream=False
55
  )
56
  print(chat_completion)
57
- return(chat_completion)
58
 
59
 
60
  def beautiful_context(self, docs):
 
54
  stream=False
55
  )
56
  print(chat_completion)
57
+ return(chat_completion.choices[0].delta.content)
58
 
59
 
60
  def beautiful_context(self, docs):