角色扮演冲突问题

Author: glide-the

src/dreamsboard/dreamsboard/dreams/builder_cosplay_code/base.py

@@ -46,6 +46,7 @@ def __init__(self,
                  kor_dreams_guidance_chain: LLMChain,
                  kor_dreams_personality_chain: LLMChain,
                  ner_dreams_personality_chain: LLMChain,
+                 user_id: str = None,
                  ):
         """
 
@@ -59,15 +60,16 @@ def __init__(self,
         self.kor_dreams_guidance_chain = kor_dreams_guidance_chain
         self.kor_dreams_personality_chain = kor_dreams_personality_chain
         self.ner_dreams_personality_chain = ner_dreams_personality_chain
-        self.storage_context = None
+        self.user_id = user_id
 
     @classmethod
     def form_builder(cls,
                      llm: BaseLanguageModel,
                      builder: StructuredStoryboardCSVBuilder,
                      dreams_guidance_context: str,
                      dreams_personality_context: str,
-                     guidance_llm: BaseLanguageModel = None) -> StructuredDreamsStoryboard:
+                     guidance_llm: BaseLanguageModel = None,
+                     user_id: str = None,) -> StructuredDreamsStoryboard:
         kor_dreams_guidance_chain = KorLoader.form_kor_dreams_guidance_builder(
             llm=llm if guidance_llm is None else guidance_llm)
         kor_dreams_personality_chain = KorLoader.form_kor_dreams_personality_builder(llm=llm)
@@ -78,7 +80,8 @@ def form_builder(cls,
                    dreams_personality_context=dreams_personality_context,
                    kor_dreams_guidance_chain=kor_dreams_guidance_chain,
                    kor_dreams_personality_chain=kor_dreams_personality_chain,
-                   ner_dreams_personality_chain=ner_dreams_personality_chain)
+                   ner_dreams_personality_chain=ner_dreams_personality_chain,
+                   user_id=user_id)
 
     def kor_dreams_guidance_context(self) -> List[DreamsStepInfo]:
         """
@@ -147,8 +150,11 @@ def loader_cosplay_builder(self, engine_template_render_data: dict = {}) -> Code
 
         # 创建一个字典，用于按照story_board组织内容和角色
         storyboard_dict = self.builder.build_dict()
-        # 获取第一个story_board_role属性的值
-        cosplay_role = list(storyboard_dict.values())[0]['story_board_role'][0]
+        if self.user_id is None:
+            # 获取第一个story_board_role属性的值
+            cosplay_role = list(storyboard_dict.values())[0]['story_board_role'][0]
+        else:
+            cosplay_role = self.user_id
 
         guidance_questions = self.kor_dreams_guidance_context()
         try:

src/dreamsboard/tests/test_batch_extract/test_batch_extract.py

@@ -140,7 +140,8 @@ def test_batch_extract() -> None:
                                                                                       builder=builder,
                                                                                       dreams_guidance_context=dreams_guidance_context,
                                                                                       dreams_personality_context=dreams_personality_context,
-                                                                                      guidance_llm=guidance_llm
+                                                                                      guidance_llm=guidance_llm,
+                                                                                      user_id=role
                                                                                       )
                         code_gen_builder = storyboard_executor.loader_cosplay_builder(
                             engine_template_render_data={