ziya 0.1.36__py3-none-any.whl → 0.1.37__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 (7) hide show
  1. app/agents/agent.py +6 -6
  2. pyproject.toml +1 -1
  3. {ziya-0.1.36.dist-info → ziya-0.1.37.dist-info}/METADATA +1 -1
  4. {ziya-0.1.36.dist-info → ziya-0.1.37.dist-info}/RECORD +7 -7
  5. {ziya-0.1.36.dist-info → ziya-0.1.37.dist-info}/LICENSE +0 -0
  6. {ziya-0.1.36.dist-info → ziya-0.1.37.dist-info}/WHEEL +0 -0
  7. {ziya-0.1.36.dist-info → ziya-0.1.37.dist-info}/entry_points.txt +0 -0
app/agents/agent.py CHANGED
@@ -5,7 +5,7 @@ import botocore
5
5
  import tiktoken
6
6
  from langchain.agents import AgentExecutor
7
7
  from langchain.agents.format_scratchpad import format_xml
8
- from langchain_aws import ChatBedrock
8
+ from langchain_community.chat_models import BedrockChat
9
9
  from langchain_community.document_loaders import TextLoader
10
10
  from langchain_core.messages import AIMessage, HumanMessage
11
11
  from langchain_core.pydantic_v1 import BaseModel, Field
@@ -29,14 +29,14 @@ if aws_profile:
29
29
  else:
30
30
  logger.info("No AWS profile specified via --aws-profile flag, using default credentials")
31
31
  model_id = {
32
- "sonnet3.5": "anthropic.claude-3-5-sonnet-20240620-v1:0",
33
- "sonnet": "anthropic.claude-3-sonnet-20240229-v1:0",
34
- "haiku": "anthropic.claude-3-haiku-20240307-v1:0",
35
- "opus": "anthropic.claude-3-opus-20240229-v1:0",
32
+ "sonnet3.5": "us.anthropic.claude-3-5-sonnet-20240620-v1:0",
33
+ "sonnet": "us.anthropic.claude-3-sonnet-20240229-v1:0",
34
+ "haiku": "us.anthropic.claude-3-haiku-20240307-v1:0",
35
+ "opus": "us.anthropic.claude-3-opus-20240229-v1:0",
36
36
  }[os.environ.get("ZIYA_AWS_MODEL", "sonnet3.5")]
37
37
  logger.info(f"Using Claude Model: {model_id}")
38
38
 
