ziya 0.1.1__tar.gz → 0.1.3__tar.gz

{ziya-0.1.1 → ziya-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ziya
-Version: 0.1.1
+Version: 0.1.3
 Summary: 
 Author: Vishnu Krishnaprasad
 Author-email: vishnukool@gmail.com
@@ -77,7 +77,9 @@ ziya
 
 ### Options
 
-`--exclude`: Comma-separated list of files or directories to exclude from the codebase.
+`--exclude`: Comma-separated list of files or directories or file suffix patterns to exclude from the codebase. Eg: `--exclude 'tst,build,*.py'`
+
+`--include`: Comma-separated list of directories to include. By default, it only searches for current directory for code files, but you can specify a list of subset directories under current folder to search instead of the entire folder. Eg: `--include='app,src/mappers'`
 
 `--profile`: AWS profile to use for the Bedrock LLM.
 
@@ -86,5 +88,5 @@ ziya
 `--port`: The port number for frontend app. Default is `6969`.
 
 ```bash
-ziya --exclude='tst,build,*.py' --profile=ziya --model=sonnet --port=8080
+ziya --include='app,src/mappers' --exclude='tst,build,*.py' --profile=ziya --model=sonnet --port=8080
 ```

{ziya-0.1.1 → ziya-0.1.3}/README.md

@@ -53,7 +53,9 @@ ziya
 
 ### Options
 
-`--exclude`: Comma-separated list of files or directories to exclude from the codebase.
+`--exclude`: Comma-separated list of files or directories or file suffix patterns to exclude from the codebase. Eg: `--exclude 'tst,build,*.py'`
+
+`--include`: Comma-separated list of directories to include. By default, it only searches for current directory for code files, but you can specify a list of subset directories under current folder to search instead of the entire folder. Eg: `--include='app,src/mappers'`
 
 `--profile`: AWS profile to use for the Bedrock LLM.
 
@@ -62,5 +64,5 @@ ziya
 `--port`: The port number for frontend app. Default is `6969`.
 
 ```bash
-ziya --exclude='tst,build,*.py' --profile=ziya --model=sonnet --port=8080
+ziya --include='app,src/mappers' --exclude='tst,build,*.py' --profile=ziya --model=sonnet --port=8080
 ```

{ziya-0.1.1 → ziya-0.1.3}/app/agents/agent.py

@@ -47,7 +47,8 @@ def get_combined_document_contents() -> str:
 
     combined_contents: str = ""
     ignored_patterns: List[str] = get_ignored_patterns(user_codebase_dir)
-    all_files: List[str] = get_complete_file_list(user_codebase_dir, ignored_patterns)
+    included_relative_dirs = [pattern.strip() for pattern in os.environ.get("ZIYA_ADDITIONAL_INCLUDE_DIRS", "").split(',')]
+    all_files: List[str] = get_complete_file_list(user_codebase_dir, ignored_patterns, included_relative_dirs)
 
     print_file_tree(all_files)
 
@@ -59,7 +60,6 @@ def get_combined_document_contents() -> str:
                 combined_contents += f"File: {file_path}\n{doc.page_content}\n\n"
             dir_path = os.path.dirname(file_path)
             if dir_path not in seen_dirs:
-                # print_file_tree(dir_path, ignored_patterns)
                 seen_dirs.add(dir_path)
         except Exception as e:
             print(f"Skipping file {file_path} due to error: {e}")
@@ -74,7 +74,6 @@ def get_combined_document_contents() -> str:
     print("--------------------------------------------------------")
     return combined_contents
 
-
 def get_child_paths(directory: str) -> Generator[str, None, None]:
     for entry in os.listdir(directory):
         child_path = os.path.join(directory, entry)

{ziya-0.1.1 → ziya-0.1.3}/app/main.py

@@ -8,6 +8,8 @@ def main():
     parser = argparse.ArgumentParser(description="Run with custom options")
     parser.add_argument("--exclude", default=[], type=lambda x: x.split(','),
                         help="List of files or directories to exclude (e.g., --exclude 'tst,build,*.py')")
+    parser.add_argument("--include", default=[], type=lambda x: x.split(','),
+                        help="List of directories to include (e.g., --include 'src,static'). Only directories for now")
     parser.add_argument("--profile", type=str, default=None,
                         help="AWS profile to use (e.g., --profile ziya)")
     parser.add_argument("--model", type=str, choices=["sonnet", "haiku", "opus"], default="haiku",
@@ -19,6 +21,9 @@ def main():
     additional_excluded_dirs = ','.join([item for item in args.exclude])
     os.environ["ZIYA_ADDITIONAL_EXCLUDE_DIRS"] = additional_excluded_dirs
 
+    additional_included_dirs = ','.join([item for item in args.include])
+    os.environ["ZIYA_ADDITIONAL_INCLUDE_DIRS"] = additional_included_dirs
+
     if args.profile:
         os.environ["ZIYA_AWS_PROFILE"] = args.profile
     if args.model:
@@ -30,4 +35,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

{ziya-0.1.1 → ziya-0.1.3}/app/utils/directory_util.py

@@ -41,17 +41,17 @@ def get_ignored_patterns(directory: str) -> List[Tuple[str, str]]:
     return ignored_patterns
 
 
-def get_complete_file_list(user_codebase_dir: str, ignored_patterns: List[str]) -> List[str]:
+def get_complete_file_list(user_codebase_dir: str, ignored_patterns: List[str], included_relative_dirs: List[str]) -> List[str]:
     should_ignore = parse_gitignore_patterns(ignored_patterns, base_dir=user_codebase_dir)
     file_set: Set[str] = set()
-
-    for root, dirs, files in os.walk(user_codebase_dir):
-        # Filter out ignored directories
-        dirs[:] = [d for d in dirs if not should_ignore(os.path.join(root, d))]
-
-        for file in files:
-            file_path = os.path.join(root, file)
-            if not should_ignore(file_path):
-                file_set.add(file_path)
+    for pattern in included_relative_dirs:
+        for root, dirs, files in os.walk(os.path.normpath(os.path.join(user_codebase_dir, pattern))):
+            # Filter out ignored directories
+            dirs[:] = [d for d in dirs if not should_ignore(os.path.join(root, d))]
+
+            for file in files:
+                file_path = os.path.join(root, file)
+                if not should_ignore(file_path):
+                    file_set.add(file_path)
 
     return list(file_set)

{ziya-0.1.1 → ziya-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ziya"
-version = "0.1.1"
+version = "0.1.3"
 description = ""
 authors = ["Vishnu Krishnaprasad <vishnukool@gmail.com>"]
 readme = "README.md"