Spaces:
Sleeping
Sleeping
Ilyas KHIAT
commited on
Commit
·
38fc6d8
1
Parent(s):
3e713fa
choisir par scene
Browse files
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 |
-
|
|
|
|
| 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:
|