This commit is contained in:
Saifeddine ALOUI 2024-07-02 23:38:56 +02:00
parent c31cae4631
commit 583b6df58e

View File

@ -2717,7 +2717,7 @@ class APScript(StateMachine):
self.step_end(f" Summary of {doc_name} - Processing chunk : {i+1}/{len(chunks)}") self.step_end(f" Summary of {doc_name} - Processing chunk : {i+1}/{len(chunks)}")
return "\n".join(summeries) return "\n".join(summeries)
def build_prompt_from_context_details(self, context_details:dict, custom_entries=""): def build_prompt_from_context_details(self, context_details:dict, custom_entries="", suppress=[]):
""" """
Builds a prompt from the provided context details. Builds a prompt from the provided context details.
@ -2737,61 +2737,54 @@ class APScript(StateMachine):
""" """
full_context = [] full_context = []
sacrifice_id = 0 sacrifice_id = 0
if context_details["conditionning"]: if context_details["conditionning"] and "conditionning" not in suppress:
full_context.append( "\n".join([ full_context.append( "\n".join([
context_details["conditionning"] context_details["conditionning"]
])) ]))
sacrifice_id += 1 sacrifice_id += 1
if context_details["documentation"]: if context_details["documentation"] and "documentation" not in suppress:
full_context.append( "\n".join([ full_context.append( "\n".join([
self.system_custom_header("documentation"), self.system_custom_header("documentation"),
context_details["documentation"] context_details["documentation"]
])) ]))
sacrifice_id += 1 sacrifice_id += 1
if context_details["knowledge"]: if context_details["knowledge"] and "knowledge" not in suppress:
full_context.append( "\n".join([ full_context.append( "\n".join([
self.system_custom_header("knowledge"), self.system_custom_header("knowledge"),
context_details["knowledge"] context_details["knowledge"]
])) ]))
sacrifice_id += 1 sacrifice_id += 1
if context_details["user_description"]: if context_details["user_description"] and "user_description" not in suppress:
full_context.append( "\n".join([ full_context.append( "\n".join([
self.system_custom_header("user_description"), self.system_custom_header("user_description"),
context_details["user_description"] context_details["user_description"]
])) ]))
sacrifice_id += 1 sacrifice_id += 1
if context_details["positive_boost"]: if context_details["positive_boost"] and "positive_boost" not in suppress:
full_context.append( "\n".join([ full_context.append( "\n".join([
self.system_custom_header("positive_boost"), self.system_custom_header("positive_boost"),
context_details["positive_boost"] context_details["positive_boost"]
])) ]))
sacrifice_id += 1 sacrifice_id += 1
if context_details["positive_boost"]: if context_details["negative_boost"] and "negative_boost" not in suppress:
full_context.append( "\n".join([
self.system_custom_header("positive_boost"),
context_details["positive_boost"]
]))
sacrifice_id += 1
if context_details["negative_boost"]:
full_context.append( "\n".join([ full_context.append( "\n".join([
self.system_custom_header("negative_boost"), self.system_custom_header("negative_boost"),
context_details["negative_boost"] context_details["negative_boost"]
])) ]))
sacrifice_id += 1 sacrifice_id += 1
if context_details["current_language"]: if context_details["current_language"] and "current_language" not in suppress:
full_context.append( "\n".join([ full_context.append( "\n".join([
self.system_custom_header("current_language"), self.system_custom_header("current_language"),
context_details["current_language"] context_details["current_language"]
])) ]))
sacrifice_id += 1 sacrifice_id += 1
if context_details["fun_mode"]: if context_details["fun_mode"] and "fun_mode" not in suppress:
full_context.append( "\n".join([ full_context.append( "\n".join([
self.system_custom_header("fun_mode"), self.system_custom_header("fun_mode"),
context_details["fun_mode"] context_details["fun_mode"]
@ -2799,13 +2792,13 @@ class APScript(StateMachine):
sacrifice_id += 1 sacrifice_id += 1
if context_details["discussion_messages"]: if context_details["discussion_messages"] and "discussion_messages" not in suppress:
full_context.append( "\n".join([ full_context.append( "\n".join([
self.system_custom_header("discussion_messages"), self.system_custom_header("discussion_messages"),
context_details["discussion_messages"] context_details["discussion_messages"]
])) ]))
if context_details["extra"]: if context_details["extra"] and "extra" not in suppress:
full_context.append( "\n".join([ full_context.append( "\n".join([
context_details["extra"] context_details["extra"]
])) ]))
@ -2815,9 +2808,10 @@ class APScript(StateMachine):
custom_entries custom_entries
])) ]))
full_context.append( "\n".join([ if "ai_prefix" not in suppress:
self.ai_custom_header(context_details["ai_prefix"]) full_context.append( "\n".join([
])) self.ai_custom_header(context_details["ai_prefix"])
]))
return self.build_prompt(full_context, sacrifice_id) return self.build_prompt(full_context, sacrifice_id)