Ilyas KHIAT commited on
Commit
38fc6d8
·
1 Parent(s): 3e713fa

choisir par scene

Browse files
Files changed (1) hide show
  1. rag.py +4 -1
rag.py CHANGED
@@ -86,9 +86,11 @@ def split_texts(text : str) -> list[str]:
86
  )
87
  return splitter.split_text(text)
88
 
 
89
  #########################################################################
90
  ### PAR ICI , CHOISIR UNE SCENE SPECIFIQUE DANS L'ARGUMENT DE LA FONCTION
91
- def get_random_chunk(scene_specific = 5) :
 
92
  if scene_specific:
93
  chunks_scene = split_texts(scenes[scene_specific-1])
94
  print(f"Scene {scene_specific} has {len(chunks_scene)} chunks")
@@ -131,6 +133,7 @@ def generate_sphinx_response() -> sphinx_output:
131
 
132
  return llm_chain.invoke({"writer":writer,"book_name":book_name,"summary":summary,"excerpt":excerpt})
133
 
 
134
  #############################################################
135
  ### PAR ICI , CHOISIR LE DEGRE DE SEVERITE DE LA VERIFICATION
136
  def verify_response(response:str,answers:list[str],question:str) -> bool:
 
86
  )
87
  return splitter.split_text(text)
88
 
89
+
90
  #########################################################################
91
  ### PAR ICI , CHOISIR UNE SCENE SPECIFIQUE DANS L'ARGUMENT DE LA FONCTION
92
+
93
+ def get_random_chunk(scene_specific = 5) : # scene_specific = None signifie qu'on considère tout le récit
94
  if scene_specific:
95
  chunks_scene = split_texts(scenes[scene_specific-1])
96
  print(f"Scene {scene_specific} has {len(chunks_scene)} chunks")
 
133
 
134
  return llm_chain.invoke({"writer":writer,"book_name":book_name,"summary":summary,"excerpt":excerpt})
135
 
136
+
137
  #############################################################
138
  ### PAR ICI , CHOISIR LE DEGRE DE SEVERITE DE LA VERIFICATION
139
  def verify_response(response:str,answers:list[str],question:str) -> bool: