Add pdf、docx parser and image、video to text function for knowledge

2023-12-05 18:08:32 +08:00
parent 66e407add5
commit 5f451107c9
3 changed files with 164 additions and 17 deletions
@@ -20,7 +20,7 @@ class TextSummaryAgent(CustomAIAgent):
        chunks = split_text(msg.body, separators=["\n\n", "\n"], chunk_size=4000, chunk_overlap=200, length_function=len)

        prompt = AgentPrompt()
-        prompt.system_message = "Your job is to generate a summary based on the input."
+        prompt.system_message = {"role":"system","content":"Your job is to generate a summary based on the input."}
        if len(chunks) == 1:
            prompt.append(AgentPrompt(chunks[0]))
            resp = await self.do_llm_complection(prompt)