fix bug

bmtools/agent/executor.py

@@ -20,8 +20,8 @@ class AgentExecutorWithTranslation(AgentExecutor):
         except ValueError as e:
             return outputs
         else:
-            if "input" in outputs:
-                outputs = self.translator(outputs)
+            # if "input" in outputs:
+            #     outputs = self.translator(outputs)
             return outputs
 
 class Executor(AgentExecutorWithTranslation):

bmtools/agent/translator.py

@@ -31,24 +31,24 @@ class Translator:
         self.chain = LLMChain(llm=llm, prompt=prompt)
 
     def __call__(self, inputs: Dict[str, str]) -> Dict[str, str]:
-        # if not self.is_create:
-        #     self.create_chain()
-        #     self.is_create = True
-        #
-        # question = inputs["input"]
-        # answer = inputs["output"]
-        #
-        # src_lang = detect_lang(answer)
-        # tgt_lang = detect_lang(question)
-        #
-        # if src_lang != tgt_lang:
-        #     translated_answer = self.chain.run(text=answer, language=tgt_lang)
-        #     outputs = deepcopy(inputs)
-        #     outputs["output"] = translated_answer
-        #     return outputs
-        # else:
-        #     return inputs
-        return inputs
+        if not self.is_create:
+            self.create_chain()
+            self.is_create = True
+
+        question = inputs["input"]
+        answer = inputs["output"]
+
+        src_lang = detect_lang(answer)
+        tgt_lang = detect_lang(question)
+
+        if src_lang != tgt_lang:
+            translated_answer = self.chain.run(text=answer, language=tgt_lang)
+            outputs = deepcopy(inputs)
+            outputs["output"] = translated_answer
+            return outputs
+        else:
+            return inputs
+        # return inputs
 
     def create_openai_model(self, openai_api_key: str, model_name: str) -> OpenAI:
         if openai_api_key is None: