add knowledge pipeline

2023-09-20 14:38:43 +08:00
parent 3721f4aaa4
commit edbd1c41da
5 changed files with 282 additions and 28 deletions
@@ -1,7 +1,7 @@
 # define a knowledge base class
 import json
 import logging
-from . import AgentPrompt, ComputeKernel
+from . import AgentPrompt, ComputeKernel, AIStorage
 from knowledge import *


@@ -239,9 +239,6 @@ class KnowledgeBase:
        prompt.messages.append({"role": "knowledge", "content": content})    

        return prompt
-                    
-                        
-                    
-                
+