def log_prompt()

in genai-on-vertex-ai/gemini/evals_playbook/utils/evals_playbook.py [0:0]

• 11 lines of code
• 5 McCabe index (conditional complexity)

    def log_prompt(self, prompt):
        try:
            if isinstance(prompt, self.Prompt):
                prompt = prompt.__dict__
                if "_sa_instance_state" in prompt: prompt.pop("_sa_instance_state") 
            if not isinstance(prompt, dict):
                raise Exception(f"Invalid task object. Expected: `dict`. Actual: {type(prompt)}")
            self._upsert("prompts", prompt)
        except Exception as e:
            print(f"Failed to log prompt due to following error.")
            raise e