Merge pull request #83 from Grigorij-Dudnik/dev

new release - bugs chroma solved

Author: Grigorij-Dudnik

manager.py

@@ -159,6 +159,7 @@ def setup_workflow(self):
         return manager_workflow.compile()
 
     def run(self):
+        """Start the Manager agent to plan and manage project tasks interactively."""
         print_formatted("😀 Hello! I'm Manager agent. Let's plan your project together!", color="green")
 
         messages = get_manager_messages(self.saved_messages_path)

non_src/tests/manual_tests/executor_scenario1.py

@@ -12,8 +12,7 @@
 load_dotenv(find_dotenv())
 
 folder_with_project_files = repo_directory.joinpath(
-    "non_src/tests/manual_tests/projects_files", "executoro"
-                                                 "_scenario_1_files"
+    "non_src/tests/manual_tests/projects_files", "executor_scenario_1_files"
 )
 tmp_folder = pathlib.Path(__file__).parent.resolve().joinpath("sandbox_work_dir")
 setup_work_dir(manual_tests_folder=tmp_folder, test_files_dir=folder_with_project_files)

single_task_coder.py

@@ -31,6 +31,7 @@
 
 
 def run_clean_coder_pipeline(task: str, work_dir: str, task_id: str=None):
+    """Execute the complete Clean Coder pipeline including research, planning, execution, and debugging phases."""
     researcher = Researcher(task_id=task_id)
     files, image_paths = researcher.research_task(task)
 

src/prompts/manager_system.prompt

@@ -8,6 +8,7 @@ Avoid creating flaky tasks, where it's unclear how to do task and if it is neede
 
 - Never make up information when describing task. Ask human or make file research to get more info instead.
 - Never write code inside of task description.
+- If you don't know how exactly task should be done - again, do not make it up. Instead, ask programmer in description to figure it out.
 
 
 Here is description of changes in the project you work on:

src/tools/rag/index_file_descriptions.py

@@ -207,6 +207,9 @@ def upload_descriptions_to_vdb():
                 # Clear the batch lists
                 docs = []
                 ids = []
+    # avoid upserting if no docs here (empty folder)
+    if not docs:
+        return
     # upsert remaining docs
     collection.upsert(documents=docs, ids=ids)
 
@@ -232,6 +235,10 @@ def upsert_file_list(file_list):
             file_path = Path(file_path)
             docs.append(content)
             ids.append(file_path.name.replace("=", "/").removesuffix(".txt"))
+
+        # avoid upserting if no docs here (no files changed)
+        if not docs:
+            return
     collection.upsert(documents=docs, ids=ids)
     print_formatted("Re-indexing of modified files completed.", color="green")
 

src/tools/rag/retrieval.py

@@ -1,5 +1,6 @@
 import os
 import chromadb
+from chromadb.errors import NotFoundError
 from pathlib import Path
 from dotenv import load_dotenv, find_dotenv
 from src.utilities.llms import init_llms_mini
@@ -42,7 +43,7 @@ def get_collection():
     # )
     try:
         return chroma_client.get_collection(name=collection_name)  # , embedding_function=embedding_function)
-    except ValueError:
+    except NotFoundError:
         return False
 
 

src/utilities/util_functions.py

@@ -197,6 +197,7 @@ def invoke_tool_native(tool_call, tools):
 
 
 def exchange_file_contents(state, files, work_dir):
+    """Update state messages by replacing old file contents with current file contents."""
     # Remove old one
     state["messages"] = [msg for msg in state["messages"] if not hasattr(msg, "contains_file_contents")]
     # Add new file contents
@@ -225,6 +226,7 @@ def bad_tool_call_looped(state):
 
 
 def create_frontend_feedback_story():
+    """Create and open a frontend feedback story template file for user configuration."""
     frontend_feedback_story_path = os.path.join(Work.dir(), ".clean_coder", "frontend_feedback_story.txt")
     if not os.path.exists(frontend_feedback_story_path):
         with open(frontend_feedback_story_path, "w") as file: