ara-cli 0.1.9.63__py3-none-any.whl → 0.1.9.64__py3-none-any.whl

ara_cli/ara_command_action.py

@@ -334,11 +334,15 @@ def reconnect_action(args):
     from ara_cli.artefact_models.artefact_load import artefact_from_content
     from ara_cli.artefact_models.artefact_model import Contribution
     from ara_cli.artefact_reader import ArtefactReader
+    from ara_cli.artefact_fuzzy_search import find_closest_rule
 
     classifier = args.classifier
     artefact_name = args.parameter
     parent_classifier = args.parent_classifier
     parent_name = args.parent_name
+    rule = None
+    if args.rule:
+        rule = ' '.join(args.rule)
 
     read_error_message = f"Could not connect {classifier} '{artefact_name}' to {parent_classifier} '{parent_name}'"
 
@@ -367,10 +371,17 @@ def reconnect_action(args):
         content=parent_content
     )
 
-    artefact.contribution = Contribution(
+    contribution = Contribution(
         artefact_name=parent.title,
         classifier=parent.artefact_type
     )
+
+    if rule:
+        closest_rule = find_closest_rule(parent, rule)
+        print("")
+        contribution.rule = closest_rule
+
+    artefact.contribution = contribution
     with open(artefact.file_path, 'w') as file:
         artefact_content = artefact.serialize()
         file.write(artefact_content)

ara_cli/ara_command_parser.py

@@ -184,6 +184,7 @@ def reconnect_parser(subparsers):
     reconnect_parser.add_argument("parameter", help="Filename of artefact").completer = ArtefactCompleter()
     reconnect_parser.add_argument("parent_classifier", choices=classifiers, help="Classifier of the parent artefact type")
     reconnect_parser.add_argument("parent_name", help="Filename of parent artefact").completer = ParentNameCompleter()
+    reconnect_parser.add_argument("-r", "--rule", dest="rule", nargs='+')
 
 
 def read_status_parser(subparsers):

ara_cli/artefact_autofix.py

@@ -83,9 +83,8 @@ def construct_prompt(artefact_type, reason, file_path, artefact_text):
 
     prompt = (
         f"Correct the following {artefact_type} artefact to fix the issue: {reason}. "
-        "Provide the complete, corrected artefact. Do not reformulate the artefact, "
+        "Provide the corrected artefact. Do not reformulate the artefact, "
         "just fix the pydantic model errors, use correct grammar. "
-        "Do not remove comments. "
         "You should follow the name of the file "
         f"from its path {file_path} for naming the arteafact's title. "
         "You are not allowed to use file extention in the artefact title. "
@@ -114,7 +113,7 @@ def run_agent(prompt, artefact_class):
     from pydantic_ai import Agent
     # gpt-4o
     # anthropic:claude-3-7-sonnet-20250219
-    agent = Agent(model="gpt-4o",
+    agent = Agent(model="anthropic:claude-3-7-sonnet-20250219",
                   result_type=artefact_class, instrument=True)
     result = agent.run_sync(prompt)
     return result.data

ara_cli/artefact_fuzzy_search.py

@@ -1,9 +1,10 @@
 import difflib
+from textwrap import indent
 from typing import Optional
 
 
-def suggest_close_names(artefact_name: str, all_artefact_names: list[str], message: str):
-    closest_matches = difflib.get_close_matches(artefact_name, all_artefact_names, cutoff=0.5)
+def suggest_close_names(artefact_name: str, all_artefact_names: list[str], message: str, cutoff=0.5):
+    closest_matches = difflib.get_close_matches(artefact_name, all_artefact_names, cutoff=cutoff)
     print(message)
     if not closest_matches:
         return
@@ -38,3 +39,21 @@ def find_closest_name_match(artefact_name: str, all_artefact_names: list[str]) -
         return None
     closest_match = closest_matches[0]
     return closest_match
+
+
+def find_closest_rule(parent_artefact, rule):
+    parent_classifier = parent_artefact.artefact_type.value
+    parent_title = parent_artefact.title
+    if not hasattr(parent_artefact, 'rules'):
+        raise TypeError(f"{parent_classifier.capitalize()} artefact '{parent_title}' can not possess rules. Only userstories and epics have rules.")
+    rules = parent_artefact.rules
+    if rule in rules:
+        return rule
+    print(f"Rule '{rule}' does not match existing rules in {parent_classifier} artefact '{parent_title}'. Attempting to find closest match among existing rules.")
+    closest_matches = difflib.get_close_matches(rule, rules, cutoff=0.5)
+    rules_list_string = indent('\n'.join(rules), prefix='\t- ')
+    if not closest_matches:
+        raise ValueError(f"Can not determine a match for rule '{rule}' in {parent_classifier} artefact '{parent_title}'. Found rules:\n{rules_list_string}")
+    closest_match = closest_matches[0]
+    print(f"Found closest matching rule of '{closest_match}'")
+    return closest_match

ara_cli/artefact_models/feature_artefact_model.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel, field_validator, model_validator, Field
-from typing import List, Dict, Tuple, Union
+from typing import List, Dict, Tuple, Union, Optional
 from ara_cli.artefact_models.artefact_model import Artefact, ArtefactType, Intent
 import re
 
@@ -98,6 +98,35 @@ class Example(BaseModel):
         return cls(values=values)
 
 
+class Background(BaseModel):
+    steps: List[str] = Field(
+        description="A list of Gherkin 'Given' type steps that describe what the background does."
+    )
+
+    @field_validator('steps', mode='before')
+    def validate_steps(cls, v: List[str]) -> List[str]:
+        """Ensure steps are non-empty and stripped."""
+        steps = [step.strip() for step in v if step.strip()]
+        if not steps:
+            raise ValueError("steps list must not be empty")
+        return steps
+
+    @classmethod
+    def from_lines(cls, lines: List[str], start_idx: int) -> Tuple['Background', int]:
+        """Parse a Background from a list of lines starting at start_idx."""
+        if not lines[start_idx].startswith('Background:'):
+            raise ValueError("Expected 'Background:' at start index")
+
+        steps = []
+        idx = start_idx + 1
+        while idx < len(lines) and not lines[idx].startswith('Background:'):
+            step = lines[idx].strip()
+            if step:
+                steps.append(step)
+            idx += 1
+        return cls(steps=steps), idx
+
+
 class Scenario(BaseModel):
     title: str = Field(
         description="The name of the scenario, giving a short summary of the test case. It comes from the 'Scenario:' line in the feature file."
@@ -213,6 +242,8 @@ class FeatureArtefact(Artefact):
     artefact_type: ArtefactType = ArtefactType.feature
     intent: FeatureIntent
     scenarios: List[Union[Scenario, ScenarioOutline]] = Field(default=None)
+    background: Optional[Background] = Field(
+        default=None, description="Highly optional background Gherkin steps for Feature Artefacts. This steps apply for all scenarios and scenario outlines in this feature file.")
 
     @field_validator('artefact_type')
     def validate_artefact_type(cls, v):
@@ -229,6 +260,16 @@ class FeatureArtefact(Artefact):
     def _artefact_type(cls) -> ArtefactType:
         return ArtefactType.feature
 
+    def _serialize_background(self) -> str:
+        """Helper method to dispatch background serialization."""
+        if not self.background:
+            return ""
+        lines = []
+        lines.append("  Background:")
+        for step in self.background.steps:
+            lines.append(f"    {step}")
+        return "\n".join(lines)
+
     def _serialize_scenario(self, scenario: Union[Scenario, ScenarioOutline]) -> str:
         """Helper method to dispatch scenario serialization."""
         if isinstance(scenario, Scenario):
@@ -309,6 +350,10 @@ class FeatureArtefact(Artefact):
         lines.append(description)
         lines.append("")
 
+        if self.background:
+            lines.append(self._serialize_background())
+            lines.append("")
+
         if self.scenarios:
             for scenario in self.scenarios:
                 lines.append(self._serialize_scenario(scenario))
@@ -321,9 +366,11 @@ class FeatureArtefact(Artefact):
         fields = super()._parse_common_fields(text)
 
         intent = FeatureIntent.deserialize(text)
+        background = cls.deserialize_background(text)
         scenarios = cls.deserialize_scenarios(text)
 
         fields['scenarios'] = scenarios
+        fields['background'] = background
         fields['intent'] = intent
 
         return cls(**fields)
@@ -348,3 +395,19 @@ class FeatureArtefact(Artefact):
             else:
                 idx += 1
         return scenarios
+
+    @classmethod
+    def deserialize_background(cls, text):
+        lines = [line.strip()
+                 for line in text.strip().splitlines() if line.strip()]
+
+        background = None
+        idx = 0
+        while idx < len(lines):
+            line = lines[idx].strip()
+            if line.startswith('Background:'):
+                background, next_idx = Background.from_lines(lines, idx)
+                break
+            else:
+                idx += 1
+        return background

ara_cli/prompt_extractor.py

@@ -184,10 +184,14 @@ def handle_existing_file(filename, block_content):
         create_file_if_not_exist(filename, block_content)
     else:
         print(f"File {filename} exists, creating modification prompt")
-        prompt = create_prompt_for_file_modification(block_content, filename)
+        prompt_text = create_prompt_for_file_modification(block_content, filename)
+        messages = [{"role": "user", "content": prompt_text}]
         response = ""
-        for chunk in send_prompt(prompt):
-            response += chunk.content
+
+        for chunk in send_prompt(messages):
+            content = chunk.choices[0].delta.content
+            if content:
+                response += content
         modify_and_save_file(response, filename)
 
 

ara_cli/prompt_handler.py

@@ -355,8 +355,12 @@ def prepend_system_prompt(message_list):
 
 
 def append_images_to_message(message, image_data_list):
-    message_content = message["content"]
-    message["content"] = message_content + image_data_list
+    if not image_data_list:
+        return message
+    new_content_list = [{"type": "text", "text": message}]
+    new_content_list.extend(image_data_list)
+    
+    message["content"] = new_content_list
 
     return message
 
@@ -376,14 +380,23 @@ def create_and_send_custom_prompt(classifier, parameter):
     append_headings(classifier, parameter, "prompt")
     write_prompt_result(classifier, parameter, prompt)
 
-    message_list = append_images_to_message(combined_content_markdown, image_data_list)
-    append_headings(classifier, parameter, "result")
+    base_message = {
+        "role": "user",
+        "content": combined_content_markdown
+    }
+
+    final_message = append_images_to_message(base_message, image_data_list)
 
+    message_list_to_send = [final_message]
+
+    append_headings(classifier, parameter, "result")
     artefact_data_path = f"ara/{sub_directory}/{parameter}.data/{classifier}.prompt_log.md"
     with open(artefact_data_path, 'a') as file:
-        for chunk in send_prompt(message_list):
-            file.write(chunk.content)
-            file.flush()
+        for chunk in send_prompt(message_list_to_send):
+            content = chunk.choices[0].delta.content
+            if content:
+                file.write(content)
+                file.flush()
     # write_prompt_result(classifier, parameter, response)
 
 

ara_cli/version.py

@@ -1,2 +1,2 @@
 # version.py
-__version__ = "0.1.9.63"   # fith parameter like .0 for local install test purposes only. official numbers should be 4 digit numbers
+__version__ = "0.1.9.64"   # fith parameter like .0 for local install test purposes only. official numbers should be 4 digit numbers

{ara_cli-0.1.9.63.dist-info → ara_cli-0.1.9.64.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ara_cli
-Version: 0.1.9.63
+Version: 0.1.9.64
 Requires-Dist: litellm
 Requires-Dist: llama-index
 Requires-Dist: llama-index-llms-openai

{ara_cli-0.1.9.63.dist-info → ara_cli-0.1.9.64.dist-info}/RECORD

@@ -1,12 +1,12 @@
 ara_cli/__init__.py,sha256=0zl7IegxTid26EBGLav_fXZ4CCIV3H5TfAoFQiOHjvg,148
 ara_cli/__main__.py,sha256=Z6XYWRLceIoZPvfC-X9EXouSZdtFOOe84kKVWJGA4r4,1861
-ara_cli/ara_command_action.py,sha256=SfWcXsLjaUdOpME2qJJ0pcCSXKY5M-zbSyj0W3u_8h4,19709
-ara_cli/ara_command_parser.py,sha256=aRxqKr0-pAuI45IuM-H4-ouc101h1_xTuEVJHXocPV0,17182
+ara_cli/ara_command_action.py,sha256=5Q9dwS4urZCkD52trqTghnYViBLvOBqwcG3oDYBNgHg,20003
+ara_cli/ara_command_parser.py,sha256=iG0IHL3So96viB7_lyetj_wbOeCLlToUpRY9HvBNY8w,17256
 ara_cli/ara_config.py,sha256=_Arkr-b9XnrNHbBlFKb9tAo3OmdP4ZZiWvbY9m6Sbo0,4178
-ara_cli/artefact_autofix.py,sha256=kkmf9hhvxqYicAm4kUDxFD1LoTr8KSPI5l12KRoZvqw,4900
+ara_cli/artefact_autofix.py,sha256=T0TJ6vBxveMoOKm6c22c6D-iMuBHlCksvNLil_O6xaM,4885
 ara_cli/artefact_creator.py,sha256=mkxKHkVIK2GdmUrKHAjKvhq66eg21S3x_cvK1ZA9DPw,5964
 ara_cli/artefact_deleter.py,sha256=Co4wwCH3yW8H9NrOq7_2p5571EeHr0TsfE-H8KqoOfY,1900
-ara_cli/artefact_fuzzy_search.py,sha256=XAvoiRafd1u21uKbX5-bow7hdq7uiLLy1KtxHNAFbCk,1337
+ara_cli/artefact_fuzzy_search.py,sha256=TD96fV5FoyJzLClXiHHlysPruK4t7yAgQ_kPnQqyMRo,2412
 ara_cli/artefact_link_updater.py,sha256=itMS_Z64jE8bBly9WA01z8PqkBeNW6ntTO7ryMeCTRg,3703
 ara_cli/artefact_lister.py,sha256=jhk4n4eqp7hDIq07q43QzS7-36BM3OfZ4EABxCeOGcw,4764
 ara_cli/artefact_reader.py,sha256=qNaMPWShmWtDU5LLdh9efFB27djI4NAoq6zEFwdTd38,6983
@@ -24,14 +24,14 @@ ara_cli/filename_validator.py,sha256=Aw9PL8d5-Ymhp3EY6lDrUBk3cudaNqo1Uw5RzPpI1jA
 ara_cli/list_filter.py,sha256=Not17hIngI37gZsLtIKxopB-BmyWoOGlBzSqBwh-Zpc,5273
 ara_cli/output_suppressor.py,sha256=ZByUwLH2DxOb-eJ31KQbtIziBKdykoyxvwxZ0tSammA,371
 ara_cli/prompt_chat.py,sha256=kd_OINDQFit6jN04bb7mzgY259JBbRaTaNp9F-webkc,1346
-ara_cli/prompt_extractor.py,sha256=gidrCI8wTLfPL0ktqiXyPeGdQEB0S0sZegSOiF1Nn0Y,7358
-ara_cli/prompt_handler.py,sha256=PzHoIPTAWtRleOMtprhyYlFfo59S5T_kzHHkrwL-cNU,17155
+ara_cli/prompt_extractor.py,sha256=a8LwPj6U8sG_v3SqDXQyPvDZQds4kHnYSO8eGissYJA,7503
+ara_cli/prompt_handler.py,sha256=1gz2VkK7zl0RAzit1UmxplNNd2Zu6J_7eHocgKiw_hQ,17473
 ara_cli/prompt_rag.py,sha256=vmlt4-rSboWibwgO_KUF79TK99YXT5KXjmbD9FeWdZY,7449
 ara_cli/run_file_lister.py,sha256=XbrrDTJXp1LFGx9Lv91SNsEHZPP-PyEMBF_P4btjbDA,2360
 ara_cli/tag_extractor.py,sha256=IrjX8R8pdas3_qWuDsXXnwCVDD0uUrgqjzfqTD81t3I,2173
 ara_cli/template_manager.py,sha256=YXPj2jGNDb-diIHFEK_vGJ-ZucodnXSGAPofKTnOofI,6633
 ara_cli/update_config_prompt.py,sha256=PZgNIN3dTw6p80GyX8Sp5apkAhSoykwnkEbHo3IOkUo,4571
-ara_cli/version.py,sha256=BPYHhB5DXV3V9HaoG9jWNjJoI6MwS8Abg07dp1EjppU,146
+ara_cli/version.py,sha256=rBCA0K3GrvOvnsjnnIVwI8EKzmt0b2Eh7phC0VgDNIM,146
 ara_cli/artefact_models/artefact_load.py,sha256=dNcwZDW2Dk0bts9YnPZ0ESmWD2NbsLIvl4Z-qQeGmTQ,401
 ara_cli/artefact_models/artefact_mapping.py,sha256=8aD0spBjkJ8toMAmFawc6UTUxB6-tEEViZXv2I-r88Q,1874
 ara_cli/artefact_models/artefact_model.py,sha256=W_jxafOd7d-1SxvB2G8dKG7_rT5t-EZo1-PnvlPHqso,14915
@@ -40,7 +40,7 @@ ara_cli/artefact_models/businessgoal_artefact_model.py,sha256=jqYFMXjWle0YW9RvcF
 ara_cli/artefact_models/capability_artefact_model.py,sha256=SZqHx4O2mj4urn77Stnj4_Jxtlq3-LgBBU9SMkByppI,3079
 ara_cli/artefact_models/epic_artefact_model.py,sha256=IadQWs6SWNcLgwvtOQWmYDyV9xLr3WwAsx-YMFan5fA,5765
 ara_cli/artefact_models/example_artefact_model.py,sha256=UXrKbaPotg1jwcrVSdCeo-XH4tTD_-U1e3giaBn5_xg,1384
-ara_cli/artefact_models/feature_artefact_model.py,sha256=TWJ7N0P2Wd8VtpbBPwJoiDnDSbHSRPPjyDfpvA-IqYc,13238
+ara_cli/artefact_models/feature_artefact_model.py,sha256=BeLsq9ZinzIDwss8nurIbxJZQ2qPgyRKRHrRlrl-zHA,15583
 ara_cli/artefact_models/issue_artefact_model.py,sha256=v6CpKnkqiUh6Wch2kkEmyyW49c8ysdy1qz8l1Ft9uJA,2552
 ara_cli/artefact_models/keyfeature_artefact_model.py,sha256=a3MyAiePN9n_GTN6QkTvamdsaorwVUff6w-9CdRZSlo,4243
 ara_cli/artefact_models/serialize_helper.py,sha256=0XCruO70-fyfLfTn7pnt8NrSQe79eYNUAjuQaV8K6_8,586
@@ -150,8 +150,8 @@ ara_cli/tests/test_list_filter.py,sha256=gSRKirTtFuhRS3QlFHqWl89WvCvAdVEnFsCWTYm
 ara_cli/tests/test_tag_extractor.py,sha256=n2xNApbDciqKO3QuaveEWSPXU1PCUa_EhxlZMrukONw,2074
 ara_cli/tests/test_template_manager.py,sha256=bRxka6cxHsCAOvXjfG8MrVO8qSZXhxW01tnph80UtNk,3143
 ara_cli/tests/test_update_config_prompt.py,sha256=vSsLvc18HZdVjVM93qXWVbJt752xTLL6VGjSVCrPufk,6729
-ara_cli-0.1.9.63.dist-info/METADATA,sha256=gxNdNrzLFrDWHMMskXW_kMFj0qBeoxoyFxilCHDsQiQ,415
-ara_cli-0.1.9.63.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ara_cli-0.1.9.63.dist-info/entry_points.txt,sha256=v4h7MzysTgSIDYfEo3oj4Kz_8lzsRa3hq-KJHEcLVX8,45
-ara_cli-0.1.9.63.dist-info/top_level.txt,sha256=zzee_PwFmKqfBi9XgIunP6xy2S4TIt593CLLxenNaAE,8
-ara_cli-0.1.9.63.dist-info/RECORD,,
+ara_cli-0.1.9.64.dist-info/METADATA,sha256=hMa8LLYlDMGNBHL5R7CwM1QamWGbbsMe7hTbajvHQ1I,415
+ara_cli-0.1.9.64.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ara_cli-0.1.9.64.dist-info/entry_points.txt,sha256=v4h7MzysTgSIDYfEo3oj4Kz_8lzsRa3hq-KJHEcLVX8,45
+ara_cli-0.1.9.64.dist-info/top_level.txt,sha256=zzee_PwFmKqfBi9XgIunP6xy2S4TIt593CLLxenNaAE,8
+ara_cli-0.1.9.64.dist-info/RECORD,,