ziya 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

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)

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()

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)

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"

{ziya-0.1.1.dist-info → ziya-0.1.3.dist-info}/METADATA

@@ -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.dist-info → ziya-0.1.3.dist-info}/RECORD

@@ -1,22 +1,22 @@
 app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 app/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-app/agents/agent.py,sha256=k0e0GMKFfU6-3e-GyAX95R9iV3sWwvLknL1gxd3TO88,4181
+app/agents/agent.py,sha256=A1GCNwPXoPY0wg9ukHyox8WuJu-i1o0ArZMpOF9I_kQ,4266
 app/agents/prompts.py,sha256=O54rH2KupyY7ZSjPEvm3j9DSGjdL3jJOz8F9PTcePQQ,1674
-app/main.py,sha256=blgAdLjoMa2HRqlr17lwbZ3hQYz2jUZJwu18KLI9FiQ,1359
+app/main.py,sha256=2-5fCFw2pMYNzyX25yxh9JnGzfV9Vvega8J59RT02MM,1704
 app/server.py,sha256=YTEHYUsFQhgkGRHZ_y7K3Ju3SmUIbVxLSdgrgi1BuEU,676
 app/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-app/utils/directory_util.py,sha256=XOImpzqVa2NrIAPl054VN2WC4kk1QUIE4COEL062ULw,2132
+app/utils/directory_util.py,sha256=qx-LCHLZgFSz8WJTJcAkRnfxtg1eQgRX55dMJ2bTI3I,2278
 app/utils/gitignore_parser.py,sha256=ksAAvXr8_GFh94cvOdkLmES5k3xxNxW2_tQuQtialIA,7496
 app/utils/logging_utils.py,sha256=8JEcc1t7L-a0G4HLmM8zFydNNSOd5l2-gkTxviLQUns,280
 app/utils/print_tree_util.py,sha256=O4Rb-GS8UODxfH7Z6bZGsr-opG6QLmvdaU1im5kh4IA,1419
-pyproject.toml,sha256=mQavdT_weig3WJLIvmy0QpC9aUtDwqT__8bNolF00Sc,716
+pyproject.toml,sha256=1MA83_N7UL7fG8SEVuiHIJdsrG6pd5UFr8SOR6KOSrQ,716
 static/app.js,sha256=qz2F0e5LxuVFI3MXWvcON9l1cN0J_ODdQOwbgT3ZqWA,3622
 static/favicon.ico,sha256=HgB8xAZdDHFK2lODUsp2H_Dds_i14pnpydx7NEJrNrU,15086
 static/sendPayload.js,sha256=VnypSWF7t8bv83026LGU3GMyr9tMrp05hmWATcbCG4k,1956
 static/ziya.css,sha256=OaieSXw_7LOk0WNNi-1mPsAaS3rOxhHycKFgFDoiZB0,1395
 templates/index.html,sha256=8auy9lGLdcDzBkzpjYB1ygrAVbXpo0rqQ8p1FgMEeck,800
-ziya-0.1.1.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
-ziya-0.1.1.dist-info/METADATA,sha256=cBLlieO7ab7odRbHumQnZOqKoSuslCWDEBBY0Szkta0,2563
-ziya-0.1.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-ziya-0.1.1.dist-info/entry_points.txt,sha256=1HspxMqCYRli3SqM6CqmT6gsW_G14G4ab3sHlrYUCAA,38
-ziya-0.1.1.dist-info/RECORD,,
+ziya-0.1.3.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
+ziya-0.1.3.dist-info/METADATA,sha256=d0kWH3BeHMjSyM8VhoUfZSy0lg8q1ks5hE5hg4w8Ga8,2921
+ziya-0.1.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+ziya-0.1.3.dist-info/entry_points.txt,sha256=1HspxMqCYRli3SqM6CqmT6gsW_G14G4ab3sHlrYUCAA,38
+ziya-0.1.3.dist-info/RECORD,,