ziya 0.1.40__tar.gz → 0.1.41__tar.gz

{ziya-0.1.40 → ziya-0.1.41}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ziya
-Version: 0.1.40
+Version: 0.1.41
 Summary: 
 Author: Vishnu Krishnaprasad
 Author-email: vishnukool@gmail.com
@@ -83,10 +83,12 @@ ziya
 
 `--profile`: AWS profile to use for the Bedrock LLM.
 
-`--model`: The AWS Bedrock Model to use, one of `sonnet`(default), `sonnet3.5`, `haiku` or `opus`.
+`--model`: The AWS Bedrock Model to use, one of `sonnet3.5`(default), `sonnet`, `haiku` or `opus`.
 
 `--port`: The port number for frontend app. Default is `6969`.
 
+`--max-depth`: Maximum depth for folder structure traversal. Default is `15`.
+
 ```bash
 ziya --exclude='tst,build,*.py' --profile=ziya --model=sonnet3.5 --port=8080
 ```

{ziya-0.1.40 → ziya-0.1.41}/README.md

@@ -57,10 +57,12 @@ ziya
 
 `--profile`: AWS profile to use for the Bedrock LLM.
 
-`--model`: The AWS Bedrock Model to use, one of `sonnet`(default), `sonnet3.5`, `haiku` or `opus`.
+`--model`: The AWS Bedrock Model to use, one of `sonnet3.5`(default), `sonnet`, `haiku` or `opus`.
 
 `--port`: The port number for frontend app. Default is `6969`.
 
+`--max-depth`: Maximum depth for folder structure traversal. Default is `15`.
+
 ```bash
 ziya --exclude='tst,build,*.py' --profile=ziya --model=sonnet3.5 --port=8080
 ```

{ziya-0.1.40 → ziya-0.1.41}/app/agents/agent.py

@@ -30,10 +30,11 @@ else:
     logger.info("No AWS profile specified via --aws-profile flag, using default credentials")
 model_id = {
     "sonnet3.5": "us.anthropic.claude-3-5-sonnet-20240620-v1:0",
+    "sonnet3.5-v2": "us.anthropic.claude-3-5-sonnet-20241022-v2:0",
     "opus": "us.anthropic.claude-3-opus-20240229-v1:0",
     "sonnet": "us.anthropic.claude-3-sonnet-20240229-v1:0",
     "haiku": "us.anthropic.claude-3-haiku-20240307-v1:0",
-}[os.environ.get("ZIYA_AWS_MODEL", "sonnet3.5")]
+}[os.environ.get("ZIYA_AWS_MODEL", "sonnet3.5-v2")]
 logger.info(f"Using Claude Model: {model_id}")
 
 model = ChatBedrock(

{ziya-0.1.40 → ziya-0.1.41}/app/main.py

@@ -17,12 +17,14 @@ def parse_arguments():
                         help="List of files or directories to exclude (e.g., --exclude 'tst,build,*.py')")
     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", "sonnet3.5", "haiku", "opus"], default="sonnet3.5",
+    parser.add_argument("--model", type=str, choices=["sonnet", "sonnet3.5", "sonnet3.5-v2", "haiku", "opus"], default="sonnet3.5-v2",
                         help="AWS Bedrock Model to use (e.g., --model sonnet)")
     parser.add_argument("--port", type=int, default=6969,
                         help="Port number to run Ziya frontend on (e.g., --port 8080)")
     parser.add_argument("--version", action="store_true",
                         help="Prints the version of Ziya")
+    parser.add_argument("--max-depth", type=int, default=15,
+                        help="Maximum depth for folder structure traversal (e.g., --max-depth 20)")
     return parser.parse_args()
 
 
@@ -37,6 +39,8 @@ def setup_environment(args):
     if args.model:
         os.environ["ZIYA_AWS_MODEL"] = args.model
 
+    os.environ["ZIYA_MAX_DEPTH"] = str(args.max_depth)
+
 
 def check_version_and_upgrade():
     current_version = get_current_version()

{ziya-0.1.40 → ziya-0.1.41}/app/server.py

@@ -43,7 +43,7 @@ async def favicon():
     return FileResponse('../templates/favicon.ico')
 
 
-def get_folder_structure(directory: str, ignored_patterns: List[Tuple[str, str]], max_depth: int = 12) -> Dict[str, Any]:
+def get_folder_structure(directory: str, ignored_patterns: List[Tuple[str, str]], max_depth: int) -> Dict[str, Any]:
     should_ignore_fn = parse_gitignore_patterns(ignored_patterns)
 
     def count_tokens(file_path: str) -> int:
@@ -87,8 +87,9 @@ def get_folder_structure(directory: str, ignored_patterns: List[Tuple[str, str]]
 async def get_folders():
     # pydevd_pycharm.settrace('localhost', port=59939, stdoutToServer=True, stderrToServer=True)
     user_codebase_dir = os.environ["ZIYA_USER_CODEBASE_DIR"]
+    max_depth = int(os.environ.get("ZIYA_MAX_DEPTH"))
     ignored_patterns: List[Tuple[str, str]] = get_ignored_patterns(user_codebase_dir)
-    return get_folder_structure(user_codebase_dir, ignored_patterns)
+    return get_folder_structure(user_codebase_dir, ignored_patterns, max_depth)
 
 
 @app.get('/api/default-included-folders')

{ziya-0.1.40 → ziya-0.1.41}/pyproject.toml

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