From 01f9e5d3315568afeca8c57905a9461e51f5d138 Mon Sep 17 00:00:00 2001 From: Yolanda Robla Date: Mon, 3 Feb 2025 10:29:20 +0100 Subject: [PATCH] fix: be more flexible when interpreting codegate version Different combinations of open interpreter send the user message in different formats, accomodate to it --- src/codegate/pipeline/cli/cli.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/codegate/pipeline/cli/cli.py b/src/codegate/pipeline/cli/cli.py index 32803e13..d9fdd3b9 100644 --- a/src/codegate/pipeline/cli/cli.py +++ b/src/codegate/pipeline/cli/cli.py @@ -80,7 +80,11 @@ def _get_cli_from_open_interpreter(last_user_message_str: str) -> Optional[re.Ma codegate_regex = re.compile(r"^codegate\s*(.*?)\s*$", re.IGNORECASE) match = codegate_regex.match(last_user_block) return match - return None + else: + # try to just get from the regex + codegate_regex = re.compile(r"^codegate\s*(.*?)\s*$", re.IGNORECASE) + match = codegate_regex.match(last_user_message_str) + return match class CodegateCli(PipelineStep):