From b14dc087b114281aef40b89aef4d6c3ace109b58 Mon Sep 17 00:00:00 2001 From: Rushil Patel Date: Tue, 19 Aug 2025 11:30:52 -0700 Subject: [PATCH] fix: safe import for create claude session method in startup hook --- .../cli/commands/claude/config/claude_session_hook.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/codegen/cli/commands/claude/config/claude_session_hook.py b/src/codegen/cli/commands/claude/config/claude_session_hook.py index 374ef0462..2f14f1542 100755 --- a/src/codegen/cli/commands/claude/config/claude_session_hook.py +++ b/src/codegen/cli/commands/claude/config/claude_session_hook.py @@ -11,7 +11,6 @@ import sys from pathlib import Path -from codegen.cli.commands.claude.claude_session_api import create_claude_session from codegen.cli.utils.org import resolve_org_id # Add the codegen CLI to the path so we can import from it @@ -19,6 +18,11 @@ codegen_cli_dir = script_dir.parent.parent.parent sys.path.insert(0, str(codegen_cli_dir)) +try: + from codegen.cli.commands.claude.claude_session_api import create_claude_session +except ImportError: + create_claude_session = None + def main(): """Main hook function called by Claude Code."""