Improved prompts to encourage better investigative flow

Improved abstraction
Fixed failing tests after refactor
This commit is contained in:
Fahad
2025-06-19 11:18:03 +04:00
parent 43485dadd6
commit b8c8e6f91e
15 changed files with 167 additions and 489 deletions

View File

@@ -164,6 +164,8 @@ class ConversationBaseTest(BaseSimulatorTest):
continuation_id = self._extract_continuation_id_from_response(response_text)
self.logger.debug(f"Tool '{tool_name}' completed successfully in-process")
if self.verbose and response_text:
self.logger.debug(f"Response preview: {response_text[:500]}...")
return response_text, continuation_id
except Exception as e:
@@ -193,6 +195,21 @@ class ConversationBaseTest(BaseSimulatorTest):
if follow_up and "continuation_id" in follow_up:
return follow_up["continuation_id"]
# Special case: files_required_to_continue may have nested content
if response_data.get("status") == "files_required_to_continue":
content = response_data.get("content", "")
if isinstance(content, str):
try:
# Try to parse nested JSON
nested_data = json.loads(content)
if isinstance(nested_data, dict):
# Check for continuation in nested data
follow_up = nested_data.get("follow_up_request", {})
if follow_up and "continuation_id" in follow_up:
return follow_up["continuation_id"]
except json.JSONDecodeError:
pass
return None
except (json.JSONDecodeError, AttributeError):