Browse Source

fix: fix PersonOpenSourceApp query error (#353)

aaishikdutta 2 years ago
parent
commit
c4797eb121
1 changed files with 2 additions and 0 deletions
  1. 2 0
      embedchain/apps/PersonApp.py

+ 2 - 0
embedchain/apps/PersonApp.py

@@ -51,12 +51,14 @@ class PersonOpenSourceApp(EmbedChainPersonApp, OpenSourceApp):
     """
     """
 
 
     def query(self, input_query, config: QueryConfig = None):
     def query(self, input_query, config: QueryConfig = None):
+        self.template = Template(self.person_prompt + " " + DEFAULT_PROMPT)
         query_config = QueryConfig(
         query_config = QueryConfig(
             template=self.template,
             template=self.template,
         )
         )
         return super().query(input_query, query_config)
         return super().query(input_query, query_config)
 
 
     def chat(self, input_query, config: ChatConfig = None):
     def chat(self, input_query, config: ChatConfig = None):
+        self.template = Template(self.person_prompt + " " + DEFAULT_PROMPT_WITH_HISTORY)
         chat_config = ChatConfig(
         chat_config = ChatConfig(
             template=self.template,
             template=self.template,
         )
         )