bugfix: #70 acquire llm output by CustomLLMModel class

Author: lightislost

muagent/codechat/code_analyzer/code_intepreter.py

@@ -32,9 +32,10 @@ def get_intepretation(self, code_list):
         res = {}
         for code in code_list:
             message = CODE_INTERPERT_TEMPLATE.format(code=code)
-            message = [HumanMessage(content=message)]
-            chat_res = chat_model.predict_messages(message)
-            content = chat_res.content
+            # message = [HumanMessage(content=message)]
+            # chat_res = chat_model.predict_messages(message)
+            # content = chat_res.content
+            content = chat_model.predict(message)
             res[code] = content
         return res
 

muagent/codechat/code_search/cypher_generator.py

@@ -54,9 +54,10 @@ def get_cypher(self, query: str):
         content = self.NGQL_GENERATION_PROMPT.format(schema=schema, question=query)
         # logger.info(content)
         ans = ''
-        message = [HumanMessage(content=content)]
-        chat_res = self.model.predict_messages(message)
-        ans = chat_res.content
+        # message = [HumanMessage(content=content)]
+        # chat_res = self.model.predict_messages(message)
+        # ans = chat_res.content
+        self.model.predict(content)
 
         ans = replace_lt_gt(ans)