diff --git a/embedchain/apps/PersonApp.py b/embedchain/apps/PersonApp.py index 4fe6c278..3a0378c1 100644 --- a/embedchain/apps/PersonApp.py +++ b/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)