test logging

This commit is contained in:
Daniel O'Connell
2023-09-29 22:30:54 +02:00
parent 3038bd131e
commit 4e0956c85b
2 changed files with 154 additions and 1 deletions
+1 -1
View File
@@ -41,7 +41,7 @@ class ChatLogger(Logger):
def is_debug(self):
return self.isEnabledFor(DEBUG)
def interaction(self, session_id, query, response, history, prompt, blocks):
def interaction(self, session_id: str, query: str, response: str, history, prompt, blocks):
prompt = [i for i in prompt if i.get('role') == 'system']
prompt = prompt[0].get('content') if prompt else None