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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of ziya might be problematic. Click here for more details.

Files changed (9) hide show
  1. app/agents/agent.py +5 -0
  2. app/main.py +1 -1
  3. app/utils/directory_util.py +5 -1
  4. pyproject.toml +1 -1
  5. {ziya-0.1.3.dist-info → ziya-0.1.4.dist-info}/METADATA +1 -1
  6. {ziya-0.1.3.dist-info → ziya-0.1.4.dist-info}/RECORD +9 -9
  7. {ziya-0.1.3.dist-info → ziya-0.1.4.dist-info}/LICENSE +0 -0
  8. {ziya-0.1.3.dist-info → ziya-0.1.4.dist-info}/WHEEL +0 -0
  9. {ziya-0.1.3.dist-info → ziya-0.1.4.dist-info}/entry_points.txt +0 -0
app/agents/agent.py CHANGED
@@ -1,4 +1,6 @@
1
1
  import os
2
+ import botocore
3
+
2
4
  from typing import Generator, List, Tuple, Set, Union
3
5
 
4
6
  import tiktoken
@@ -38,6 +40,9 @@ model = ChatBedrock(
38
40
  model_id=model_id,
39
41
  model_kwargs={"max_tokens": 4096},
40
42
  credentials_profile_name=aws_profile if aws_profile else None,
43
+ config=botocore.config.Config(
44
+ read_timeout=900
45
+ )
41
46
  )
42
47
 
43
48
 
app/main.py CHANGED
@@ -12,7 +12,7 @@ def main():
12
12
  help="List of directories to include (e.g., --include 'src,static'). Only directories for now")
13
13
  parser.add_argument("--profile", type=str, default=None,
14
14
  help="AWS profile to use (e.g., --profile ziya)")
15
- parser.add_argument("--model", type=str, choices=["sonnet", "haiku", "opus"], default="haiku",
15
+ parser.add_argument("--model", type=str, choices=["sonnet", "haiku", "opus"], default="sonnet",
16
16
  help="AWS Bedrock Model to use (e.g., --model sonnet)")
17
17
  parser.add_argument("--port", type=int, default=6969,
18
18
  help="Port number to run Ziya frontend on (e.g., --port 8080)")
app/utils/directory_util.py CHANGED
@@ -51,7 +51,11 @@ def get_complete_file_list(user_codebase_dir: str, ignored_patterns: List[str],
51
51
 
52
52
  for file in files:
53
53
  file_path = os.path.join(root, file)
54
- if not should_ignore(file_path):
54
+ if not should_ignore(file_path) and not is_image_file(file_path):
55
55
  file_set.add(file_path)
56
56
 
57
57
  return list(file_set)
58
+
59
+ def is_image_file(file_path: str) -> bool:
60
+ image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.svg', '.ico']
61
+ return any(file_path.lower().endswith(ext) for ext in image_extensions)
pyproject.toml CHANGED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "ziya"
3
- version = "0.1.3"
3
+ version = "0.1.4"
4
4
  description = ""
5
5
  authors = ["Vishnu Krishnaprasad <vishnukool@gmail.com>"]
6
6
  readme = "README.md"
{ziya-0.1.3.dist-info → ziya-0.1.4.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ziya
3
- Version: 0.1.3
3
+ Version: 0.1.4
4
4
  Summary:
5
5
  Author: Vishnu Krishnaprasad
6
6
  Author-email: vishnukool@gmail.com
{ziya-0.1.3.dist-info → ziya-0.1.4.dist-info}/RECORD RENAMED
@@ -1,22 +1,22 @@
1
1
  app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
2
  app/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
- app/agents/agent.py,sha256=A1GCNwPXoPY0wg9ukHyox8WuJu-i1o0ArZMpOF9I_kQ,4266
3
+ app/agents/agent.py,sha256=Qb1xgZ7fkpLSbKzSsygze7UL2H8jMtnOiTD-Gm8uXWU,4349
4
4
  app/agents/prompts.py,sha256=O54rH2KupyY7ZSjPEvm3j9DSGjdL3jJOz8F9PTcePQQ,1674
5
- app/main.py,sha256=2-5fCFw2pMYNzyX25yxh9JnGzfV9Vvega8J59RT02MM,1704
5
+ app/main.py,sha256=muUq4TlNUHJPP4AXaVrMsWUzbNo_aYk6XLDk2wAMqgU,1705
6
6
  app/server.py,sha256=YTEHYUsFQhgkGRHZ_y7K3Ju3SmUIbVxLSdgrgi1BuEU,676
7
7
  app/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
8
- app/utils/directory_util.py,sha256=qx-LCHLZgFSz8WJTJcAkRnfxtg1eQgRX55dMJ2bTI3I,2278
8
+ app/utils/directory_util.py,sha256=fdYumUhLcmbaGI0OvGY13CkQSKxrCZi6ZNwhz9_oJmQ,2512
9
9
  app/utils/gitignore_parser.py,sha256=ksAAvXr8_GFh94cvOdkLmES5k3xxNxW2_tQuQtialIA,7496
10
10
  app/utils/logging_utils.py,sha256=8JEcc1t7L-a0G4HLmM8zFydNNSOd5l2-gkTxviLQUns,280
11
11
  app/utils/print_tree_util.py,sha256=O4Rb-GS8UODxfH7Z6bZGsr-opG6QLmvdaU1im5kh4IA,1419
12
- pyproject.toml,sha256=1MA83_N7UL7fG8SEVuiHIJdsrG6pd5UFr8SOR6KOSrQ,716
12
+ pyproject.toml,sha256=-81QM7yQm3lDFoO6M5dgR-mbMITwEpUfc3iZbGh0W9g,716
13
13
  static/app.js,sha256=qz2F0e5LxuVFI3MXWvcON9l1cN0J_ODdQOwbgT3ZqWA,3622
14
14
  static/favicon.ico,sha256=HgB8xAZdDHFK2lODUsp2H_Dds_i14pnpydx7NEJrNrU,15086
15
15
  static/sendPayload.js,sha256=VnypSWF7t8bv83026LGU3GMyr9tMrp05hmWATcbCG4k,1956
16
16
  static/ziya.css,sha256=OaieSXw_7LOk0WNNi-1mPsAaS3rOxhHycKFgFDoiZB0,1395
17
17
  templates/index.html,sha256=8auy9lGLdcDzBkzpjYB1ygrAVbXpo0rqQ8p1FgMEeck,800
18
- ziya-0.1.3.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
19
- ziya-0.1.3.dist-info/METADATA,sha256=d0kWH3BeHMjSyM8VhoUfZSy0lg8q1ks5hE5hg4w8Ga8,2921
20
- ziya-0.1.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
21
- ziya-0.1.3.dist-info/entry_points.txt,sha256=1HspxMqCYRli3SqM6CqmT6gsW_G14G4ab3sHlrYUCAA,38
22
- ziya-0.1.3.dist-info/RECORD,,
18
+ ziya-0.1.4.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
19
+ ziya-0.1.4.dist-info/METADATA,sha256=e57AePaIYYdGb5OgVjJbfoiaW1f9BWBI4WwstFY5Q-U,2921
20
+ ziya-0.1.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
21
+ ziya-0.1.4.dist-info/entry_points.txt,sha256=1HspxMqCYRli3SqM6CqmT6gsW_G14G4ab3sHlrYUCAA,38
22
+ ziya-0.1.4.dist-info/RECORD,,
{ziya-0.1.3.dist-info → ziya-0.1.4.dist-info}/LICENSE RENAMED
File without changes
{ziya-0.1.3.dist-info → ziya-0.1.4.dist-info}/WHEEL RENAMED
File without changes