39
- model = ChatBedrock(
39
+ model = BedrockChat(
40
40
  model_id=model_id,
41
41
  model_kwargs={"max_tokens": 4096, "temperature": 0.3, "top_k": 15},
42
42
  credentials_profile_name=aws_profile if aws_profile else None,
pyproject.toml CHANGED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "ziya"
3
- version = "0.1.36"
3
+ version = "0.1.37"
4
4
  description = ""
5
5
  authors = ["Vishnu Krishnaprasad <vishnukool@gmail.com>"]
6
6
  readme = "README.md"
{ziya-0.1.36.dist-info → ziya-0.1.37.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ziya
3
- Version: 0.1.36
3
+ Version: 0.1.37
4
4
  Summary:
5
5
  Author: Vishnu Krishnaprasad
6
6
  Author-email: vishnukool@gmail.com
{ziya-0.1.36.dist-info → ziya-0.1.37.dist-info}/RECORD RENAMED
@@ -1,6 +1,6 @@
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=OwO1pWhQc8gAmNQO5mOxRe7Ravc_LHZjXOn5uKjiz8Y,3618
3
+ app/agents/agent.py,sha256=zRaeEy-1BmiTqovrrpJgKIY0J5AzTU5RKOU831j1anc,3648
4
4
  app/agents/prompts.py,sha256=s1gtElttFwaXCrW6FHOlvTPXbguyuZvrd1BEGmLkfhU,2785
5
5
  app/main.py,sha256=8kNIuHBnbPC4ldA0wC7BhI9TzNjyDHZQX1caqhByEaY,4068
6
6
  app/server.py,sha256=EOuQ_DqNjYmD2prB4_ttiUxVIeSxGxno76feoD0zfXM,3488
@@ -11,7 +11,7 @@ app/utils/langchain_validation_util.py,sha256=RgmKayKMApvUA7SPF_DrBAcYdIzzLUVJfn
11
11
  app/utils/logging_utils.py,sha256=8JEcc1t7L-a0G4HLmM8zFydNNSOd5l2-gkTxviLQUns,280
12
12
  app/utils/print_tree_util.py,sha256=O4Rb-GS8UODxfH7Z6bZGsr-opG6QLmvdaU1im5kh4IA,1419
13
13
  app/utils/version_util.py,sha256=NcvMWIImJDcO9K5OMX6jJD4q6Zb8Y117Mu8sWIkhxEc,631
14
- pyproject.toml,sha256=LmhYMMc04dm8iJnIFcHG1cgajPZyx8hKdChqSwT9Yp0,804
14
+ pyproject.toml,sha256=MPmcDze9hGWcr2n7Z3mPZzcURQhSKLiFmFE6as83p5U,804
15
15
  scripts.py,sha256=70BOvYoboMhl3Bxjy73m1ozmI8UV8JZVyi2RwUptkK8,809
16
16
  templates/asset-manifest.json,sha256=UKfHq1FKhejTUZgnopTN1Z3FZjnc92z9jh0UQF9_NLc,1157
17
17
  templates/favicon.ico,sha256=HgB8xAZdDHFK2lODUsp2H_Dds_i14pnpydx7NEJrNrU,15086
@@ -29,8 +29,8 @@ templates/static/media/fa-solid-900.4d986b00ff9ca3828fbd.woff2,sha256=rhfBavvqIW
29
29
  templates/static/media/fa-solid-900.bacd5de623fb563b961a.ttf,sha256=tJkNDQxfXTjWLpNu6hIGdOWEx-6o3O44qXXAz5o3U5s,420332
30
30
  templates/static/media/fa-v4compatibility.c8e090db312b0bea2aa2.ttf,sha256=_49SX7BQxdJFGczI9XI9hbLlHt0_m8ZUivVa663U8mk,10832
31
31
  templates/static/media/fa-v4compatibility.cf7f5903d06b79ad60f1.woff2,sha256=x6hp-sopnRW-EKAfGdB2WnxNRtiSLZuTFyNcHkpvCYI,4792
32
- ziya-0.1.36.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
33
- ziya-0.1.36.dist-info/METADATA,sha256=ZeQaxKU309-stcYwSsS3iGn6T5vsspLKaVHAmElCOCU,2694
34
- ziya-0.1.36.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
35
- ziya-0.1.36.dist-info/entry_points.txt,sha256=6-KUolj5XXeOPVCJGTJgUL_3lDVGxG-YtK5BTvFPyBg,147
36
- ziya-0.1.36.dist-info/RECORD,,
32
+ ziya-0.1.37.dist-info/LICENSE,sha256=8CfErGEG13yY1Z-CDlIhAQawX2KOv-QI_2Ge2z6x8SU,1064
33
+ ziya-0.1.37.dist-info/METADATA,sha256=V7WH5PgRzt8eIQ7vJc_cnp_WNqL0byhQ_BTeh1p_6Cw,2694
34
+ ziya-0.1.37.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
35
+ ziya-0.1.37.dist-info/entry_points.txt,sha256=6-KUolj5XXeOPVCJGTJgUL_3lDVGxG-YtK5BTvFPyBg,147
36
+ ziya-0.1.37.dist-info/RECORD,,
{ziya-0.1.36.dist-info → ziya-0.1.37.dist-info}/LICENSE RENAMED
File without changes
{ziya-0.1.36.dist-info → ziya-0.1.37.dist-info}/WHEEL RENAMED
File without changes