|
@@ -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)
|