Browse Source

fixed dry_run not working in PersonApp (#357)

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

+ 8 - 8
embedchain/apps/PersonApp.py

@@ -29,19 +29,19 @@ class PersonApp(EmbedChainPersonApp, App):
     Extends functionality from EmbedChainPersonApp and App
     """
 
-    def query(self, input_query, config: QueryConfig = None):
+    def query(self, input_query, config: QueryConfig = None, dry_run=False):
         self.template = Template(self.person_prompt + " " + DEFAULT_PROMPT)
         query_config = QueryConfig(
             template=self.template,
         )
-        return super().query(input_query, query_config)
+        return super().query(input_query, query_config, dry_run)
 
-    def chat(self, input_query, config: ChatConfig = None):
+    def chat(self, input_query, config: ChatConfig = None, dry_run=False):
         self.template = Template(self.person_prompt + " " + DEFAULT_PROMPT_WITH_HISTORY)
         chat_config = ChatConfig(
             template=self.template,
         )
-        return super().chat(input_query, chat_config)
+        return super().chat(input_query, chat_config, dry_run)
 
 
 class PersonOpenSourceApp(EmbedChainPersonApp, OpenSourceApp):
@@ -50,16 +50,16 @@ class PersonOpenSourceApp(EmbedChainPersonApp, OpenSourceApp):
     Extends functionality from EmbedChainPersonApp and OpenSourceApp
     """
 
-    def query(self, input_query, config: QueryConfig = None):
+    def query(self, input_query, config: QueryConfig = None, dry_run=False):
         self.template = Template(self.person_prompt + " " + DEFAULT_PROMPT)
         query_config = QueryConfig(
             template=self.template,
         )
-        return super().query(input_query, query_config)
+        return super().query(input_query, query_config, dry_run)
 
-    def chat(self, input_query, config: ChatConfig = None):
+    def chat(self, input_query, config: ChatConfig = None, dry_run=False):
         self.template = Template(self.person_prompt + " " + DEFAULT_PROMPT_WITH_HISTORY)
         chat_config = ChatConfig(
             template=self.template,
         )
-        return super().chat(input_query, chat_config)
+        return super().chat(input_query, chat_config, dry_run)