def init_chat()

in agents/DebugSQLAgent.py [0:0]

• 21 lines of code
• 4 McCabe index (conditional complexity)

    def init_chat(self,source_type,user_grouping, tables_schema,columns_schema,similar_sql="-No examples provided..-"):

        if f'usecase_{source_type}_{user_grouping}' in PROMPTS:
            usecase_context = PROMPTS[f'usecase_{source_type}_{user_grouping}']
        else:
            usecase_context = "No extra context for the usecase is provided"
            
        context_prompt = PROMPTS[f'debugsql_{source_type}']

        context_prompt = format_prompt(context_prompt,
                                       usecase_context = usecase_context,
                                       similar_sql=similar_sql, 
                                       tables_schema=tables_schema, 
                                       columns_schema = columns_schema)

        # print(f"Prompt to Debug SQL after formatting: \n{context_prompt}")
        
        if self.model_id == 'codechat-bison-32k':
            with telemetry.tool_context_manager('opendataqna-debugsql-v2'):

                chat_session = self.model.start_chat(context=context_prompt)
        elif 'gemini' in self.model_id:
            with telemetry.tool_context_manager('opendataqna-debugsql-v2'):

                chat_session = self.model.start_chat(response_validation=False)
                chat_session.send_message(context_prompt)
        else:
            raise ValueError('Invalid Chat Model Specified')
        
        return chat_session