ziya 0.1.3__tar.gz → 0.1.5__tar.gz

{ziya-0.1.3 → ziya-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ziya
-Version: 0.1.3
+Version: 0.1.5
 Summary: 
 Author: Vishnu Krishnaprasad
 Author-email: vishnukool@gmail.com

{ziya-0.1.3 → ziya-0.1.5}/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
+    )
 )
 
 

{ziya-0.1.3 → ziya-0.1.5}/app/main.py

@@ -1,7 +1,25 @@
 import os
+import subprocess
+import sys
+from typing import Optional
+
 from langchain_cli.cli import serve
 import argparse
 
+from app.utils.logging_utils import logger
+from app.utils.version_util import get_current_version, get_latest_version
+
+
+def update_package(current_version: str, latest_version: Optional[str]) -> None:
+    try:
+        logger.info(f"Updating ziya from {current_version} to {latest_version}")
+        subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', 'ziya'])
+        logger.info("Update completed. Next time you run ziya it will be with the latest version.")
+
+    except Exception as e:
+        logger.info(f"Unexpected error upgrading ziya: {e}")
+
+
 def main():
     os.environ["ZIYA_USER_CODEBASE_DIR"] = os.getcwd()
 
@@ -12,12 +30,26 @@ def main():
                         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)")
+    parser.add_argument("--version", action="store_true",
+                        help="Prints the version of Ziya")
     args = parser.parse_args()
 
+    current_version = get_current_version()
+    latest_version = get_latest_version()
+
+    if args.version:
+        print(f"Ziya version {current_version}")
+        return
+
+    if latest_version and current_version != latest_version:
+        update_package(current_version, latest_version)
+    else:
+        logger.info(f"Ziya version {current_version} is up to date.")
+
     additional_excluded_dirs = ','.join([item for item in args.exclude])
     os.environ["ZIYA_ADDITIONAL_EXCLUDE_DIRS"] = additional_excluded_dirs
 
@@ -35,4 +67,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

{ziya-0.1.3 → ziya-0.1.5}/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)

ziya-0.1.5/app/utils/version_util.py

@@ -0,0 +1,23 @@
+import requests
+from packaging import version
+import subprocess
+import sys
+from typing import Optional
+
+
+def get_latest_version() -> Optional[str]:
+    try:
+        response = requests.get('https://pypi.org/pypi/ziya/json')
+        response.raise_for_status()
+        return str(version.parse(response.json()['info']['version']))
+    except requests.exceptions.RequestException:
+        return None
+
+
+def update_package(package_name: str) -> None:
+    subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', package_name])
+
+
+def get_current_version() -> str:
+    from importlib.metadata import version
+    return str(version('ziya'))

{ziya-0.1.3 → ziya-0.1.5}/pyproject.toml

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