diff --git a/apps/onboarding/consumers/knowledge.py b/apps/onboarding/consumers/knowledge.py index 386c43a..9286141 100644 --- a/apps/onboarding/consumers/knowledge.py +++ b/apps/onboarding/consumers/knowledge.py @@ -81,7 +81,7 @@ class OnboardingKnowledgeConsumer(BaseOnboardingConsumer): assistant_message = OnboardingPrompts.KA_HELP_FALLBACK await self.save_page_help(session, str(page_uuid), str(user_message), assistant_message) - await self.log_interaction(session, str(user_message), assistant_message, str(page_uuid), mode, updated_page) + await self.log_interaction(session, str(user_message), assistant_message, str(page_uuid), mode, updated_page, config=config) await self.send_log(LogType.COMPLETED, assistant_message, { 'updated_page': updated_page, @@ -192,16 +192,18 @@ class OnboardingKnowledgeConsumer(BaseOnboardingConsumer): @database_sync_to_async def log_interaction( self, session, user_message: str, assistant_message: str, - page_uuid: str, mode: str, updated_page: bool, + page_uuid: str, mode: str, updated_page: bool, config=None, ): AgentInteractionLog.objects.create( session=session, + agent_config=config, sender_type='user', content=user_message, tool_call_metadata={'action': 'ask_ka', 'page_uuid': page_uuid, 'mode': mode}, ) AgentInteractionLog.objects.create( session=session, + agent_config=config, sender_type='ai', content=assistant_message, tool_call_metadata={