ziya 0.1.0__tar.gz → 0.1.2__tar.gz

{ziya-0.1.0 → ziya-0.1.2}/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2023 LangChain, Inc.
+Copyright (c) 2024 ziya-ai
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

{ziya-0.1.0 → ziya-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ziya
-Version: 0.1.0
+Version: 0.1.2
 Summary: 
 Author: Vishnu Krishnaprasad
 Author-email: vishnukool@gmail.com
@@ -24,6 +24,10 @@ Description-Content-Type: text/markdown
 
 # Ziya
 
+## Documentation
+See the [GitHub Repository](https://github.com/ziya-ai/ziya)
+
+## Overview
 Ziya is a code assist tool for AWS Bedrock models. It can read your entire codebase and answer questions.
 
 The current version only performs read operations. However, future versions will be able to:

{ziya-0.1.0 → ziya-0.1.2}/README.md

@@ -1,5 +1,9 @@
 # Ziya
 
+## Documentation
+See the [GitHub Repository](https://github.com/ziya-ai/ziya)
+
+## Overview
 Ziya is a code assist tool for AWS Bedrock models. It can read your entire codebase and answer questions.
 
 The current version only performs read operations. However, future versions will be able to:

{ziya-0.1.0 → ziya-0.1.2}/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.0 → ziya-0.1.2}/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-dirs", default=[], type=lambda x: x.split(','),
+                        help="List of directories to include (e.g., --include 'src,static')")
     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_dirs])
+    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.0 → ziya-0.1.2}/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.0 → ziya-0.1.2}/pyproject.toml

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