ziya 0.1.37__tar.gz → 0.1.39__tar.gz

{ziya-0.1.37 → ziya-0.1.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ziya
-Version: 0.1.37
+Version: 0.1.39
 Summary: 
 Author: Vishnu Krishnaprasad
 Author-email: vishnukool@gmail.com
@@ -12,14 +12,14 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: boto3 (>=1.34.88,<2.0.0)
 Requires-Dist: jinja2 (>=3.1.3,<4.0.0)
-Requires-Dist: langchain (>=0.1,<0.2)
-Requires-Dist: langchain-anthropic (>=0.1.4,<0.2.0)
-Requires-Dist: langchain-aws (>=0.1.0,<0.2.0)
+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: langchainhub (>=0.1.15)
-Requires-Dist: pydantic (<2)
-Requires-Dist: pydevd-pycharm (>=241.15989.57,<242.0.0)
-Requires-Dist: tiktoken (>=0.6.0,<0.7.0)
+Requires-Dist: pydantic (>=2.9.2,<3.0.0)
+Requires-Dist: pydevd-pycharm (>=243.18137.19,<244.0.0)
+Requires-Dist: tiktoken (>=0.8.0,<0.9.0)
 Requires-Dist: uvicorn (>=0.23.2,<0.24.0)
 Description-Content-Type: text/markdown
 

{ziya-0.1.37 → ziya-0.1.39}/app/agents/agent.py

@@ -5,10 +5,10 @@ import botocore
 import tiktoken
 from langchain.agents import AgentExecutor
 from langchain.agents.format_scratchpad import format_xml
-from langchain_community.chat_models import BedrockChat
+from langchain_aws import ChatBedrock
 from langchain_community.document_loaders import TextLoader
 from langchain_core.messages import AIMessage, HumanMessage
-from langchain_core.pydantic_v1 import BaseModel, Field
+from pydantic import BaseModel, Field
 
 from app.agents.prompts import conversational_prompt, parse_output
 from app.utils.logging_utils import logger
@@ -30,13 +30,13 @@ 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",
+    "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",
-    "opus": "us.anthropic.claude-3-opus-20240229-v1:0",
 }[os.environ.get("ZIYA_AWS_MODEL", "sonnet3.5")]
 logger.info(f"Using Claude Model: {model_id}")
 
-model = BedrockChat(
+model = ChatBedrock(
     model_id=model_id,
     model_kwargs={"max_tokens": 4096, "temperature": 0.3, "top_k": 15},
     credentials_profile_name=aws_profile if aws_profile else None,

{ziya-0.1.37 → ziya-0.1.39}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ziya"
-version = "0.1.37"
+version = "0.1.39"
 description = ""
 authors = ["Vishnu Krishnaprasad <vishnukool@gmail.com>"]
 readme = "README.md"
@@ -12,16 +12,16 @@ packages = [
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"
 uvicorn = "^0.23.2"
-pydantic = "<2"
+pydantic = "^2.9.2"
 jinja2 = "^3.1.3"
-tiktoken = "^0.6.0"
+tiktoken = "^0.8.0"
 boto3 = "^1.34.88"
-langchain-aws = "^0.1.0"
-langchain = "^0.1"
+langchain-aws = ">=0.2,<0.3"
+langchain = ">=0.3,<0.4"
 langchainhub = ">=0.1.15"
-langchain-anthropic = "^0.1.4"
+langchain-anthropic = ">=0.2,<0.3"
 langchain-cli = ">=0.0.15"
-pydevd-pycharm = "^241.15989.57"
+pydevd-pycharm = "^243.18137.19"
 
 [build-system]
 requires = ["poetry-core"]