ziya 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

app/agents/agent.py

@@ -1,4 +1,6 @@
 import os
+import botocore
+
 from typing import Generator, List, Tuple, Set, Union
 
 import tiktoken
@@ -38,6 +40,9 @@ model = ChatBedrock(
     model_id=model_id,
     model_kwargs={"max_tokens": 4096},
     credentials_profile_name=aws_profile if aws_profile else None,
+    config=botocore.config.Config(
+        read_timeout=900
+    )
 )
 
 

app/main.py

@@ -8,11 +8,11 @@ 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("--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",
+    parser.add_argument("--model", type=str, choices=["sonnet", "haiku", "opus"], default="sonnet",
                         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)")
@@ -21,7 +21,7 @@ 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])
+    additional_included_dirs = ','.join([item for item in args.include])
     os.environ["ZIYA_ADDITIONAL_INCLUDE_DIRS"] = additional_included_dirs
 
     if args.profile:

app/utils/directory_util.py

@@ -51,7 +51,11 @@ def get_complete_file_list(user_codebase_dir: str, ignored_patterns: List[str],
 
             for file in files:
                 file_path = os.path.join(root, file)
-                if not should_ignore(file_path):
+                if not should_ignore(file_path) and not is_image_file(file_path):
                     file_set.add(file_path)
 
     return list(file_set)
+
+def is_image_file(file_path: str) -> bool:
+    image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.svg', '.ico']
+    return any(file_path.lower().endswith(ext) for ext in image_extensions)

pyproject.toml

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

{ziya-0.1.2.dist-info → ziya-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ziya
-Version: 0.1.2
+Version: 0.1.4
 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.2.dist-info → ziya-0.1.4.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=A1GCNwPXoPY0wg9ukHyox8WuJu-i1o0ArZMpOF9I_kQ,4266
+app/agents/agent.py,sha256=Qb1xgZ7fkpLSbKzSsygze7UL2H8jMtnOiTD-Gm8uXWU,4349
 app/agents/prompts.py,sha256=O54rH2KupyY7ZSjPEvm3j9DSGjdL3jJOz8F9PTcePQQ,1674
-app/main.py,sha256=p7FGRXeN-rtC56dHQiivEu5uKKniU_ULvUjsjFEe4-k,1688
+app/main.py,sha256=muUq4TlNUHJPP4AXaVrMsWUzbNo_aYk6XLDk2wAMqgU,1705
 app/server.py,sha256=YTEHYUsFQhgkGRHZ_y7K3Ju3SmUIbVxLSdgrgi1BuEU,676
 app/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-app/utils/directory_util.py,sha256=qx-LCHLZgFSz8WJTJcAkRnfxtg1eQgRX55dMJ2bTI3I,2278
+app/utils/directory_util.py,sha256=fdYumUhLcmbaGI0OvGY13CkQSKxrCZi6ZNwhz9_oJmQ,2512
 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=XQTaNA-iGsOAzHReXPn4heww08imAKKAeR4rXw8svu4,716
+pyproject.toml,sha256=-81QM7yQm3lDFoO6M5dgR-mbMITwEpUfc3iZbGh0W9g,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.2.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
-ziya-0.1.2.dist-info/METADATA,sha256=S24oPWhu0DnlG7THzeRUald_tyGxDK8C4ajFr3grIDA,2563
-ziya-0.1.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-ziya-0.1.2.dist-info/entry_points.txt,sha256=1HspxMqCYRli3SqM6CqmT6gsW_G14G4ab3sHlrYUCAA,38
-ziya-0.1.2.dist-info/RECORD,,
+ziya-0.1.4.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
+ziya-0.1.4.dist-info/METADATA,sha256=e57AePaIYYdGb5OgVjJbfoiaW1f9BWBI4WwstFY5Q-U,2921
+ziya-0.1.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+ziya-0.1.4.dist-info/entry_points.txt,sha256=1HspxMqCYRli3SqM6CqmT6gsW_G14G4ab3sHlrYUCAA,38
+ziya-0.1.4.dist-info/RECORD,,