ziya 0.1.39__py3-none-any.whl → 0.1.41__py3-none-any.whl

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(

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

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

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ziya"
-version = "0.1.39"
+version = "0.1.41"
 description = ""
 authors = ["Vishnu Krishnaprasad <vishnukool@gmail.com>"]
 readme = "README.md"
@@ -22,6 +22,7 @@ langchainhub = ">=0.1.15"
 langchain-anthropic = ">=0.2,<0.3"
 langchain-cli = ">=0.0.15"
 pydevd-pycharm = "^243.18137.19"
+langchain-community = "^0.3.1"
 
 [build-system]
 requires = ["poetry-core"]

{ziya-0.1.39.dist-info → ziya-0.1.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ziya
-Version: 0.1.39
+Version: 0.1.41
 Summary: 
 Author: Vishnu Krishnaprasad
 Author-email: vishnukool@gmail.com
@@ -16,6 +16,7 @@ Requires-Dist: langchain (>=0.3,<0.4)
 Requires-Dist: langchain-anthropic (>=0.2,<0.3)
 Requires-Dist: langchain-aws (>=0.2,<0.3)
 Requires-Dist: langchain-cli (>=0.0.15)
+Requires-Dist: langchain-community (>=0.3.1,<0.4.0)
 Requires-Dist: langchainhub (>=0.1.15)
 Requires-Dist: pydantic (>=2.9.2,<3.0.0)
 Requires-Dist: pydevd-pycharm (>=243.18137.19,<244.0.0)
@@ -82,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.39.dist-info → ziya-0.1.41.dist-info}/RECORD

@@ -1,9 +1,9 @@
 app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 app/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-app/agents/agent.py,sha256=bjGH8IwcgAaR2T5kz0RzhhC17_pKKx3qk-eq7XXtRO4,3612
+app/agents/agent.py,sha256=Ck5MzWszl7hjGbHfS19aZDpNZiFVuIqtngbWVr4WkDM,3683
 app/agents/prompts.py,sha256=s1gtElttFwaXCrW6FHOlvTPXbguyuZvrd1BEGmLkfhU,2785
-app/main.py,sha256=8kNIuHBnbPC4ldA0wC7BhI9TzNjyDHZQX1caqhByEaY,4068
-app/server.py,sha256=EOuQ_DqNjYmD2prB4_ttiUxVIeSxGxno76feoD0zfXM,3488
+app/main.py,sha256=lk9wAwK6_9EkYyBsZEVNgM3FUY2ddK7xlgP0QXso9-A,4304
+app/server.py,sha256=uvZh24gNx_Z2-0IxYZkj6YEzDf0VjvOvsQFyOOhCtkM,3548
 app/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 app/utils/directory_util.py,sha256=2zxsSxSOZKesjhgjF3KguY__fC9XkjxHHnDbcxtnhXk,2762
 app/utils/gitignore_parser.py,sha256=Li3hQd60eclIuN2qxn8-NQ5sxRyMcnS88DFUxVHdx-A,7487
@@ -11,7 +11,7 @@ app/utils/langchain_validation_util.py,sha256=RgmKayKMApvUA7SPF_DrBAcYdIzzLUVJfn
 app/utils/logging_utils.py,sha256=8JEcc1t7L-a0G4HLmM8zFydNNSOd5l2-gkTxviLQUns,280
 app/utils/print_tree_util.py,sha256=O4Rb-GS8UODxfH7Z6bZGsr-opG6QLmvdaU1im5kh4IA,1419
 app/utils/version_util.py,sha256=NcvMWIImJDcO9K5OMX6jJD4q6Zb8Y117Mu8sWIkhxEc,631
-pyproject.toml,sha256=F9Lvrij_ATgn419v5Jp3nx-DW4-JtepMncgfJ3-nnDU,822
+pyproject.toml,sha256=cdLVN9nOqczPFHZdqy_oJWwIiDAHetwPsLOI157kCv0,853
 scripts.py,sha256=70BOvYoboMhl3Bxjy73m1ozmI8UV8JZVyi2RwUptkK8,809
 templates/asset-manifest.json,sha256=UKfHq1FKhejTUZgnopTN1Z3FZjnc92z9jh0UQF9_NLc,1157
 templates/favicon.ico,sha256=HgB8xAZdDHFK2lODUsp2H_Dds_i14pnpydx7NEJrNrU,15086
@@ -29,8 +29,8 @@ templates/static/media/fa-solid-900.4d986b00ff9ca3828fbd.woff2,sha256=rhfBavvqIW
 templates/static/media/fa-solid-900.bacd5de623fb563b961a.ttf,sha256=tJkNDQxfXTjWLpNu6hIGdOWEx-6o3O44qXXAz5o3U5s,420332
 templates/static/media/fa-v4compatibility.c8e090db312b0bea2aa2.ttf,sha256=_49SX7BQxdJFGczI9XI9hbLlHt0_m8ZUivVa663U8mk,10832
 templates/static/media/fa-v4compatibility.cf7f5903d06b79ad60f1.woff2,sha256=x6hp-sopnRW-EKAfGdB2WnxNRtiSLZuTFyNcHkpvCYI,4792
-ziya-0.1.39.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
-ziya-0.1.39.dist-info/METADATA,sha256=s5Atd9hbCXgfzhLeNTebIxClbgkedfzD9_aosJzq9hw,2698
-ziya-0.1.39.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-ziya-0.1.39.dist-info/entry_points.txt,sha256=6-KUolj5XXeOPVCJGTJgUL_3lDVGxG-YtK5BTvFPyBg,147
-ziya-0.1.39.dist-info/RECORD,,
+ziya-0.1.41.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
+ziya-0.1.41.dist-info/METADATA,sha256=g4lVlVwIa5UMK8fxZzX1VYYOllNpCsUWWuFNIvirp0M,2829
+ziya-0.1.41.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+ziya-0.1.41.dist-info/entry_points.txt,sha256=6-KUolj5XXeOPVCJGTJgUL_3lDVGxG-YtK5BTvFPyBg,147
+ziya-0.1.41.dist-info/RECORD,,